From f6d90d988d74845a7a77c59ce6ee9b3e33cb412d Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Thu, 17 Jun 2021 11:31:20 +0800
Subject: [PATCH] Move ssim script

---
 benchmark/ATD12K.py                             | 2 +-
 benchmark/HD.py                                 | 2 +-
 benchmark/HD_multi.py                           | 2 +-
 benchmark/MiddleBury_Other.py                   | 2 +-
 benchmark/UCF101.py                             | 2 +-
 benchmark/Vimeo90K.py                           | 3 +--
 inference_video.py                              | 2 +-
 {benchmark => model}/pytorch_msssim/__init__.py | 0
 8 files changed, 7 insertions(+), 8 deletions(-)
 rename {benchmark => model}/pytorch_msssim/__init__.py (100%)

diff --git a/benchmark/ATD12K.py b/benchmark/ATD12K.py
index 0d8be5e..adf17a6 100644
--- a/benchmark/ATD12K.py
+++ b/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")
 
diff --git a/benchmark/HD.py b/benchmark/HD.py
index f91153d..981f07e 100644
--- a/benchmark/HD.py
+++ b/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
diff --git a/benchmark/HD_multi.py b/benchmark/HD_multi.py
index 988ddc7..3ff1ffe 100644
--- a/benchmark/HD_multi.py
+++ b/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
diff --git a/benchmark/MiddleBury_Other.py b/benchmark/MiddleBury_Other.py
index a6676ce..d0683cf 100644
--- a/benchmark/MiddleBury_Other.py
+++ b/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")
 
diff --git a/benchmark/UCF101.py b/benchmark/UCF101.py
index 1ea20f8..9f2c424 100644
--- a/benchmark/UCF101.py
+++ b/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")
 
diff --git a/benchmark/Vimeo90K.py b/benchmark/Vimeo90K.py
index ccc7d28..977e137 100644
--- a/benchmark/Vimeo90K.py
+++ b/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")
 
diff --git a/inference_video.py b/inference_video.py
index e12b38d..57bc21c 100644
--- a/inference_video.py
+++ b/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")
 
diff --git a/benchmark/pytorch_msssim/__init__.py b/model/pytorch_msssim/__init__.py
similarity index 100%
rename from benchmark/pytorch_msssim/__init__.py
rename to model/pytorch_msssim/__init__.py