gptsubtitler

Sous-titreur

Sous-titrez automatiquement n'importe quelle vidéo parlée dans n'importe quelle langue dans la langue de votre choix à l'aide de l'IA.

Modèles utilisés :

• Port C++ OpenAI Whisper - pour l'audio en texte
• Facebook M2M10 - pour la traduction

Outils utilisés :

• ffmpeg

N'oubliez pas de mettre en vedette le référentiel si vous le trouvez utile ou éducatif !

Avant:

soldat.mp4

Après (en roumain - model_type=medium, language_model_type=base ) :

soldat_subtitled.mp4

Installation

Installez en utilisant pip .

 pip install gptsubtitler

Installez ffmpeg :

 # Ubuntu or Debian
sudo apt update && sudo apt install ffmpeg

# MacOS
brew install ffmpeg

# Windows using Chocolatey https://chocolatey.org/
choco install ffmpeg

Guide rapide

Exemple d'utilisation pour ajouter des sous-titres et les traduire en roumain :

Ligne de commande :

gptsubtitler soldier.mp4 --source_language en --target_language ro --captioning_model_type medium --language_model_type base

Ou en Python

 from gptsubtitler import Transcriber

# I strongly recommend using the "medium" model_type.
Transcriber . transcribe ( "soldier.mp4" , source_language = "en" , target_language = "ro" , captioning_model_type = "medium" , language_model_type = "base" )

Vous pouvez également utiliser la classe Translator de translator.py si vous souhaitez simplement traduire du texte.

Exemple d'utilisation pour traduire de l'anglais vers le roumain :

 from gptsubtitler import Translator

print ( Translator . translate ( "Hi!" , target_language = "ro" , source_language = "en" ))

Si vous avez généré un fichier .srt et souhaitez simplement ajouter des sous-titres :

 from gptsubtitler import create_video_with_subtitles
create_video_with_subtitles ( "video.mp4" , "output.srt" , "video_subtitled.mp4" )

Possibilités

 Args:
    video_file (str): Path to video file.

    output_video_file (str, optional): Path to output video file. Defaults to video_file_subtitled.

    output_subtitle_file (str, optional): Path to output SRT file. Defaults to "output.srt".

    source_language (str, optional): Source language for translation. Defaults to en.

    target_language (str, optional): Target language for translation. Defaults to None.

    captioning_model_type (str, optional): Model type. Defaults to "base".

    language_model_type (str, optional): Language model type. Defaults to "base".

    model_dir (str, optional): Path to model directory. Defaults to None.

Options disponibles pour captioning_model_type (le modèle audio en texte) :

• minuscule
• base - par défaut
• petit
• moyen
• grand

Options disponibles pour language_model_type (le modèle de traducteur de langue) :

• base - par défaut
• grand