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

cli版字幕去除

Browse Source
This commit is contained in:
YaoFANGUK
2023-10-25 17:07:52 +08:00
parent 2b9360c299
commit d897dede0b
26 changed files with 4 additions and 2573 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/main.py
View File

@@ -55,6 +55,7 @@ class SubtitleDetect:
def find_subtitle_frame_no(self):
video_cap = cv2.VideoCapture(self.video_path)
frame_count = video_cap.get(cv2.CAP_PROP_FRAME_COUNT)
current_frame_no = 0
subtitle_frame_no_list = {}
@@ -80,6 +81,7 @@ class SubtitleDetect:
else:
temp_list.append((xmin, xmax, ymin, ymax))
subtitle_frame_no_list[current_frame_no] = temp_list
print(f'[字幕查找]{current_frame_no}/{int(frame_count)}')
return subtitle_frame_no_list
@@ -144,7 +146,7 @@ class SubtitleRemover:
masks = self.create_mask(frame, sub_list[index])
frame = self.inpaint_frame(frame, masks)
self.video_writer.write(frame)
print(f'{index}/{int(self.frame_count)}')
print(f'[字幕去除]{index}/{int(self.frame_count)}')
self.video_cap.release()
self.video_writer.release()