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

Update inference_video.py

Browse Source
This commit is contained in:
hzwer
2022-05-05 14:40:11 +08:00
committed by GitHub
parent cc612e9d45
commit d6a21eac6b
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
inference_video.py
View File

@@ -87,20 +87,26 @@ if torch.cuda.is_available():
try:
try:
from model.oldmodel.RIFE_HDv2 import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v2.x HD model.")
try:
from model.RIFE_HDv2 import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v2.x HD model.")
except:
from train_log.RIFE_HDv3 import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v3.x HD model.")
except:
from train_log.RIFE_HDv3 import Model
from model.RIFE_HD import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v3.x HD model.")
print("Loaded v1.x HD model")
except:
from model.oldmodel.RIFE_HD import Model
from model.RIFE import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v1.x HD model")
print("Loaded ArXiv-RIFE model")
model.eval()
model.device()