(self, featmap_size)
| 349 | return preds |
| 350 | |
| 351 | def _resize_aabb(self, featmap_size): |
| 352 | if featmap_size[0] != self.featmap_size[0] or featmap_size[1] != self.featmap_size[1] or featmap_size[2] != self.featmap_size[2]: |
| 353 | scale = torch.tensor([featmap_size[0] / self.featmap_size[0], featmap_size[1] / self.featmap_size[1], featmap_size[2] / self.featmap_size[2]], device=self.device) |
| 354 | new_aabb = self.aabb.clone() |
| 355 | new_aabb[:3] = self.aabb[:3] * scale |
| 356 | new_aabb[3:] = self.aabb[3:] * scale |
| 357 | print("resized aabb:", new_aabb) |
| 358 | return new_aabb |
| 359 | else: |
| 360 | return self.aabb |
| 361 | |
| 362 | @torch.no_grad() |
| 363 | def decode_texmesh(self, save_dir, triplane_feat, reso, n_faces=10000, n_surf_pc=-1, texture_reso=2048, only_largest_cc=True, |
no outgoing calls
no test coverage detected