mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-04-14 00:17:32 +08:00
f78e985e1c
添加任务列表组件并优化视频加载逻辑 支持可视化显示字幕区域 整理所有模型, 分别为STTN智能擦除, STTN字幕检测, LAMA, ProPainter, OpenCV 提高处理性能 新增CPU运行模式并优化多语言支持 修复Propainter模式部分视频报错 本次提交新增了CPU运行模式，适用于无GPU加速的场景。同时，优化了多语言支持，新增了日语、韩语、越南语等语言配置文件，并更新了README文档以反映新的运行模式和多语言支持。此外，修复了部分代码逻辑，提升了系统的稳定性和兼容性。
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import argparse
|
|
from enum import Enum
|
|
|
|
from .constant import InpaintMode
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(
|
|
description="Video Subtitle Remover Command Line Tool"
|
|
)
|
|
parser.add_argument(
|
|
"--input", "-i", required=True, type=str,
|
|
help="Input video file path"
|
|
)
|
|
parser.add_argument(
|
|
"--output", "-o", required=False, type=str, default=None,
|
|
help="Output video file path (optional)"
|
|
)
|
|
parser.add_argument(
|
|
"--ymin", type=int, default=None,
|
|
help="Subtitle area ymin (optional)"
|
|
)
|
|
parser.add_argument(
|
|
"--ymax", type=int, default=None,
|
|
help="Subtitle area ymax (optional)"
|
|
)
|
|
parser.add_argument(
|
|
"--xmin", type=int, default=None,
|
|
help="Subtitle area xmin (optional)"
|
|
)
|
|
parser.add_argument(
|
|
"--xmax", type=int, default=None,
|
|
help="Subtitle area xmax (optional)"
|
|
)
|
|
parser.add_argument(
|
|
"--inpaint-mode", type=str, default="sttn-auto",
|
|
choices=[mode.name.lower().replace('_','-') for mode in InpaintMode],
|
|
help="Inpaint mode, default is sttn-auto"
|
|
)
|
|
args = parser.parse_args()
|
|
args.inpaint_mode = InpaintMode[args.inpaint_mode.replace('-','_').upper()]
|
|
return args |