From 6211c353aaf7ce6ad2bdb8bb59ef4589ecb9a324 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 6 Jul 2024 08:54:02 -0400
Subject: [PATCH] faceswap handle no faces found

---
 CHANGELOG.md             | 1 +
 modules/face/faceswap.py | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index e04990523..2d0cb6b41 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -31,6 +31,7 @@ TODO:
 - fix current step for higher order samplers
 - fix control input type video  
 - fix reset pipeline at the end of each iteration  
+- fix faceswap when no faces detected  
 - multiple ModernUI fixes
 
 ## Update for 2024-06-23
diff --git a/modules/face/faceswap.py b/modules/face/faceswap.py
index 2d03a5232..2868d6c78 100644
--- a/modules/face/faceswap.py
+++ b/modules/face/faceswap.py
@@ -22,6 +22,9 @@ def face_swap(p: processing.StableDiffusionProcessing, app, input_images: List[I
 
     np_image = cv2.cvtColor(np.array(source_image), cv2.COLOR_RGB2BGR)
     faces = app.get(np_image)
+    if faces is None or len(faces) == 0:
+        shared.log.warning('FaceSwap: No faces detected')
+        return
     source_face = faces[0]
     processed_images = []
     for image in input_images: