Merge 76b69a2 into 10198b7

open-mmlab · Mar 23, 2023 · 9b611f8 · 9b611f8
2 parents 10198b7 + 76b69a2
commit 9b611f8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mmedit/models/editors/tof/tof_vfi_net.py b/mmedit/models/editors/tof/tof_vfi_net.py
@@ -60,11 +60,11 @@ def forward(self, imgs):
         flow_10 = self.spynet(imgs[:, 0], imgs[:, 1]).permute(0, 2, 3, 1)
         flow_01 = self.spynet(imgs[:, 1], imgs[:, 0]).permute(0, 2, 3, 1)
 
-        wrap_frame0 = flow_warp(imgs[:, 0], flow_01 / 2)
-        wrap_frame1 = flow_warp(imgs[:, 1], flow_10 / 2)
+        warp_frame0 = flow_warp(imgs[:, 0], flow_01 / 2)
+        warp_frame1 = flow_warp(imgs[:, 1], flow_10 / 2)
 
-        wrap_frames = torch.stack([wrap_frame0, wrap_frame1], dim=1)
-        output = self.resnet(wrap_frames)
+        warp_frames = torch.stack([warp_frame0, warp_frame1], dim=1)
+        output = self.resnet(warp_frames)
 
         return output