diff --git a/inference_video.py b/inference_video.py index 1069fc1..94d7bb3 100644 --- a/inference_video.py +++ b/inference_video.py @@ -7,12 +7,20 @@ from torch.nn import functional as F import warnings import _thread -import skvideo.io from queue import Queue, Empty from model.pytorch_msssim import ssim_matlab warnings.filterwarnings("ignore") +def vreader(filename): + cap = cv2.VideoCapture(filename) + while True: + ret, frame = cap.read() + if not ret: + break + yield frame[:, :, ::-1].copy() # BGR to RGB + cap.release() + def transferAudio(sourceVideo, targetVideo): import shutil import moviepy.editor @@ -108,7 +116,7 @@ def transferAudio(sourceVideo, targetVideo): args.fps = fps * args.multi else: fpsNotAssigned = False - videogen = skvideo.io.vreader(args.video) + videogen = vreader(args.video) lastframe = next(videogen) fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v') video_path_wo_ext, ext = os.path.splitext(args.video) diff --git a/requirements.txt b/requirements.txt index f405577..c1b497d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,5 @@ numpy>=1.16, <=1.23.5 tqdm>=4.35.0 -sk-video>=1.1.10 torch>=1.3.0 opencv-python>=4.1.2 moviepy>=1.0.3