fix small bug when no face is detected

2024-10-08 14:32:21 +02:00
parent 8c721d8202
commit 4919e4e931
1 changed files with 1 additions and 1 deletions
@@ -702,7 +702,7 @@ class FaceWarp:

            if shape_from is None or shape_to is None:
                print(f"\033[96mNo landmarks detected at frame {i}\033[0m")
-                img = img_to.unsqueeze(0)
+                img = image_to[i].unsqueeze(0)
                mask = torch.zeros_like(img)[:,:,:1]
                result_image.append(img)
                result_mask.append(mask)