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

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:
try: try:
from model.oldmodel.RIFE_HDv2 import Model try:
model = Model() from model.RIFE_HDv2 import Model
model.load_model(args.modelDir, -1) model = Model()
print("Loaded v2.x HD 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: except:
from train_log.RIFE_HDv3 import Model from model.RIFE_HD import Model
model = Model() model = Model()
model.load_model(args.modelDir, -1) model.load_model(args.modelDir, -1)
print("Loaded v3.x HD model.") print("Loaded v1.x HD model")
except: except:
from model.oldmodel.RIFE_HD import Model from model.RIFE import Model
model = Model() model = Model()
model.load_model(args.modelDir, -1) model.load_model(args.modelDir, -1)
print("Loaded v1.x HD model") print("Loaded ArXiv-RIFE model")
model.eval() model.eval()
model.device() model.device()