(self, aabb)
| 262 | list(self.rgb_decoder.parameters()) + list(self.mr_decoder.parameters()) + list(self.normal_decoder.parameters()) |
| 263 | |
| 264 | def reset_aabb(self, aabb): |
| 265 | print("set net aabb:", aabb) |
| 266 | if not isinstance(aabb, torch.Tensor): |
| 267 | aabb = torch.tensor(aabb, dtype=torch.float32) |
| 268 | # self.register_buffer("aabb", aabb.to(self.encoder.weight.device)) |
| 269 | self.aabb = aabb.to(self.geo_encoder.weight.device) |
| 270 | |
| 271 | def encode(self, vol): |
| 272 | geo_feat = self.geo_encoder(vol[:, :1]) |
nothing calls this directly
no outgoing calls
no test coverage detected