Muhammad Taqi Raza commited on
Commit
6f3d83d
·
1 Parent(s): c920a94

video process resolution

Browse files
Files changed (1) hide show
  1. inference/v2v_data/demo.py +2 -2
inference/v2v_data/demo.py CHANGED
@@ -159,8 +159,8 @@ class GetAnchorVideos:
159
  masks.append(mask2)
160
  cond_video = (torch.cat(warped_images) + 1.0) / 2.0
161
 
162
- cond_video_save = process_video_tensor(cond_video).cpu()
163
- ori_video_save = process_video_tensor((frames+1.0) / 2.0).cpu()
164
  save_cated = torch.cat([ori_video_save, cond_video_save], dim=3)
165
  # post_t captions depth intrinsics joint_videos
166
  save_name = os.path.basename(opts.video_path).split('.')[0]
 
159
  masks.append(mask2)
160
  cond_video = (torch.cat(warped_images) + 1.0) / 2.0
161
 
162
+ cond_video_save = process_video_tensor(cond_video, resolution=(opts.height, opts.width)).cpu()
163
+ ori_video_save = process_video_tensor((frames+1.0) / 2.0, resolution=(opts.height, opts.width)).cpu()
164
  save_cated = torch.cat([ori_video_save, cond_video_save], dim=3)
165
  # post_t captions depth intrinsics joint_videos
166
  save_name = os.path.basename(opts.video_path).split('.')[0]