Muhammad Taqi Raza commited on
Commit
a653e07
·
1 Parent(s): 35ab635

absolute paths

Browse files
Files changed (1) hide show
  1. inference/v2v_data/demo.py +3 -0
inference/v2v_data/demo.py CHANGED
@@ -133,6 +133,8 @@ class GetAnchorVideos:
133
  frames = (
134
  torch.from_numpy(frames).permute(0, 3, 1, 2).to(opts.device) * 2.0 - 1.0
135
  ) # 49 576 1024 3 -> 49 3 576 1024, [-1,1]
 
 
136
  assert frames.shape[0] == opts.video_length
137
 
138
  depths = center_crop_to_ratio(depths, resolution=target_resolution)
@@ -162,6 +164,7 @@ class GetAnchorVideos:
162
  # post_t captions depth intrinsics joint_videos
163
  save_name = os.path.basename(opts.video_path).split('.')[0]
164
  save_name = opts.save_name
 
165
 
166
  os.makedirs(f'{opts.out_dir}', exist_ok=True)
167
  os.makedirs(f'{opts.out_dir}/videos', exist_ok=True)
 
133
  frames = (
134
  torch.from_numpy(frames).permute(0, 3, 1, 2).to(opts.device) * 2.0 - 1.0
135
  ) # 49 576 1024 3 -> 49 3 576 1024, [-1,1]
136
+ print(f"==> video frames shape: {frames.shape}")
137
+ print(f"==> opts video length: {opts.video_length}")
138
  assert frames.shape[0] == opts.video_length
139
 
140
  depths = center_crop_to_ratio(depths, resolution=target_resolution)
 
164
  # post_t captions depth intrinsics joint_videos
165
  save_name = os.path.basename(opts.video_path).split('.')[0]
166
  save_name = opts.save_name
167
+ print("video save name:", save_name)
168
 
169
  os.makedirs(f'{opts.out_dir}', exist_ok=True)
170
  os.makedirs(f'{opts.out_dir}/videos', exist_ok=True)