Daryl Fung commited on
Commit
b10e574
·
1 Parent(s): 7698439

added print

Browse files
Files changed (1) hide show
  1. animate_motion/demo.py +1 -1
animate_motion/demo.py CHANGED
@@ -103,7 +103,7 @@ def make_animation(source_image, driving_video, generator, kp_detector, relative
103
  kp_driving_initial = kp_detector(driving[:, :, 0])
104
 
105
  for frame_idx in tqdm(range(driving.shape[2])):
106
- print(frame_idx)
107
  driving_frame = driving[:, :, frame_idx]
108
  if not cpu:
109
  driving_frame = driving_frame.cuda()
 
103
  kp_driving_initial = kp_detector(driving[:, :, 0])
104
 
105
  for frame_idx in tqdm(range(driving.shape[2])):
106
+ print(frame_idx, '/', driving.shape[2])
107
  driving_frame = driving[:, :, frame_idx]
108
  if not cpu:
109
  driving_frame = driving_frame.cuda()