Muhammad Taqi Raza commited on
Commit
419a65f
·
1 Parent(s): 2ecb7ee

adding camera offsets values

Browse files
Files changed (1) hide show
  1. inference/v2v_data/demo.py +4 -4
inference/v2v_data/demo.py CHANGED
@@ -620,7 +620,7 @@ class GetAnchorVideos:
620
  ]
621
  )
622
  .unsqueeze(0)
623
- .repeat(c2ws.shape[0], 1, 1)
624
  .to(opts.device)
625
  )
626
 
@@ -637,12 +637,12 @@ class GetAnchorVideos:
637
  ]
638
  )
639
  .unsqueeze(0)
640
- .repeat(c2ws.shape[0], 1, 1)
641
  .to(opts.device)
642
  )
643
 
644
- c2ws = torch.matmul(rot_mat_x, c2ws)
645
- c2w_init = torch.matmul(rot_mat_y, c2ws)
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