diff --git a/backend/inpaint/sttn_inpaint.py b/backend/inpaint/sttn_inpaint.py index c721115..ceed187 100644 --- a/backend/inpaint/sttn_inpaint.py +++ b/backend/inpaint/sttn_inpaint.py @@ -291,10 +291,10 @@ class STTNVideoInpaint: mask_area = mask[inpaint_area[k][0]:inpaint_area[k][1], :] frame[inpaint_area[k][0]:inpaint_area[k][1], :, :] = mask_area * comp + (1 - mask_area) * frame[inpaint_area[k][0]:inpaint_area[k][1], :, :] writer.write(frame) - if input_sub_remover is not None and input_sub_remover.gui_mode: + if input_sub_remover is not None: if tbar is not None: input_sub_remover.update_progress(tbar, increment=1) - if original_frame is not None: + if original_frame is not None and input_sub_remover.gui_mode: input_sub_remover.preview_frame = cv2.hconcat([original_frame, frame]) # 释放视频写入对象 writer.release() diff --git a/backend/main.py b/backend/main.py index 3c19993..d59614f 100644 --- a/backend/main.py +++ b/backend/main.py @@ -670,12 +670,14 @@ class SubtitleRemover: print('[Processing] start removing subtitles...') if self.sub_area is not None: ymin, ymax, xmin, xmax = self.sub_area - mask_area_coordinates = [(xmin, xmax, ymin, ymax)] - mask = create_mask(self.mask_size, mask_area_coordinates) - sttn_video_inpaint = STTNVideoInpaint(self.video_path) - sttn_video_inpaint(input_mask=mask, input_sub_remover=self, tbar=tbar) else: - print('please set subtitle area first') + print('[Info] No subtitle area has been set. Video will be processed in full screen. As a result, the final outcome might be suboptimal.') + ymin, ymax, xmin, xmax = 0, self.frame_height, 0, self.frame_width + mask_area_coordinates = [(xmin, xmax, ymin, ymax)] + mask = create_mask(self.mask_size, mask_area_coordinates) + sttn_video_inpaint = STTNVideoInpaint(self.video_path) + sttn_video_inpaint(input_mask=mask, input_sub_remover=self, tbar=tbar) + def sttn_mode(self, tbar): # 是否跳过字幕帧寻找