Update src/facerender/modules/make_animation.py

src/facerender/modules/make_animation.py

@@ -3,6 +3,7 @@ import torch
 import torch.nn.functional as F
 import numpy as np
 from tqdm import tqdm 
+import time
 
 def normalize_kp(kp_source, kp_driving, kp_driving_initial, adapt_movement_scale=False,
                  use_relative_movement=False, use_relative_jacobian=False):
@@ -154,11 +155,16 @@ def make_animation(source_image, source_semantics, target_semantics,
     source_semantics = source_semantics.to(device)
     target_semantics = target_semantics.to(device)
 
+    
     with torch.no_grad():  # No gradients needed
         predictions = []
+        start = time.time()
         kp_canonical = kp_detector(source_image)
         he_source = mapping(source_semantics)
         kp_source = keypoint_transformation(kp_canonical, he_source)
+        end = time.time()
+        diff = end-start
+        print("time taken for 3 values within make animation:", diff)
 
         with autocast():
             for frame_idx in tqdm(range(target_semantics.shape[1]), desc='Face Renderer:', unit='frame'):