(self, aabb)
| 52 | return list(self.tex_encoder.parameters()) + list(self.tex_convs.parameters()) + list(self.tex_decoder.parameters()) |
| 53 | |
| 54 | def reset_aabb(self, aabb): |
| 55 | print("set net aabb:", aabb) |
| 56 | if not isinstance(aabb, torch.Tensor): |
| 57 | aabb = torch.tensor(aabb, dtype=torch.float32) |
| 58 | # self.register_buffer("aabb", aabb.to(self.encoder.weight.device)) |
| 59 | self.aabb = aabb.to(self.geo_encoder.weight.device) |
| 60 | |
| 61 | def encode(self, vol): |
| 62 | geo_feat = self.geo_encoder(vol[:, :1]) |