Muhammad Taqi Raza
commited on
Commit
·
6f3d83d
1
Parent(s):
c920a94
video process resolution
Browse files
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]
|