Ma/video-subtitle-remover
1
0
Fork 0
mirror of https://github.com/YaoFANGUK/video-subtitle-remover.git synced 2026-05-18 03:27:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

去除propainter

Browse Source
This commit is contained in:
yaofanguk
2026-04-10 16:08:17 +08:00
parent c189f8368e
commit bc31d1ff2d
24 changed files with 20 additions and 1164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
backend/tools/constant.py
View File

@@ -8,7 +8,6 @@ class InpaintMode(Enum):
STTN_AUTO = "sttn-auto"
STTN_DET = "sttn-det"
LAMA = "lama"
PROPAINTER = "propainter"
OPENCV = "opencv"
@unique

2
backend/tools/model_config.py
View File

@@ -13,7 +13,6 @@ class ModelConfig:
self.LAMA_MODEL_DIR = os.path.join(BASE_DIR, 'models', 'big-lama')
self.STTN_AUTO_MODEL_PATH = os.path.join(BASE_DIR, 'models', 'sttn-auto', 'infer_model.pth')
self.STTN_DET_MODEL_PATH = os.path.join(BASE_DIR, 'models', 'sttn-det', 'sttn.pth')
self.PROPAINTER_MODEL_DIR = os.path.join(BASE_DIR,'models', 'propainter')
if config.subtitleDetectMode.value == SubtitleDetectMode.PP_OCRv5_MOBILE:
self.DET_MODEL_DIR = os.path.join(BASE_DIR,'models', 'V5', 'ch_det_fast')
elif config.subtitleDetectMode.value == SubtitleDetectMode.PP_OCRv5_SERVER:
@@ -23,4 +22,3 @@ class ModelConfig:
self.DET_MODEL_NAME = _MODEL_NAME_MAP[config.subtitleDetectMode.value]
merge_big_file_if_not_exists(self.LAMA_MODEL_DIR, 'bit-lama.pt')
merge_big_file_if_not_exists(self.PROPAINTER_MODEL_DIR, 'ProPainter.pth')