Update src/facerender/modules/make_animation.py
Browse files
src/facerender/modules/make_animation.py
CHANGED
@@ -235,6 +235,10 @@ def make_animation(source_image, source_semantics, target_semantics,
|
|
235 |
kp_driving = keypoint_transformation(kp_canonical, he_driving)
|
236 |
kp_norm = kp_driving
|
237 |
|
|
|
|
|
|
|
|
|
238 |
# Ensure correct input shape for conv1d layers
|
239 |
source_image_reshaped = source_image.view(source_image.size(0), source_image.size(1), -1)
|
240 |
kp_source_reshaped = kp_source.view(kp_source.size(0), kp_source.size(1), -1)
|
|
|
235 |
kp_driving = keypoint_transformation(kp_canonical, he_driving)
|
236 |
kp_norm = kp_driving
|
237 |
|
238 |
+
print("source_image",source_image)
|
239 |
+
print("kp_source",kp_source)
|
240 |
+
print("kp_norm",kp_norm)
|
241 |
+
|
242 |
# Ensure correct input shape for conv1d layers
|
243 |
source_image_reshaped = source_image.view(source_image.size(0), source_image.size(1), -1)
|
244 |
kp_source_reshaped = kp_source.view(kp_source.size(0), kp_source.size(1), -1)
|