davrot/gevi
1
0
Fork 0
mirror of https://github.com/davrot/gevi.git synced 2025-06-03 22:00:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-07-10 13:06:54 +02:00 • committed by GitHub
parent cc7de6dbbb
commit 71c2784437
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 97 additions and 94 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ImageAlignment.py
View file

@ -762,7 +762,7 @@ class ImageAlignment(torch.nn.Module):
bgval: torch.Tensor | None = None,
invert=False,
) -> torch.Tensor:
if invert is True:
if invert:
if scale is not None:
scale = 1.0 / scale
if angle is not None:
@ -927,7 +927,7 @@ class ImageAlignment(torch.nn.Module):
if succ2[pos] > succ[pos]:
pick_rotated = True
if pick_rotated is True:
if pick_rotated:
tvec[pos, :] = tvec2[pos, :]
succ[pos] = succ2[pos]
angle[pos] += 180