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

Reorg benchmark

Browse Source
This commit is contained in:
hzwer
2020-11-26 18:39:18 +08:00
parent 344d8100c0
commit d3f99506b1
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Vimeo90K_benchmark.py → benchmark/Vimeo90K_benchmark.py
View File

@@ -1,4 +1,6 @@
import os import os
import sys
sys.path.append('.')
import cv2 import cv2
import math import math
import torch import torch
@@ -7,7 +9,6 @@ import numpy as np
from torch.nn import functional as F from torch.nn import functional as F
from pytorch_msssim import ssim_matlab from pytorch_msssim import ssim_matlab
from model.RIFE import Model from model.RIFE import Model
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Model() model = Model()

0
pytorch_msssim/__init__.py → benchmark/pytorch_msssim/__init__.py
View File