【初心者歓迎】動画⇔画像変換ツールを作ろう（ffmpegで動画から画像の切り出し＆結合）

       
import os
import sys
sys.path.append(os.path.join(os.path. dirname(__file__), '../libs'))
from create_ui import CreateUI
from  videoimageconverter import VideoImageConverter

# ボタンが押された時の処理
def func(param):
    ff = VideoImageConverter(param["extract_folder"])
    if param["_clicked_widget"] == "extract_button":
        ff.movie_to_image(param["input_file"])
    if param["_clicked_widget"] == "merge_button":
        ff.image_to_movie(param["output_save"])
   
# インスタンス生成
ui = CreateUI(func)

# 画面の表示
ui.show()

import subprocess
import os
import shutil

class VideoImageConverter:
    def __init__(self,temp_folder,format="png",fps=30):
        """
        コンストラクタ

        Args:
            temp_folder (str): 出力フォルダ
            format (str): 画像の種類(png,jpg,bmp など)
            fps (int): フレームレート
        """      
        self.fps = fps
        self.format = format
        self.image_folder = f"{temp_folder}/image"
        self.audio_folder = f"{temp_folder}/audio"  
    
    def movie_to_image(self,input_video,mode="av"):
        """
        動画を画像と音声に分解する。
        output_folder配下に temp/image と temp/audio フォルダが作成される

        Args:
            input_video (str): 入力動画ファイル
            mode (str): "a":動画から音声を抽出 "v":動画から画像を抽出
        """      
            
        if "v" in mode:                  
            # 画像の格納先フォルダ作成     
            self.drop_and_create_folder(self.image_folder)
            # 画像の抽出
            command = f'ffmpeg -i "{input_video}" -q:v 1 -r {self.fps} "{self.image_folder}/%10d.{self.format}"'
            subprocess.call(command, shell=True)
            
        if "a" in mode:
            # 音声の格納先フォルダ作成 
            self.drop_and_create_folder(self.audio_folder)
            # 音声の抽出
            command = f'ffmpeg -i "{input_video}" -c:a aac "{self.audio_folder}/audio.aac"'
            subprocess.call(command, shell=True)

    def image_to_movie(self,output_video):
        """
        画像ファイルを動画に結合する

        Args:
            output_video (str): 出力動画ファイル名
        """

        # 出力先のファイルがあれば削除
        if os.path.isfile(output_video):
            os.remove(output_video)
        
        # 音声ファイルがあれば、動画に加える
        audio_path = f"{self.audio_folder}/audio.aac"
        option = ""
        if os.path.isfile(audio_path):
            option = f' -i "{audio_path}" '
            
        # ffmpegコマンドを作成
        command = f'ffmpeg -r {self.fps} -i "{self.image_folder}/%10d.{self.format}" {option} -c:v libx264 -pix_fmt yuv420p "{output_video}"'

        # ffmpegを実行
        subprocess.call(command, shell=True)    

    def drop_and_create_folder(self,folder):
        """
        フォルダを作成する。既に存在すれば削除して再作成する

        Args:
            folder (str): フォルダ名
        """
        # 既にフォルダがあれば削除
        if os.path.exists(folder):
            shutil.rmtree(folder)         
        # フォルダを作成
        os.makedirs(folder, exist_ok=True) 

# 作業フォルダを指定してインスタンスを生成
ff = VideoImageConverter("./temp")

# 動画から画像と音声を抽出して作業フォルダに保存
ff.movie_to_image("d:/hoge.mp4")

# 作業フォルダ内の画像と音声から動画を作成
ff.image_to_movie("d:/join.mp4")

title:動画⇔画像変換ツール
mode:light
max:1024,350
size:600,350
min:600,350
input_file:入力ファイル,ファイル選択
extract_button:画像と音声を抽出
extract_folder:作業フォルダ,フォルダ選択
merge_button:画像と音声から動画を作成
output_save:出力ファイル,ファイル選択