(self, point_clouds: torch.Tensor)
| 124 | return self.n_query |
| 125 | |
| 126 | def encode_point_clouds(self, point_clouds: torch.Tensor) -> Dict[str, torch.Tensor]: |
| 127 | h = self.encoder_input_proj(point_clouds.permute(0, 2, 1)) |
| 128 | h = self.encoder(h) |
| 129 | return dict(latents=h) |
| 130 | |
| 131 | def predict_sdf( |
| 132 | self, x: torch.Tensor, encoded: Optional[Dict[str, torch.Tensor]] |
nothing calls this directly
no outgoing calls
no test coverage detected