Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 04:19:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Release v2 model

Browse Source
This commit is contained in:
hzwer
2021-02-09 09:32:04 +08:00
parent cf104c9b0a
commit 4b93d33eb2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
inference_img.py
View File

@@ -3,7 +3,7 @@ import cv2
import torch
import argparse
from torch.nn import functional as F
from model.RIFE_HD import Model
from model.RIFE_HDv2 import Model
import warnings
warnings.filterwarnings("ignore")

2
inference_video.py
View File

@@ -76,7 +76,7 @@ assert (not args.video is None or not args.img is None)
if not args.img is None:
args.png = True
from model.RIFE_HD import Model
from model.RIFE_HDv2 import Model
model = Model()
model.load_model(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'train_log'), -1)
model.eval()