Python Opencv —— 获取指定时间内的视频片段以及帧

参考：OpenCV截取指定时间范围内的视频（python）_Qyun_lucky_star的博客-CSDN博客

import cv2


def capture_video(video_path, result_video_path, video, result_video, start_time, end_time):
    """
    功能：截取短视频
    参数：
        video_path：需要截取的视频路径
        result_video_path：截取后的视频存放的路径
        video：需要截取的视频的名称（不带后缀）
        result_video：截取了的视频的名称（不带后缀）
        start_time：截取开始时间（单位s）
        end_time：截取结束时间（单位s）
    """

    # 读取视频
    cap = cv2.VideoCapture(video_path + video)

    # 读取视频帧率
    fps_video = cap.get(cv2.CAP_PROP_FPS)

    # 设置写入视频的编码格式
    fourcc = cv2.VideoWriter_fourcc(*"mp4v")

    # 获取视频宽度和高度
    frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

    # 设置写视频的对象
    videoWriter = cv2.VideoWriter(result_video_path + result_video, fourcc, fps_video, (frame_width, frame_height))

    # 初始化一个计数器
    count = 0

    while (cap.isOpened()):

        # 按帧读取视频里的图片
        ret, frame = cap.read()

        # 如果视频没有读取结束
        if ret == True:

            # 计数器加一
            count += 1

            # 截取相应时间内的视频信息
            if(count > (start_time * fps_video) and count <= (end_time * fps_video)):
                # 将图片写入视屏
                videoWriter.write(frame)

            if(count == (end_time * fps_video)):
                break

        else:
            # 写入视屏结束
            videoWriter.release()
            break

if __name__ == '__main__':

    video_path = "./"
    result_video_path = "result/"
    video = "中长跑视频回放2.mp4"
    result_video = "result_test_video.mp4"
    start_time = 7 * 60 + 0
    end_time = 7 * 60 + 6
    capture_video(video_path, result_video_path, video, result_video, start_time, end_time)
    # 切成帧
    """
    import cv2
    video = "./result/result_test_video.mp4"
    cap = cv2.VideoCapture(video)
    frame_id = 0
    while (cap.isOpened()):
        ret, frame = cap.read()
        if ret==True:
            cv2.imwrite('./2/'+str(frame_id)+'.jpg',frame)
            frame_id+=1
        else:
            cap.release()
    """