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

Move ssim script

Browse Source
This commit is contained in:
hzwer
2021-06-17 11:31:20 +08:00
parent 011d301004
commit f6d90d988d
8 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/ATD12K.py
View File

@@ -7,7 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.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")

2
benchmark/HD.py
View File

@@ -7,7 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.pytorch_msssim import ssim_matlab
from model.RIFE import Model from model.RIFE import Model
from skimage.color import rgb2yuv, yuv2rgb from skimage.color import rgb2yuv, yuv2rgb
from yuv_frame_io import YUV_Read,YUV_Write from yuv_frame_io import YUV_Read,YUV_Write

2
benchmark/HD_multi.py
View File

@@ -7,7 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.pytorch_msssim import ssim_matlab
from model.RIFE import Model from model.RIFE import Model
from skimage.color import rgb2yuv, yuv2rgb from skimage.color import rgb2yuv, yuv2rgb
from yuv_frame_io import YUV_Read,YUV_Write from yuv_frame_io import YUV_Read,YUV_Write

2
benchmark/MiddleBury_Other.py
View File

@@ -7,7 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.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")

2
benchmark/UCF101.py
View File

@@ -7,7 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.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")

3
benchmark/Vimeo90K.py
View File

@@ -7,8 +7,7 @@ import torch
import argparse import argparse
import numpy as np 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 model.pytorch_msssim import ssim_matlab
# from model.RIFE2F15C import Model
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")

2
inference_video.py
View File

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

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