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

修复bug

Browse Source
This commit is contained in:
YaoFANGUK
2023-12-13 19:50:15 +08:00
parent 6b817bd57a
commit 29c5317a69
3 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
backend/main.py
View File

@@ -525,15 +525,12 @@ class SubtitleRemover:
self.progress_total = 50 + self.progress_remover
def run(self):
# 记录开始时间
start_time = time.time()
# 寻找字幕帧
# 重置进度条
self.progress_total = 0
# 寻找字幕帧
sub_list = self.sub_detector.find_subtitle_frame_no(sub_remover=self)
# 测试代码
# from test1_dict_raw import test1_raw
# sub_list = self.sub_detector.unify_regions(test1_raw)
continuous_frame_no_list = self.sub_detector.find_continuous_ranges_with_same_mask(sub_list)
# 获取场景分割的帧号
scene_div_points = self.sub_detector.get_scene_div_frame_no(self.video_path)
@@ -546,9 +543,11 @@ class SubtitleRemover:
self.lama_inpaint = LamaInpaint()
original_frame = cv2.imread(self.video_path)
mask = create_mask(original_frame.shape[0:2], sub_list[1])
frame = self.lama_inpaint(original_frame, mask)
cv2.imencode(self.ext, frame)[1].tofile(self.video_out_name)
self.preview_frame = cv2.hconcat([original_frame, frame])
inpainted_frame = self.lama_inpaint(original_frame, mask)
print(original_frame.shape)
print(inpainted_frame.shape)
self.preview_frame = cv2.hconcat([original_frame, inpainted_frame])
cv2.imencode(self.ext, inpainted_frame)[1].tofile(self.video_out_name)
tbar.update(1)
self.progress_total = 100
else: