File tree Expand file tree Collapse file tree 1 file changed +1
-6
lines changed
Expand file tree Collapse file tree 1 file changed +1
-6
lines changed Original file line number Diff line number Diff 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 ,))
You can’t perform that action at this time.
0 commit comments