# -*- encoding: utf-8 -*-
# @Author: SWHL
# @Contact: [email protected]

# 提取 + 识别
from rapid_videocr import RapidVideoSubFinderOCR

vsf_exe = r"G:\ProgramFiles\VideoSubFinder_6.10_x64\Release_x64\VideoSubFinderWXW.exe"
vsf_ocr = RapidVideoSubFinderOCR(vsf_exe_path=vsf_exe, is_concat_rec=True)

# video_path可以是目录或者具体video路径
video_path = "test_files/tiny/2.mp4"
save_dir = "outputs"
vsf_ocr(video_path, save_dir)


# 只识别
from rapid_videocr import RapidVideOCR

extractor = RapidVideOCR(is_concat_rec=True, is_print_console=False)

rgb_dir = "test_files/RGBImagesTiny"
save_dir = "outputs"
save_name = "a"

# outputs/a.srt  outputs/a.txt
extractor(rgb_dir, save_dir, save_name=save_name)