Skip to content

Commit fc8b2bb

Browse files
committed
removed commented lines of code in rotate and reflect function
1 parent 6fbc772 commit fc8b2bb

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

XPointMLTest.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,6 @@ def rotate(frameData,deg):
7777
plotSimple(all[2], f"{frameData['fnum']}_rotation{deg}_all2.png")
7878
plotSimple(all[3], f"{frameData['fnum']}_rotation{deg}_all3.png")
7979

80-
# mask = v2.functional.rotate(frameData["mask"], deg, v2.InterpolationMode.BILINEAR)
8180
mask = v2.functional.rotate(frameData["mask"], deg, v2.InterpolationMode.NEAREST)
8281
return {
8382
"fnum": frameData["fnum"],
@@ -96,11 +95,7 @@ def reflect(frameData,axis):
9695
if axis not in [0,1]:
9796
print(f"invalid reflection axis specified... exiting")
9897
sys.exit()
99-
100-
# psi = torch.flip(frameData["psi"][0], dims=(axis,)).unsqueeze(0)
101-
# all = torch.flip(frameData["all"], dims=(axis,))
102-
# mask = torch.flip(frameData["mask"][0], dims=(axis,)).unsqueeze(0)
103-
98+
10499
psi = torch.flip(frameData["psi"], dims=(axis+1,))
105100
all = torch.flip(frameData["all"], dims=(axis+1,))
106101
mask = torch.flip(frameData["mask"], dims=(axis+1,))

0 commit comments

Comments
 (0)