mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-05-07 01:57:28 +08:00
使用PySide6-Fluent-Widgets重构整套UI
添加任务列表组件并优化视频加载逻辑 支持可视化显示字幕区域 整理所有模型, 分别为STTN智能擦除, STTN字幕检测, LAMA, ProPainter, OpenCV 提高处理性能 新增CPU运行模式并优化多语言支持 修复Propainter模式部分视频报错 本次提交新增了CPU运行模式,适用于无GPU加速的场景。同时,优化了多语言支持,新增了日语、韩语、越南语等语言配置文件,并更新了README文档以反映新的运行模式和多语言支持。此外,修复了部分代码逻辑,提升了系统的稳定性和兼容性。
This commit is contained in:
41
backend/tools/args_handler.py
Normal file
41
backend/tools/args_handler.py
Normal file
@@ -0,0 +1,41 @@
|
||||
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
|
||||
Reference in New Issue
Block a user