Update src/facerender/modules/make_animation.py

src/facerender/modules/make_animation.py

@@ -206,7 +206,7 @@ def make_animation(source_image, source_semantics, target_semantics,
     source_image = source_image.to(device)
     source_semantics = source_semantics.to(device)
     target_semantics = target_semantics.to(device)
-
+    print("source_image",source_image)
     num_frames = target_semantics.shape[1]
     predictions = []
 
@@ -214,9 +214,9 @@ def make_animation(source_image, source_semantics, target_semantics,
         kp_canonical = kp_detector(source_image)
         he_source = mapping(source_semantics)
         kp_source = keypoint_transformation(kp_canonical, he_source)
-        print("source_image",source_image)
+        print("kp_canonical",kp_canonical)
+        print("he_source",he_source)
         print("kp_source",kp_source)
-        print("kp_norm",kp_norm)
 
         for start_idx in tqdm(range(0, num_frames, batch_size), desc='Face Renderer:', unit='batch'):
             end_idx = min(start_idx + batch_size, num_frames)