Python3使用ffmpeg、opencv操作音视频文件记录

Python3使用ffmpeg、opencv操作音视频文件记录

opencv

• pip install opencv-python
  

获取首帧

import cv2

def keyframe_test():
    vidcap = cv2.VideoCapture(r F:	oolsffmpegin	ime.mp4 )
    success, image = vidcap.read()
    if success:
        cv2.imwrite("frame1.jpg", image)  # save frame as JPEG file
        cv2.imshow( frame , image)
        cv2.waitKey()

格式信息

def get_source_info_opencv(source_name):
    return_value = 0
    try:
        cap = cv2.VideoCapture(source_name)
        width = cap.get(cv2.CAP_PROP_FRAME_WIDTH )
        height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
        fps = cap.get(cv2.CAP_PROP_FPS)
        num_frames = cap.get(cv2.CAP_PROP_FRAME_COUNT)
        print("width:{} 
height:{} 
fps:{} 
num_frames:{}".format(width, height, fps, num_frames))
    except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
        print("init_source:{} error. {}
".format(source_name, str(e)))
        return_value = -1
    return return_value

参数项

capture.get(0)   视频文件的当前位置（播放）以毫秒为单位
capture.get(1)   基于以0开始的被捕获或解码的帧索引
capture.get(2)   视频文件的相对位置（播放）：0=电影开始，1=影片的结尾。
capture.get(3)   在视频流的帧的宽度
capture.get(4)   在视频流的帧的高度
capture.get(5)   帧速率
capture.get(6)   编解码的4字-字符代码
capture.get(7)   视频文件中的帧数
capture.get(8)   返回对象的格式
capture.get(9)   返回后端特定的值，该值指示当前捕获模式
capture.get(10)   图像的亮度(仅适用于照相机)
capture.get(11)   图像的对比度(仅适用于照相机)
capture.get(12)   图像的饱和度(仅适用于照相机)
capture.get(13)   色调图像(仅适用于照相机)
capture.get(14)   图像增益(仅适用于照相机)（Gain在摄影中表明白平衡提升）
capture.get(15)   曝光(仅适用于照相机)
capture.get(16)   指示是否应将图像转换为RGB布尔标志
capture.get(17)   × 暂时不支持
capture.get(18)   立体摄像机的矫正标注（目前只有DC1394 v.2.x后端支持这个功能）

ffmpeg

主机需安装ffmpeg

方式一

pip install ffmpeg-python; ffmpeg

Ubuntu机器上测试的时候获取不到数据

import ffmpeg

def get_source_info_ffmpeg(source_name):
    return_value = 0
    try:
        info = ffmpeg.probe(source_name, capture_stdout=True, capture_stderr=True)
        # print(info)
        # print("---------------------------------")
        vs = next(c for c in info[ streams ] if c[ codec_type ] ==  video )
        format_name = info[ format ][ format_name ]
        codec_name = vs[ codec_name ]
        duration_ts = float(vs[ duration_ts ])
        fps = vs[ r_frame_rate ]
        width = vs[ width ]
        height = vs[ height ]
        print("format_name:{} 
codec_name:{} 
duration_ts:{} 
width:{} 
height:{} 
fps:{}".format(format_name, codec_name, duration_ts, width, height, fps))
    except (OSError, TypeError, ValueError, KeyError, SyntaxError, Exception) as e:
        print("init_source:{} error. {}
".format(source_name, str(e)))
        return_value = 0
    return return_value

方式二 使用进程调用

此方式成功获取到数据

import subprocess

def use_system_ffmpeg(source_name):

    ffmpeger = subprocess.Popen(
        f ffprobe -show_streams {source_name} ,
        shell=False, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr = subprocess.PIPE, encoding= utf-8 )
    relog = ffmpeger.stdout.read()
    print( ====== ,relog)
    info_list = relog.split( 
 )
    frame_nums = 0
    frame_rate = 0
    for item in info_list:
        if item.startswith( r_frame_rate ):
            frame_rate = int(eval(item.split( = )[1]))
        if item.startswith( nb_frames ):
            frame_nums = int(item.split( = )[1])
            break
    print(frame_nums, frame_rate)
    return frame_nums,frame_rate