欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python实现多个视频合成一个视频的功能

程序员文章站 2022-03-08 20:21:52
目录前言环境依赖代码验证一下前言本文提供将多个视频拼接为一个视频的python工具代码,其中有一些限制条件,下面的代码说明会提到。环境依赖ffmpeg环境安装,可以参考:本文主要使用到的不是ffmpe...

前言

本文提供将多个视频拼接为一个视频的python工具代码,其中有一些限制条件,下面的代码说明会提到。

环境依赖

ffmpeg环境安装,可以参考:

本文主要使用到的不是ffmpeg,而是ffprobe也在上面这篇文章中的zip包中。

Python实现多个视频合成一个视频的功能

ffmpy安装:

pip install ffmpy -i https://pypi.douban.com/simple

代码

不废话了,上代码。

#!/user/bin/env python
# coding=utf-8
"""
@project : csdn
@author  : 剑客阿良_aliang
@file   : concat_video.py
@ide    : pycharm
@time   : 2021-12-23 15:23:16
"""
 
from ffmpy import ffmpeg
import os
import uuid
import subprocess
 
 
# 视频拼接
def concat(video_list: list, output_dir: str):
    if len(video_list) == 0:
        raise exception('video_list can not empty')
    _ext = check_format(video_list)
    _fps = check_fps(video_list)
    _result_path = os.path.join(
        output_dir, '{}{}'.format(
            uuid.uuid1().hex, _ext))
    _tmp_config = make_tmp_concat_config(video_list, output_dir)
    ff = ffmpeg(inputs={'{}'.format(_tmp_config): '-f concat -safe 0 -y'}, outputs={
        _result_path: '-c copy'})
    print(ff.cmd)
    ff.run()
    os.remove(_tmp_config)
    return _result_path
 
 
# 构造拼接所需临时文件
def make_tmp_concat_config(video_list: list, output_dir: str):
    _tmp_concat_config_path = os.path.join(output_dir, '{}.txt'.format(uuid.uuid1().hex))
    with open(_tmp_concat_config_path, mode='w', encoding='utf-8') as f:
        f.writelines(list(map(lambda x: 'file {}\n'.format(x), video_list)))
    return _tmp_concat_config_path
 
 
# 校验每个视频的格式
def check_format(video_list: list):
    _video_format = ''
    for x in video_list:
        _ext = os.path.splitext(x)[-1]
        if _video_format == '' and _ext != '':
            _video_format = _ext
            continue
        if _video_format != '' and _ext == _video_format:
            continue
        if _video_format != '' and _ext != _video_format:
            raise exception('inconsistent video format')
    return _video_format
 
 
# 校验每个视频的fps
def check_fps(video_list: list):
    _video_fps = 0
    for x in video_list:
        _fps = get_video_fps(x)
        if _video_fps == 0 and _fps:
            _video_fps = _fps
            continue
        if _video_fps != 0 and _fps == _video_fps:
            continue
        if _video_fps != '' and _fps != _video_fps:
            raise exception('inconsistent video fps')
    if _video_fps == 0:
        raise exception('video fps error')
    return _video_fps
 
 
# 获取视频fps
def get_video_fps(video_path: str):
    ext = os.path.splitext(video_path)[-1]
    if ext != '.mp4' and ext != '.avi' and ext != '.flv':
        raise exception('format not support')
    ffprobe_cmd = 'ffprobe -v error -select_streams v -of default=noprint_wrappers=1:nokey=1 -show_entries stream=r_frame_rate {}'
    p = subprocess.popen(
        ffprobe_cmd.format(video_path),
        stdout=subprocess.pipe,
        stderr=subprocess.pipe,
        shell=true)
    out, err = p.communicate()
    print("subprocess 执行结果:out:{} err:{}".format(out, err))
    fps_info = str(out, 'utf-8').strip()
    if fps_info:
        if fps_info.find("/") > 0:
            video_fps_str = fps_info.split('/', 1)
            fps_result = int(int(video_fps_str[0]) / int(video_fps_str[1]))
        else:
            fps_result = int(fps_info)
    else:
        raise exception('get fps error')
    return fps_result
 
 
if __name__ == '__main__':
    print(concat(['d:/tmp/100.mp4', 'd:/tmp/101.mp4'], 'c:/users/huyi/desktop'))

代码说明

1、主要拼接方法为concat,入参分别为:视频列表、输出目录。

2、该视频拼接命令对视频本身有所限制,需要保证都是相同格式的视频,其次是每个视频的fps得一致,不然最终合成的视频会无法打开或者出现花屏现象。

3、临时的拼接文件会在使用后删除。

4、最终输出的文件名为了保证唯一使用uuid。

验证一下

下面是我准备的两个视频:

Python实现多个视频合成一个视频的功能Python实现多个视频合成一个视频的功能

执行结果

