mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-05-05 08:27:31 +08:00
f78e985e1c
添加任务列表组件并优化视频加载逻辑 支持可视化显示字幕区域 整理所有模型, 分别为STTN智能擦除, STTN字幕检测, LAMA, ProPainter, OpenCV 提高处理性能 新增CPU运行模式并优化多语言支持 修复Propainter模式部分视频报错 本次提交新增了CPU运行模式，适用于无GPU加速的场景。同时，优化了多语言支持，新增了日语、韩语、越南语等语言配置文件，并更新了README文档以反映新的运行模式和多语言支持。此外，修复了部分代码逻辑，提升了系统的稳定性和兼容性。
15 lines
325 B
Python
15 lines
325 B
Python
import cv2
|
|
|
|
class OpenCVInpaint:
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def inpaint(self, frame, mask):
|
|
return cv2.inpaint(frame, mask, 3, cv2.INTER_LINEAR)
|
|
|
|
def __call__(self, frames, mask):
|
|
comp = []
|
|
for frame in frames:
|
|
comp.append(self.inpaint(frame, mask))
|
|
return comp |