From 76b69a25c0acd97e4e27441f3232b319381f36c8 Mon Sep 17 00:00:00 2001 From: Qunliang Xing Date: Thu, 23 Mar 2023 10:30:34 +0800 Subject: [PATCH] Fix warp typo --- mmedit/models/editors/tof/tof_vfi_net.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mmedit/models/editors/tof/tof_vfi_net.py b/mmedit/models/editors/tof/tof_vfi_net.py index e9faceec6a..badea6a085 100644 --- 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