Add debugging for face alignment to diagnose smile editing issues

runners/simple_runner.py

@@ -77,8 +77,11 @@ def run_alignment(image_path):
   import dlib
   from scripts.align_all_parallel import align_face
 
+  print(f"Loading dlib shape predictor from: pretrained_models/shape_predictor_68_face_landmarks.dat")
   predictor = dlib.shape_predictor("pretrained_models/shape_predictor_68_face_landmarks.dat")
+  print(f"Running face alignment on: {image_path}")
   aligned_image, unalign_dict = align_face(filepath=image_path, predictor=predictor)
+  print(f"Face alignment completed successfully")
   return aligned_image, unalign_dict
 
 
@@ -163,11 +166,13 @@ class SimpleRunner:
         aligned_image_pth = orig_img_pth
 
         if align:
+            print(f"Running face alignment on {orig_img_pth}")
             aligned_image, unalign_dict = run_alignment(orig_img_pth)
             save_align_pth = save_pth.parents[0] / (save_pth.stem + "_aligned.jpg")
             print(f"Save aligned image to {save_align_pth}")
             aligned_image.convert('RGB').save(save_align_pth)
             aligned_image_pth = save_align_pth
+            print(f"Face alignment completed. Using aligned image: {aligned_image_pth}")
 
         if use_mask and mask_path is None:
             print("Prepearing mask")

scripts/align_all_parallel.py

@@ -60,6 +60,7 @@ def align_face(filepath, predictor):
     """
     unalign_dict = {}
     lm = get_landmark(filepath, predictor)
+    print(f"Detected {len(lm)} facial landmarks")
 
     lm_chin = lm[0: 17]  # left-right
     lm_eyebrow_left = lm[17: 22]  # left-right