Muhammad Taqi Raza
commited on
Commit
·
b5a313b
1
Parent(s):
0f464ea
adding camera offsets values
Browse files- gradio_app.py +1 -1
- inference/v2v_data/demo.py +10 -0
gradio_app.py
CHANGED
@@ -45,7 +45,7 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
|
|
45 |
depth_inference_steps, depth_guidance_scale,
|
46 |
window_size, overlap, max_res, sample_size,
|
47 |
seed_input, height, width, aspect_ratio_inputs,
|
48 |
-
init_dx, init_dy, init_dz):
|
49 |
|
50 |
temp_input_path = "/app/temp_input.mp4"
|
51 |
output_dir = "/app/output_anchor"
|
|
|
45 |
depth_inference_steps, depth_guidance_scale,
|
46 |
window_size, overlap, max_res, sample_size,
|
47 |
seed_input, height, width, aspect_ratio_inputs,
|
48 |
+
init_dx, init_dy, init_dz):
|
49 |
|
50 |
temp_input_path = "/app/temp_input.mp4"
|
51 |
output_dir = "/app/output_anchor"
|
inference/v2v_data/demo.py
CHANGED
@@ -592,6 +592,10 @@ class GetAnchorVideos:
|
|
592 |
camera_y = getattr(opts, "init_dy", 0.0)
|
593 |
camera_z = getattr(opts, "init_dz", 0.0)
|
594 |
|
|
|
|
|
|
|
|
|
595 |
c2w_init = (
|
596 |
torch.tensor(
|
597 |
[
|
@@ -602,6 +606,12 @@ class GetAnchorVideos:
|
|
602 |
]
|
603 |
).to(opts.device).unsqueeze(0)
|
604 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
605 |
|
606 |
# c2w_init = (
|
607 |
# torch.tensor(
|
|
|
592 |
camera_y = getattr(opts, "init_dy", 0.0)
|
593 |
camera_z = getattr(opts, "init_dz", 0.0)
|
594 |
|
595 |
+
print("camera_x", camera_x)
|
596 |
+
print("camera_y", camera_y)
|
597 |
+
print("camera_z", camera_z)
|
598 |
+
|
599 |
c2w_init = (
|
600 |
torch.tensor(
|
601 |
[
|
|
|
606 |
]
|
607 |
).to(opts.device).unsqueeze(0)
|
608 |
)
|
609 |
+
|
610 |
+
# c2ws[:, 2, 3] -= r
|
611 |
+
# if x is not None:
|
612 |
+
# c2ws[:, 1, 3] += y
|
613 |
+
# if y is not None:
|
614 |
+
# c2ws[:, 0, 3] -= x
|
615 |
|
616 |
# c2w_init = (
|
617 |
# torch.tensor(
|