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 inference_img.py

Browse Source
This commit is contained in:
hzwer
2022-05-05 14:30:35 +08:00
committed by GitHub
parent 1415f00957
commit 2ebd789972
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
inference_img.py
View File

@@ -23,6 +23,7 @@ parser.add_argument('--model', dest='modelDir', type=str, default='train_log', h
args = parser.parse_args()
try:
try:
try:
from model.RIFE_HDv2 import Model
model = Model()
@@ -33,11 +34,16 @@ try:
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v3.x HD model.")
except:
except:
from model.RIFE_HD import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded v1.x HD model")
expect:
from model.RIFE import Model
model = Model()
model.load_model(args.modelDir, -1)
print("Loaded ArXiv-RIFE model")
model.eval()
model.device()