From 0d12922b50ad472ad68ea1ad3d92aa9ec718800f Mon Sep 17 00:00:00 2001 From: YaoFANGUK Date: Thu, 28 Dec 2023 08:55:59 +0800 Subject: [PATCH] minor --- backend/main.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/main.py b/backend/main.py index e01ee54..c888eb1 100644 --- a/backend/main.py +++ b/backend/main.py @@ -774,8 +774,11 @@ class SubtitleRemover: sub_list = self.sub_detector.find_subtitle_frame_no(sub_remover=self) self.lama_inpaint = LamaInpaint() original_frame = cv2.imread(self.video_path) - mask = create_mask(original_frame.shape[0:2], sub_list[1]) - inpainted_frame = self.lama_inpaint(original_frame, mask) + if len(sub_list): + mask = create_mask(original_frame.shape[0:2], sub_list[1]) + inpainted_frame = self.lama_inpaint(original_frame, mask) + else: + inpainted_frame = original_frame if self.gui_mode: self.preview_frame = cv2.hconcat([original_frame, inpainted_frame]) cv2.imencode(self.ext, inpainted_frame)[1].tofile(self.video_out_name)