Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2025-12-16 08:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix scale mode

Browse Source
This commit is contained in:
hzwer
2021-02-27 19:54:01 +08:00
parent f75bbacbd1
commit e59b084b7d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model/IFNet_HDv2.py
View File

@@ -42,14 +42,14 @@ class IFBlock(nn.Module):
def forward(self, x, scale=1.0): def forward(self, x, scale=1.0):
scale = self.scale / scale scale = self.scale / scale
if scale != 1.0: if scale != 1.0:
x = F.interpolate(x, scale_factor=1. / self.scale, mode="bilinear", x = F.interpolate(x, scale_factor=1. / scale, mode="bilinear",
align_corners=False) align_corners=False)
x = self.conv0(x) x = self.conv0(x)
x = self.convblock(x) x = self.convblock(x)
x = self.conv1(x) x = self.conv1(x)
flow = x flow = x
if scale != 1.0: if scale != 1.0:
flow = F.interpolate(flow, scale_factor=self.scale, mode="bilinear", flow = F.interpolate(flow, scale_factor=scale, mode="bilinear",
align_corners=False) align_corners=False)
return flow return flow