python-opencv写入相机拍摄的视频

最新推荐文章于 2024-05-11 15:26:08 发布

吴天德少侠

最新推荐文章于 2024-05-11 15:26:08 发布

阅读量141

点赞数

文章标签： python opencv 音视频

本文链接：https://blog.csdn.net/sdhdsf132452/article/details/129275123

版权

opencv 专栏收录该内容

34 篇文章 3 订阅

订阅专栏

该代码段定义了一个函数saveVideo，它使用OpenCV库从指定设备_id_捕获视频，并将其保存到output目录下。文件名基于当前时间戳，格式为MP4。在循环中，它读取视频帧，显示帧并将其写入输出视频文件，直到用户按q键退出。如果摄像头无法打开，程序会打印错误信息。

摘要由CSDN通过智能技术生成

代码

import os
import cv2
from datetime import datetime


def saveVideo(device_id_,out_dir_='output/'):
    cap = cv2.VideoCapture(device_id_)
    frame_w = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
    frame_h = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
    size = (int(frame_w),int(frame_h))
    fps = cap.get(cv2.CAP_PROP_FPS)
    
    now = datetime.now()
    suffix = f'{now.year:04}{now.month:02}{now.day:02}_{now.hour:02}{now.minute:02}{now.second:02}'
    
    video_name = suffix+".mp4"
    if not os.path.exists(out_dir_):
        os.mkdir(out_dir_)
    out_path = os.path.join(out_dir_,video_name)

    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    video_writer = cv2.VideoWriter(out_path,fourcc,fps,size)

    while(cap.isOpened()):
        ret, frame = cap.read()
        if ret == True:
            # 说明读取视频帧成功
            cv2.imshow('frame',frame)
            video_writer.write(frame)

            if cv2.waitKey(int(1000/fps)) & 0xFF == ord('q'):
                break
        else:
            print('相机打开失败')
            break




if __name__ == '__main__':
    saveVideo(device_id_=0,)
    pass