pydev console: starting.
python 3.6.13 |anaconda, inc.| (default, mar 16 2021, 11:37:27) [msc v.1916 64 bit (amd64)] on win32
runfile('d:/spyder/csdn/tool/concat_video.py', wdir='d:/spyder/csdn/tool')
subprocess 执行结果:out:b'25/1\r\n' err:b''
subprocess 执行结果:out:b'25/1\r\n' err:b''
ffmpeg -f concat -safe 0 -y -i c:/users/huyi/desktop\6d02df4b63d111eca6eee454e8bf1461.txt -c copy c:/users/huyi/desktop\6d02df4a63d111ec9dbbe454e8bf1461.mp4
ffmpeg version n4.3.1-20-g8a2acdc6da copyright (c) 2000-2020 the ffmpeg developers
  built with gcc 9.3-win32 (gcc) 20200320
  configuration: --prefix=/ffbuild/prefix --pkg-config-flags=--static --pkg-config=pkg-config --cross-prefix=x86_64-w64-mingw32- --arch=x86_64 --target-os=mingw32 --enable-gpl --enable-version3 --disable-debug --enable-iconv --enable-zlib --enable-libxml2 --enable-libfreetype --enable-libfribidi --enable-gmp --enable-lzma --enable-fontconfig --enable-libvmaf --disable-vulkan --enable-libvorbis --enable-amf --enable-libaom --enable-avisynth --enable-libdav1d --enable-ffnvcodec --enable-cuda-llvm --disable-libglslang --enable-libass --enable-libbluray --enable-libmp3lame --enable-libopus --enable-libtheora --enable-libvpx --enable-libwebp --enable-libmfx --enable-libopencore-amrnb --enable-libopencore-amrwb --enable-libopenjpeg --enable-librav1e --enable-schannel --enable-sdl2 --enable-libsoxr --enable-libsrt --enable-libtwolame --enable-libvidstab --enable-libx264 --enable-libx265 --enable-libxvid --enable-libzimg --extra-cflags=-dlibtwolame_static --extra-cxxflags= --extra-ldflags=-pthread --extra-libs=-lgomp
  libavutil      56. 51.100 / 56. 51.100
  libavcodec     58. 91.100 / 58. 91.100
  libavformat    58. 45.100 / 58. 45.100
  libavdevice    58. 10.100 / 58. 10.100
  libavfilter     7. 85.100 /  7. 85.100
  libswscale      5.  7.100 /  5.  7.100
  libswresample   3.  7.100 /  3.  7.100
  libpostproc    55.  7.100 / 55.  7.100
[mov,mp4,m4a,3gp,3g2,mj2 @ 000001de4a7aebc0] auto-inserting h264_mp4toannexb bitstream filter
input #0, concat, from 'c:/users/huyi/desktop\6d02df4b63d111eca6eee454e8bf1461.txt':
  duration: n/a, start: -0.064000, bitrate: 1098 kb/s
    stream #0:0(und): audio: aac (lc) (mp4a / 0x6134706d), 16000 hz, mono, fltp, 69 kb/s
    metadata:
      handler_name    : soundhandler
    stream #0:1(und): video: h264 (main) (avc1 / 0x31637661), yuv420p, 1080x1920, 1028 kb/s, 25 fps, 25 tbr, 12800 tbn, 50 tbc
    metadata:
      handler_name    : videohandler
output #0, mp4, to 'c:/users/huyi/desktop\6d02df4a63d111ec9dbbe454e8bf1461.mp4':
  metadata:
    encoder         : lavf58.45.100
    stream #0:0(und): video: h264 (main) (avc1 / 0x31637661), yuv420p, 1080x1920, q=2-31, 1028 kb/s, 25 fps, 25 tbr, 12800 tbn, 12800 tbc
    metadata:
      handler_name    : videohandler
    stream #0:1(und): audio: aac (lc) (mp4a / 0x6134706d), 16000 hz, mono, fltp, 69 kb/s
    metadata:
      handler_name    : soundhandler
stream mapping:
  stream #0:1 -> #0:0 (copy)
  stream #0:0 -> #0:1 (copy)
press [q] to stop, [?] for help
[mov,mp4,m4a,3gp,3g2,mj2 @ 000001de4a7aebc0] auto-inserting h264_mp4toannexb bitstream filter
[mp4 @ 000001de4acdabc0] non-monotonous dts in output stream 0:1; previous: 6176768, current: 6176608; changing to 6176769. this may result in incorrect timestamps in the output file.
frame=22199 fps=0.0 q=-1.0 lsize=  119649kb time=00:14:48.05 bitrate=1103.7kbits/s speed=2.41e+03x    
video:111571kb audio:7524kb subtitle:0kb other streams:0kb global headers:0kb muxing overhead: 0.465183%
c:/users/huyi/desktop\6d02df4a63d111ec9dbbe454e8bf1461.mp4

结果验证

Python实现多个视频合成一个视频的功能

到此这篇关于python实现多个视频合成一个视频的功能的文章就介绍到这了,更多相关python视频合成内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!