python-----opencv截取按幀截取視頻

最近有需求把一個視頻從指定幀截取一部分，demo代碼如下：

import cv2
video_path = r'F:\temp\temp_0806\1\video\test.dat'

videoCapture = cv2.VideoCapture(video_path)

fps = 16  # 保存視頻的幀率
size = (1280, 720)  # 保存視頻的大小

videoWriter = cv2.VideoWriter('video4.avi', cv2.VideoWriter_fourcc('X', 'V', 'I', 'D'), fps, size)
i = 0

while True:
    success, frame = videoCapture.read()
    if success:
        i += 1
        print('i = ', i)
        if (i >= 1 and i <= 960):
            videoWriter.write(frame)
    else:
        print('end')
        break