1. 程式人生 > 程式設計 >Python呼叫ffmpeg開源視訊處理庫,批量處理視訊

Python呼叫ffmpeg開源視訊處理庫,批量處理視訊

程式碼示例

# coding=utf-8
import os
import subprocess
import datetime
import json,pprint
import re,time
import threading
import random
import shutil


class FFmpeg:

  def __init__(self,editvdo,addlogo=None,addmusic=None,addvdohead=None,addvdotail=None):
    self.editvdo = editvdo
    self.addlogo = addlogo
    self.addmusic = addmusic
    self.addvdohead = addvdohead
    self.addvdotail = addvdotail
    self.vdo_time,self.vdo_width,self.vdo_height,self.attr_dict = self.get_attr()
    self.editvdo_path = os.path.dirname(editvdo)
    self.editvdo_name = os.path.basename(editvdo)

  def get_attr(self):
    """
    獲取視訊屬性引數
    :return:
    """
    strcmd = r'ffprobe -print_format json -show_streams -i "{}"'.format(self.editvdo)
    status,output = subprocess.getstatusoutput(strcmd)
    agrs = eval(re.search('{.*}',output,re.S).group().replace("\n","").replace(" ",''))
    streams = agrs.get('streams',[])
    agrs_dict = dict()
    [agrs_dict.update(x) for x in streams]
    vdo_time = agrs_dict.get('duration')
    vdo_width = agrs_dict.get('width')
    vdo_height = agrs_dict.get('height')
    attr = (vdo_time,vdo_width,vdo_height,agrs_dict)
    return attr

  def edit_head(self,start_time,end_time,deposit=None):
    """
    擷取指定長度視訊
    :param second: 去除開始的多少秒
    :param deposit: 另存為檔案
    :return: True/Flase
    """
    if None == deposit:
      deposit = self.editvdo_path+'/'+'edit_head'+self.editvdo_name
    start = time.strftime('%H:%M:%S',time.gmtime(start_time))
    end = time.strftime('%H:%M:%S',time.gmtime(end_time))
    strcmd = 'ffmpeg -i "{}" -vcodec copy -acodec copy -ss {} -to {} "{}" -y'.format(
      self.editvdo,start,end,deposit)
    result = subprocess.run(args=strcmd,stdout=subprocess.PIPE,shell=True)
    if os.path.exists(deposit):
      os.remove(self.editvdo)
      os.rename(deposit,self.editvdo)
      return True
    else:
      return False

  def edit_logo(self,deposit=None):
    """
    新增水印
    :param deposit:新增水印後另存為路徑,為空則覆蓋
    :return: True/False
    """
    if None == deposit:
      deposit = self.editvdo_path+'/'+'edit_logo'+self.editvdo_name
    strcmd = r'ffmpeg -i "{}" -vf "movie=\'{}\' [watermark];[in] ' \
         r'[watermark] overlay=main_w-overlay_w-10:10 [out]" "{}"'.format(
          self.editvdo,self.addlogo,self.editvdo)
      return True
    else:
      return False

  def edit_music(self,deposit=None):
    if None == deposit:
      deposit = self.editvdo_path+'/'+'edit_music'+self.editvdo_name
    strcmd = r'ffmpeg -y -i "{}" -i "{}" -filter_complex "[0:a] ' \
         r'pan=stereo|c0=1*c0|c1=1*c1 [a1],[1:a] ' \
         r'pan=stereo|c0=1*c0|c1=1*c1 [a2],[a1][a2]amix=duration=first,' \
         r'pan=stereo|c0<c0+c1|c1<c2+c3,pan=mono|c0=c0+c1[a]" ' \
         r'-map "[a]" -map 0:v -c:v libx264 -c:a aac ' \
         r'-strict -2 -ac 2 "{}"'.format(self.editvdo,self.addmusic,self.editvdo)
      return True
    else:
      return False

  def edit_rate(self,rete=30,deposit=None):
    """
    改變幀率
    :param rete: 修改大小幀率
    :param deposit: 修改後儲存路徑
    :return:
    """
    if None == deposit:
      deposit = self.editvdo_path+'/'+'edit_music'+self.editvdo_name
    strcmd = r'ffmpeg -i "{}" -r {} "{}"' % (self.editvdo,rete,self.editvdo)
      return True
    else:
      return False

  def edit_power(self,power='1280x720',deposit=None):
    """
    修改解析度
    :param power: 解析度
    :param deposit: 修改後儲存路徑,為空則覆蓋
    :return:
    """
    if None == deposit:
      deposit = self.editvdo_path+'/'+'edit_power'+self.editvdo_name
    strcmd = r'ffmpeg -i "{}" -s {} "{}"'.format(self.editvdo,power,self.editvdo)
      return True
    else:
      return False

  def rdit_marge(self,vdo_head,vdo_tail,deposit=None):
    if None == deposit:
      deposit = self.editvdo_path+'/'+'rdit_marge'+self.editvdo_name
    with open(self.editvdo_path+'/'+'rdit_marge.txt','w',encoding='utf-8') as f:
      f.write("file '{}' \nfile '{}' \nfile '{}'" .format(
        vdo_head,self.editvdo,vdo_tail))
    strcmd = r'ffmpeg -f concat -safe 0 -i "{}" -c copy "{}"'.format(
      self.editvdo_path + '/' + 'rdit_marge.txt',self.editvdo)
      return True
    else:
      return False



  # ffmpeg - i input.mkv - filter_complex "[0:v]setpts=0.5*PTS[v];[0:a]atempo=2.0[a]" - map"[v]" - map"[a]" output.mkv



test = FFmpeg(r"D:\vdio\4.mp4")
pass

以上就是Python呼叫ffmpeg開源視訊處理庫,批量處理視訊的詳細內容,更多關於python 批量處理視訊的資料請關注我們其它相關文章!