(
self,
H_c2w: torch.Tensor, # (b, q, 4, 4) camera pose in the world coord
intrinsic: torch.Tensor, # (b, q, 3, 3) camera intrinsics
width_px: int,
height_px: int,
)
| 1598 | |
| 1599 | class Camera: |
| 1600 | def __init__( |
| 1601 | self, |
| 1602 | H_c2w: torch.Tensor, # (b, q, 4, 4) camera pose in the world coord |
| 1603 | intrinsic: torch.Tensor, # (b, q, 3, 3) camera intrinsics |
| 1604 | width_px: int, |
| 1605 | height_px: int, |
| 1606 | ): |
| 1607 | self.H_c2w = H_c2w |
| 1608 | self.intrinsic = intrinsic |
| 1609 | self.width_px = width_px |
| 1610 | self.height_px = height_px |
| 1611 | |
| 1612 | self.attr_names = ['H_c2w', 'intrinsic', 'width_px', 'height_px'] |
| 1613 | |
| 1614 | def index_select(self, dim: int, index: torch.Tensor) -> 'Camera': |
| 1615 | camera = self.clone() |
nothing calls this directly
no outgoing calls
no test coverage detected