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

Added Docker setup

Browse Source
This commit is contained in:
ko1N
2021-01-17 15:24:57 +01:00
parent 9239bf4df4
commit ea994bbe34
7 changed files with 49 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/MiddleBury_Other.py
View File

@@ -12,7 +12,7 @@ from model.RIFE import Model
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Model()
model.load_model('./train_log')
model.load_model(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'train_log'))
model.eval()
model.device()

2
benchmark/Vimeo90K.py
View File

@@ -13,7 +13,7 @@ from model.RIFE import Model
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Model()
model.load_model('./train_log')
model.load_model(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'train_log'))
model.eval()
model.device()