Move ssim script

benchmark/ATD12K.py

@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

benchmark/HD.py

@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 from skimage.color import rgb2yuv, yuv2rgb
 from yuv_frame_io import YUV_Read,YUV_Write

benchmark/HD_multi.py

@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 from skimage.color import rgb2yuv, yuv2rgb
 from yuv_frame_io import YUV_Read,YUV_Write

benchmark/MiddleBury_Other.py

@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

benchmark/UCF101.py

@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

benchmark/Vimeo90K.py

@@ -7,8 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
-# from model.RIFE2F15C import Model
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

inference_video.py

@@ -9,7 +9,7 @@ import warnings
 import _thread
 import skvideo.io
 from queue import Queue, Empty
-from benchmark.pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 
 warnings.filterwarnings("ignore")