Files
video-subtitle-remover/backend/tools/constant.py
2025-05-22 19:52:36 +08:00

22 lines
481 B
Python

from enum import Enum, unique
@unique
class InpaintMode(Enum):
"""
图像重绘算法枚举
"""
STTN_AUTO = "sttn-auto"
STTN_DET = "sttn-det"
LAMA = "lama"
PROPAINTER = "propainter"
OPENCV = "opencv"
@unique
class SubtitleDetectMode(Enum):
"""
字幕检测算法枚举
"""
PP_OCRv4_SERVER = "PP-OCRv4-Server"
PP_OCRv4_MOBILE = "PP-OCRv4-Mobile"
PP_OCRv5_SERVER = "PP-OCRv5-Server"
PP_OCRv5_MOBILE = "PP-OCRv5-Mobile"