gptsubtitler

Субтитры

Автоматически добавляйте субтитры к любому видео, произнесенному на любом языке, на язык по вашему выбору с помощью искусственного интеллекта.

Используемые модели:

• Порт OpenAI шепот C++ — для преобразования аудио в текст
• Facebook M2M10 - для перевода

Используемые инструменты:

• ffmpeg

Пожалуйста, не забудьте отметить репозиторий, если вы найдете его полезным или познавательным!

До:

солдат.mp4

После (по-румынски — model_type=medium, language_model_type=base ):

солдат_субтитров.mp4

Настраивать

Установите с помощью pip .

 pip install gptsubtitler

Установите ffmpeg :

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

# MacOS
brew install ffmpeg

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

Краткое руководство

Пример использования для добавления субтитров и их перевода на румынский язык:

Командная строка:

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

Или на 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" )

Вы также можете использовать класс Translator из translator.py если хотите просто перевести текст.

Пример использования для перевода с английского на румынский:

 from gptsubtitler import Translator

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

Если вы создали файл .srt и просто хотите добавить субтитры:

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

Параметры

 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.

Доступные параметры для captioning_model_type (модель преобразования звука в текст):

• крошечный
• база - по умолчанию
• маленький
• середина
• большой

Доступные параметры для language_model_type (модель языкового переводчика):

• база - по умолчанию
• большой