From 4afb18ae218a822b35d668c49ebf1de032d5e5a9 Mon Sep 17 00:00:00 2001 From: Jeffrey Wu Date: Tue, 28 May 2024 10:58:25 +0800 Subject: [PATCH] feat: skip for no face frame, instead of raise Exception --- faceless/processors/face_enhancer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/faceless/processors/face_enhancer.py b/faceless/processors/face_enhancer.py index fc33cf5..5d46bb7 100644 --- a/faceless/processors/face_enhancer.py +++ b/faceless/processors/face_enhancer.py @@ -152,7 +152,8 @@ def process_frames(target_frames_dir: str, queue_payloads: List[str]): raise Exception("invalid target image") output_vision_frame = process_frame(target_vision_frame) if output_vision_frame is None: - raise Exception("process frame failed") + continue + # raise Exception("process frame failed") write_image(frame_filepath, output_vision_frame) def enhance_video(frames_dir: str):