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

Update RIFE.py

Browse Source
This commit is contained in:
hzwer
2025-07-25 17:02:10 +08:00
committed by GitHub
parent faf78cd5a1
commit eff2c82671
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model/RIFE.py
View File

@@ -53,7 +53,9 @@ class Model:
if rank == 0:
torch.save(self.flownet.state_dict(),'{}/flownet.pkl'.format(path))
def inference(self, img0, img1, scale=1, scale_list=[4, 2, 1], TTA=False, timestep=0.5):
def inference(self, img0, img1, scale=1, scale_list=None, TTA=False, timestep=0.5):
if scale_list is None:
scale_list = [4, 2, 1]
for i in range(3):
scale_list[i] = scale_list[i] * 1.0 / scale
imgs = torch.cat((img0, img1), 1)