支持设置VSF核心数和解码器

Author: eritpchy

backend/config.py

@@ -3,7 +3,7 @@
 from pathlib import Path
 from qfluentwidgets import (qconfig, ConfigItem, QConfig, OptionsValidator, BoolValidator, OptionsConfigItem, 
                             EnumSerializer, RangeValidator, RangeConfigItem)
-from backend.tools.constant import SubtitleArea
+from backend.tools.constant import SubtitleArea, VideoSubFinderDecoder
 import configparser
 
 # 项目版本号
@@ -90,6 +90,10 @@ class Config(QConfig):
     hardwareAcceleration = ConfigItem("Main", "HardwareAcceleration", HARDWARD_ACCELERATION_OPTION, BoolValidator())
     # 启动时检查应用更新
     checkUpdateOnStartup = ConfigItem("Main", "CheckUpdateOnStartup", True, BoolValidator())
+    # VideoSubFinder CPU核心数
+    videoSubFinderCpuCores = RangeConfigItem("Main", "VideoSubFinderCpuCores", 0, RangeValidator(0, os.cpu_count()))
+    # VideoSubFinder 视频解码组件
+    videoSubFinderDecoder = OptionsConfigItem("Main", "VideoSubFinderDecoder", VideoSubFinderDecoder.OPENCV, OptionsValidator(VideoSubFinderDecoder), EnumSerializer(VideoSubFinderDecoder))
 
 CONFIG_FILE = 'config/config.json'
 config = Config()

backend/interface/ch.ini

@@ -10,6 +10,7 @@ ProjectLinkDesc = 视频硬字幕提取，生成srt文件。无需申请第三
 BasicSetting = 基础设置
 AdvancedSetting = 高级设置
 DevSetting = 开发配置
+VideoSubFinderSetting = VideoSubFinder配置
 AboutSetting = 关于
 GenerateTxt = 生成TXT文本字幕
 WordSegmentation = 重新分词
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = 有可用更新
 UpdatesAvailableDesc = 发现新版本 {}, 是否更新？
 NoUpdatesAvailableTitle = 无可用更新
 NoUpdatesAvailableDesc = 软件已是最新版本
+VideoSubFinderCpuCores = CPU核心数
+VideoSubFinderCpuCoresDesc = 字幕提取时使用的CPU核心数，默认为0，即自动, 选择所有核心并不会变得快
+VideoSubFinderDecoder = 视频解码组件
+VideoSubFinderDecoderDesc = 用于针对不同视频兼容性需求, 默认为OpenCV(推荐), 当遇到无法正常提取时可以切换为FFmpeg(可能会造成时间轴细微偏移)
 
 [SubtitleArea]
 LowerPart = 下半部分

backend/interface/chinese_cht.ini

@@ -9,6 +9,7 @@ ProjectLinkTitle = 字幕提取器
 ProjectLinkDesc = 視頻硬字幕提取，生成srt文件。無需申請第三方API，本地實現文本識別。基於深度學習的視頻字幕提取框架，包含字幕區域檢測、字幕內容提取。
 BasicSetting = 基本設置
 AdvancedSetting = 高級設置
+VideoSubFinderSetting = VideoSubFinder設定
 DevSetting = 開發配置
 AboutSetting = 關於
 GenerateTxt = 生成TXT文本字幕
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = 有可用更新
 UpdatesAvailableDesc = 發現新版本 {}，是否更新？
 NoUpdatesAvailableTitle = 無可用更新
 NoUpdatesAvailableDesc = 軟體已是最新版本
+VideoSubFinderCpuCores = CPU核心數
+VideoSubFinderCpuCoresDesc = 提取字幕時使用的CPU核心數，預設為0（自動），選擇所有核心不一定更快
+VideoSubFinderDecoder = 視頻解碼組件
+VideoSubFinderDecoderDesc = 用於不同影片相容性需求，預設為OpenCV（推薦），若無法正常提取可切換為FFmpeg（可能導致時間軸略有偏移）
 
 [SubtitleArea]
 LowerPart = 下半部分

backend/interface/en.ini

@@ -9,6 +9,7 @@ ProjectLinkTitle = Subtitle Extractor
 ProjectLinkDesc = Video hard subtitle extraction, generating srt files. No third-party API required, local text recognition. A video subtitle extraction framework based on deep learning, including subtitle area detection and subtitle content extraction.
 BasicSetting = Basic Settings
 AdvancedSetting = Advanced Settings
+VideoSubFinderSetting = VideoSubFinder Settings
 DevSetting = Development Settings
 AboutSetting = About
 GenerateTxt = Generate TXT Subtitles
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = Updates Available
 UpdatesAvailableDesc = New version {} found, update now?
 NoUpdatesAvailableTitle = No Updates Available
 NoUpdatesAvailableDesc = The software is up to date
