Muhammad Taqi Raza
commited on
Commit
·
a653e07
1
Parent(s):
35ab635
absolute paths
Browse files
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)
|