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

12
inference_video.py
View File

@@ -87,7 +87,8 @@ if torch.cuda.is_available():
try:
try:
from model.oldmodel.RIFE_HDv2 import Model
try:
from model.RIFE_HDv2 import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v2.x HD model.")
@@ -96,11 +97,16 @@ try:
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v3.x HD model.")
except:
from model.oldmodel.RIFE_HD import Model
except:
from model.RIFE_HD import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v1.x HD model")
except:
from model.RIFE import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded ArXiv-RIFE model")
model.eval()
model.device()