harrisonpy / rapidvideocr Goto Github PK

简体中文 | English

• 视频硬字幕提取，自动生成对应srt | txt文件。
• 支持字幕语言：中文 | 英文 （其他可以支持的语言参见：支持语种列表)
• 可加入QQ群：706807542
• 更快更准确地提取视频硬字幕，并提供srt| txtl两种格式的输出：
  • 更快：与VideoSubFinder软件结合使用，提取关键字幕帧更快。
  • 更准：采用RapidOCR作为识别库。
  • 更方便：pip直接安装即可使用。
• **该工具处于发展中。在使用过程中，如果遇到任何问题，欢迎提issue或者入群反馈。
• 如果不愿意用的话，不用就好，不要影响自己心情。**
• 如果有帮助到您的话，请给个小星星⭐或者赞助一杯咖啡（点击页面最上面的Sponser中链接）。

• 增加对VideoSubFinder软件提取字幕帧结果的处理接口
• 尝试将VideoSubFinder核心功能整合到本项目中，通过其开放的CLI mode

flowchart LR
    A(VideoSubFinder) --提取字幕关键帧--> B(RapidVideOCR)  --OCR--> C(SRT)

• 修复生成的srt文件中的格式错误， #19

• 针对传入的TXTImages目录，作了优化处理。相比于传入RGBImages，会更快和更准。推荐传入TXTImages目录

1. 安装使用VideoSubFinder软件
   • 下载地址：videosubfinder / QQ群（706807542）共享文件
   • 使用教程：VideoSubFinder提取字幕关键帧教程
   • 最终生成的RGBImages和TXTImages目录一般会在软件安装目录下
   • ✧✧✧ 推荐用TXTImages目录中图像，会更加准确
2. 安装rapid_videocr

   pip install rapid_videocr

3. 使用RapidVideOCR工具
   • 脚本运行：

     from rapid_videocr import RapidVideOCR
     
     extractor = RapidVideOCR()
     
     rgb_dir = 'test_files/TXTImages'
     save_dir = 'result'
     extractor(rgb_dir, save_dir)

   • 命令行运行：
     • Usage:

       $ rapid_videocr -h
       usage: rapid_videocr [-h] [-i IMG_DIR] [-s SAVE_DIR] [-o {srt,txt,all}]
       
       optional arguments:
       -h, --help            show this help message and exit
       -i IMG_DIR, --img_dir IMG_DIR
                               The full path of mp4 video.
       -s SAVE_DIR, --save_dir SAVE_DIR
                               The path of saving the recognition result.
       -o {srt,txt,all}, --out_format {srt,txt,all}
                               Output file format. Default is "all"

     • Example:

       $ rapid_videocr -i RGBImages -s Results -o srt

4. 查看结果
   • 前往save_dir目录下即可查看结果。
   • 值得注意的是，如果想要让视频播放软件自动挂载srt文件，需要更改srt文件名字为视频文件名字，且放到同一目录下，亦或者手动指定加载。

• 扫码加入组织：