1. 程式人生 > >FFmpeg+python實現批量視訊分幀

FFmpeg+python實現批量視訊分幀

使用FFmpeg工具對短視訊批量分幀,以下是一段很簡單的程式碼:

import PIL.Image as Image
import pylab
import imageio
#註釋的程式碼執行一次就好,以後都會預設下載完成
#imageio.plugins.ffmpeg.download()  #第一次執行是刪除註釋,下載ffmpeg工具
import skimage
import numpy as np
import os
from subprocess import call
from tqdm import tqdm
import pandas as pd
#視訊的絕對路徑和圖片儲存的目標路徑
def extract_frames(src_path,target_path):

    new_path = target_path

    for video_name in tqdm(os.listdir(src_path)):
        #video_name = "ZJL35.mp4"
        filename = src_path + video_name
        cur_new_path = new_path+video_name.split('.')[0]+'/'
        if not os.path.exists(cur_new_path):
            os.mkdir(cur_new_path)
        dest = cur_new_path + video_name.split('.')[0]+'-%04d.jpg'
        call(["ffmpeg", "-i", filename,"-r","5", dest]) #這裡的5為5fps,幀率可修改

執行

extract_frames(src_path='./Charades_test_video/',target_path='./Charades_test_video_pics/')