batch singe images

kex0 · Jun 3, 2023 · bdb501b · bdb501b
2 parents 1072b86 + d30265b
commit bdb501b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/roop/core.py b/roop/core.py
@@ -204,7 +204,7 @@ def start():
                 quit()
             process_img(args['source_img'], target_path, os.path.splitext(args['output_file'])[0] + f"_{i+1}.png")
             status("swap successful!")
-            return
+            continue
         seconds, probabilities = predict_video_frames(video_path=args['target_path'], frame_interval=100)
         if any(probability > 0.85 for probability in probabilities):
             quit()

diff --git a/roop/swapper.py b/roop/swapper.py
@@ -60,6 +60,9 @@ def process_video(source_img, frame_paths):
 def process_img(source_img, target_path, output_file):
     frame = cv2.imread(target_path)
     face = get_face_single(frame)
+    if not face:
+        print(f"No face found in {target_path}")
+        return
     source_face = get_face_single(cv2.imread(source_img))
     result = get_face_swapper().get(frame, face, source_face, paste_back=True)
     cv2.imwrite(output_file, result)