From 12c99227acad12b69b113b31a1f08789b606d08a Mon Sep 17 00:00:00 2001 From: syzygy137 Date: Sun, 10 May 2026 21:32:24 -0700 Subject: [PATCH] Replace skvideo with cv2 for video reading --- inference_video.py | 12 ++++++++++-- requirements.txt | 1 - 2 files changed, 10 insertions(+), 3 deletions(-) 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