Muhammad Taqi Raza
commited on
Commit
·
419a65f
1
Parent(s):
2ecb7ee
adding camera offsets values
Browse files
inference/v2v_data/demo.py
CHANGED
@@ -620,7 +620,7 @@ class GetAnchorVideos:
|
|
620 |
]
|
621 |
)
|
622 |
.unsqueeze(0)
|
623 |
-
.repeat(
|
624 |
.to(opts.device)
|
625 |
)
|
626 |
|
@@ -637,12 +637,12 @@ class GetAnchorVideos:
|
|
637 |
]
|
638 |
)
|
639 |
.unsqueeze(0)
|
640 |
-
.repeat(
|
641 |
.to(opts.device)
|
642 |
)
|
643 |
|
644 |
-
|
645 |
-
c2w_init = torch.matmul(rot_mat_y,
|
646 |
# c2ws[:, 2, 3] -= r
|
647 |
# if x is not None:
|
648 |
# c2ws[:, 1, 3] += y
|
|
|
620 |
]
|
621 |
)
|
622 |
.unsqueeze(0)
|
623 |
+
.repeat(c2w_init.shape[0], 1, 1)
|
624 |
.to(opts.device)
|
625 |
)
|
626 |
|
|
|
637 |
]
|
638 |
)
|
639 |
.unsqueeze(0)
|
640 |
+
.repeat(c2w_init.shape[0], 1, 1)
|
641 |
.to(opts.device)
|
642 |
)
|
643 |
|
644 |
+
c2w_init = torch.matmul(rot_mat_x, c2w_init)
|
645 |
+
c2w_init = torch.matmul(rot_mat_y, c2w_init)
|
646 |
# c2ws[:, 2, 3] -= r
|
647 |
# if x is not None:
|
648 |
# c2ws[:, 1, 3] += y
|