opencv视频裁剪

import cv2
import os
import numpy as np
from tqdm import tqdm


def process(source_dir, target_dir, video_name):
    source_video = os.path.join(source_dir, video_name)
    target_video = os.path.join(target_dir, video_name)
    print(f"Processing video: {video_name}")
    if os.path.exists(target_video):
        print(f"{video_name} has been processed")
    else:
        capture = cv2.VideoCapture(source_video)
        fps = capture.get(cv2.CAP_PROP_FPS)
        total_frame = capture.get(cv2.CAP_PROP_FRAME_COUNT)
        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
        # 宽，高
        cols, rows = int(capture.get(cv2.CAP_PROP_FRAME_WIDTH)), int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT))  # 宽高
        p1_rows = [97, 950]
        p2_rows = [1030, rows]
        row_sum = p1_rows[1] - p1_rows[0] + p2_rows[1] - p2_rows[0]

        out = cv2.VideoWriter(target_video, fourcc, fps, (cols, row_sum), True)

        for i in tqdm(range(int(total_frame))):
            ret, frame = capture.read()
            if not ret:
                break

            processed_frame = np.zeros((row_sum, cols, 3), np.uint8)
            processed_frame[:p1_rows[1]-p1_rows[0], :] = frame[p1_rows[0]:p1_rows[1],:]
            processed_frame[p1_rows[1]-p1_rows[0]:, :] = frame[p2_rows[0]:p2_rows[1],:]
            out.write(processed_frame)
            processed_frame_small = cv2.resize(processed_frame, (int(processed_frame.shape[1]/4),int(processed_frame.shape[0]/4)))
            cv2.imshow(target_video, processed_frame_small)
            if cv2.waitKey(1) & 0xFF == ord('q'):  # q退出
                break
        
        capture.release()
        out.release()
        cv2.destroyAllWindows()

def main(source_dir, target_dir):
    video_list = sorted(os.listdir(source_dir))
    if not os.path.exists(target_dir):
        os.mkdir(target_dir)
    for video_name in video_list:
        process(source_dir, target_dir, video_name)
    # process(source_dir, target_dir, video_list[0])
    cv2.destroyAllWindows()

if __name__=="__main__":
    source_dir = f"/mnt/dataDisk/0608_videos_to_cut"
    target_dir = f"/mnt/dataDisk/0608_videos_cutted"
    main(source_dir, target_dir)

补充：
opencv生成的MP4中的视频格式是H265，web上便于显示的MP4需要H264格式，可以使用 ffmpeg -i <input_video> -vcodec h264 <output_video> 处理得到H264格式视频的MP4文件，并且生成的视频保持画质的同时大小更小一些。