Hard coding for camera parameters TODO: Do some soft coding
(self)
| 189 | return normal_renderer |
| 190 | |
| 191 | def initialize_camera_params(self): |
| 192 | """Hard coding for camera parameters |
| 193 | TODO: Do some soft coding""" |
| 194 | |
| 195 | # Extrinsics |
| 196 | self.R = torch.diag( |
| 197 | torch.tensor([1, 1, 1]) |
| 198 | ).float().to(self.device).unsqueeze(0) |
| 199 | |
| 200 | self.T = torch.tensor( |
| 201 | [0, 0, 0] |
| 202 | ).unsqueeze(0).float().to(self.device) |
| 203 | |
| 204 | # Intrinsics |
| 205 | self.K = self.K.unsqueeze(0).float().to(self.device) |
| 206 | self.bboxes = torch.tensor([[0, 0, self.width, self.height]]).float() |
| 207 | self.K_full, self.image_sizes = update_intrinsics_from_bbox(self.K, self.bboxes) |
| 208 | self.cameras = self.create_camera() |
| 209 | |
| 210 | def render_normal(self, vertices): |
| 211 | vertices = vertices.unsqueeze(0) |
no test coverage detected