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

Update RIFE.py

Browse Source
This commit is contained in:
hzwer
2021-06-28 15:33:21 +08:00
committed by GitHub
parent b7b503dc69
commit 856d550d66
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model/RIFE.py
View File

@@ -183,7 +183,7 @@ class Model:
else: else:
return pred return pred
def inference(self, img0, img1): def inference(self, img0, img1, scale=None):
imgs = torch.cat((img0, img1), 1) imgs = torch.cat((img0, img1), 1)
flow, _ = self.flownet(torch.cat((img0, img1), 1)) flow, _ = self.flownet(torch.cat((img0, img1), 1))
return self.predict(imgs, flow, training=False) return self.predict(imgs, flow, training=False)