+VideoSubFinderCpuCores = CPU Cores
+VideoSubFinderCpuCoresDesc = Number of CPU cores used for subtitle extraction. Default is 0 (auto). Using all cores may not be faster.
+VideoSubFinderDecoder = Video Decoder
+VideoSubFinderDecoderDesc = For different video compatibility needs. Default is OpenCV (recommended). Switch to FFmpeg if extraction fails (may cause slight timeline shift).
 
 [SubtitleArea]
 LowerPart = Lower Part

backend/interface/es.ini

@@ -8,7 +8,8 @@ CopyrightDesc = ©Derechos reservados 2020, YaoFANGUK, Jason Eric (UI), Versión
 ProjectLinkTitle = Extractor de subtítulos
 ProjectLinkDesc = Extracción de subtítulos incrustados en video, generando archivos srt. No requiere API de terceros, reconocimiento de texto local. Un marco de extracción de subtítulos de video basado en aprendizaje profundo, que incluye detección de área de subtítulos y extracción de contenido de subtítulos.
 BasicSetting = Configuración básica
-AdvancedSetting = Configuración avanzada
+AdvancedSetting = Configuración de avanzada
+VideoSubFinderSetting = Configuración de VideoSubFinder
 DevSetting = Configuración de desarrollo
 AboutSetting = Acerca de
 GenerateTxt = Generar subtítulos TXT
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = Actualizaciones disponibles
 UpdatesAvailableDesc = Se encontró una nueva versión {}, ¿actualizar ahora?
 NoUpdatesAvailableTitle = No hay actualizaciones disponibles
 NoUpdatesAvailableDesc = El software está actualizado
+VideoSubFinderCpuCores = Núcleos de CPU
+VideoSubFinderCpuCoresDesc = Núcleos de CPU usados para extraer subtítulos. Por defecto es 0 (automático). Usar todos los núcleos no siempre es más rápido.
+VideoSubFinderDecoder = Componente de decodificación de video
+VideoSubFinderDecoderDesc = Para diferentes necesidades de compatibilidad de video. Por defecto es OpenCV (recomendado). Cambia a FFmpeg si la extracción falla (puede causar un pequeño desfase en la línea de tiempo).
 
 [SubtitleArea]
 LowerPart = Parte inferior

backend/interface/japan.ini

@@ -9,6 +9,7 @@ ProjectLinkTitle = 字幕抽出ツール
 ProjectLinkDesc = 動画の硬字幕を抽出し、srtファイルを生成します。サードパーティAPIは不要で、ローカルでテキスト認識を実現。深層学習に基づく動画字幕抽出フレームワークで、字幕領域の検出と字幕内容の抽出を含みます。
 BasicSetting = 基本設定
 AdvancedSetting = 詳細設定
+VideoSubFinderSetting = VideoSubFinder設定
 DevSetting = 開発設定
 AboutSetting = について
 GenerateTxt = TXT字幕を生成
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = 利用可能なアップデート
 UpdatesAvailableDesc = 新しいバージョン{}が見つかりました。今すぐアップデートしますか？
 NoUpdatesAvailableTitle = 利用可能なアップデートはありません
 NoUpdatesAvailableDesc = ソフトウェアは最新バージョンです
+VideoSubFinderCpuCores = CPUコア数
+VideoSubFinderCpuCoresDesc = 字幕抽出に使用するCPUコア数。デフォルトは0（自動）。全コアを選択しても速くなるとは限りません。
+VideoSubFinderDecoder = ビデオデコーダー
+VideoSubFinderDecoderDesc = 異なる動画互換性のニーズに対応。デフォルトはOpenCV（推奨）。抽出できない場合はFFmpegに切り替えてください（タイムラインがわずかにずれる場合があります）。
 
 [SubtitleArea]
 LowerPart = 下部

backend/interface/ko.ini

@@ -9,6 +9,7 @@ ProjectLinkTitle = 자막 추출기
 ProjectLinkDesc = 비디오 하드 자막 추출, srt 파일 생성. 제3자 API가 필요 없으며, 로컬에서 텍스트 인식 구현. 딥러닝 기반 비디오 자막 추출 프레임워크로, 자막 영역 감지 및 자막 내용 추출 포함.
 BasicSetting = 기본 설정
 AdvancedSetting = 고급 설정
+VideoSubFinderSetting = VideoSubFinder 설정
 DevSetting = 개발 설정
 AboutSetting = 관한
 GenerateTxt = TXT 자막 생성
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = 사용 가능한 업데이트
 UpdatesAvailableDesc = 새 버전 {}이(가) 발견되었습니다. 지금 업데이트하시겠습니까?
 NoUpdatesAvailableTitle = 사용 가능한 업데이트 없음
 NoUpdatesAvailableDesc = 소프트웨어가 최신 버전입니다
