mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-16 16:37:51 +01:00
Reorg benchmark
This commit is contained in:
@@ -1,4 +1,6 @@
|
||||
import os
|
||||
import sys
|
||||
sys.path.append('.')
|
||||
import cv2
|
||||
import math
|
||||
import torch
|
||||
@@ -7,7 +9,6 @@ import numpy as np
|
||||
from torch.nn import functional as F
|
||||
from pytorch_msssim import ssim_matlab
|
||||
from model.RIFE import Model
|
||||
|
||||
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
||||
|
||||
model = Model()
|
||||
Reference in New Issue
Block a user