+VideoSubFinderCpuCores = CPU 코어 수
+VideoSubFinderCpuCoresDesc = 자막 추출에 사용할 CPU 코어 수입니다. 기본값은 0(자동)이며, 모든 코어를 선택해도 더 빨라지지 않을 수 있습니다.
+VideoSubFinderDecoder = 비디오 디코더
+VideoSubFinderDecoderDesc = 다양한 비디오 호환성 요구에 사용. 기본값은 OpenCV(추천)이며, 추출이 실패할 경우 FFmpeg로 전환하세요(타임라인이 약간 어긋날 수 있음).
 
 [SubtitleArea]
 LowerPart = 하단 부분

backend/interface/vi.ini

@@ -9,6 +9,7 @@ ProjectLinkTitle = Trích xuất phụ đề
 ProjectLinkDesc = Trích xuất phụ đề cứng từ video, tạo file srt. Không cần API bên thứ ba, nhận dạng văn bản cục bộ. Một framework trích xuất phụ đề video dựa trên học sâu, bao gồm phát hiện vùng phụ đề và trích xuất nội dung phụ đề.
 BasicSetting = Cài đặt cơ bản
 AdvancedSetting = Cài đặt nâng cao
+VideoSubFinderSetting = Cài đặt VideoSubFinder
 DevSetting = Cài đặt phát triển
 AboutSetting = Liên quan đến
 GenerateTxt = Tạo phụ đề TXT
@@ -50,6 +51,10 @@ UpdatesAvailableTitle = Có bản cập nhật mới
 UpdatesAvailableDesc = Phát hiện phiên bản mới {}, cập nhật ngay bây giờ?
 NoUpdatesAvailableTitle = Không có bản cập nhật mới
 NoUpdatesAvailableDesc = Phần mềm đã là phiên bản mới nhất
+VideoSubFinderCpuCores = Số lõi CPU
+VideoSubFinderCpuCoresDesc = Số lõi CPU dùng để tách phụ đề. Mặc định là 0 (tự động). Dùng tất cả lõi không phải lúc nào cũng nhanh hơn.
+VideoSubFinderDecoder = Bộ giải mã video
+VideoSubFinderDecoderDesc = Dùng cho các nhu cầu tương thích video khác nhau. Mặc định là OpenCV (khuyến nghị). Chuyển sang FFmpeg nếu tách phụ đề thất bại (có thể lệch nhẹ thời gian).
 
 [SubtitleArea]
 LowerPart = Phần dưới

backend/main.py

@@ -488,10 +488,13 @@ def vsf_output(out, ):
         else:
             # 留2核心来给其他任务使用
             cpu_count = max(multiprocessing.cpu_count() - 2, 1)
+        if config.videoSubFinderCpuCores.value > 0:
+            cpu_count = config.videoSubFinderCpuCores.value
         if platform.system() == 'Windows':
             # 定义执行命令
             cmd = f"{path_vsf} --use_cuda -c -r -i \"{self.video_path}\" -o \"{self.temp_output_dir}\" -ces \"{self.vsf_subtitle}\" "
-            cmd += f"-te {top_end} -be {bottom_end} -le {left_end} -re {right_end} -nthr {cpu_count} -nocrthr {cpu_count}"
+            cmd += f"-te {top_end} -be {bottom_end} -le {left_end} -re {right_end} -nthr {cpu_count} -nocrthr {cpu_count} "
+            cmd += f"--open_video_{config.videoSubFinderDecoder.value.value.lower()} "
             # 计算进度
             try:
                 self.vsf_running = True
@@ -509,7 +512,8 @@ def vsf_output(out, ):
             cmd = f"{path_vsf} -c -r -i \"{self.video_path}\" -o \"{self.temp_output_dir}\" -ces \"{self.vsf_subtitle}\" "
             if self.hardware_accelerator.has_accelerator():
                 cmd += "--use_cuda "
-            cmd += f"-te {top_end} -be {bottom_end} -le {left_end} -re {right_end} -nthr {cpu_count} -dsi"
+            cmd += f"-te {top_end} -be {bottom_end} -le {left_end} -re {right_end} -nthr {cpu_count} -dsi "
+            cmd += f"--open_video_{config.videoSubFinderDecoder.value.value.lower()} "
             self.vsf_running = True
             try:
                 p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, bufsize=1,

backend/tools/constant.py

@@ -20,6 +20,10 @@ class BackgroundColor(Enum):
     UNKNOWN = 2
 
 
+class VideoSubFinderDecoder(Enum):
+    OPENCV = "OpenCV"
+    FFMPEG = "FFmpeg"
+
 BGR_COLOR_GREEN = (0, 0xff, 0)
 BGR_COLOR_BLUE = (0xff, 0, 0)
 BGR_COLOR_RED = (0, 0, 0xff)