(self, aabb)
| 155 | return list(self.tex_encoder.parameters()) + list(self.tex_convs.parameters()) + list(self.tex_decoder.parameters()) |
| 156 | |
| 157 | def reset_aabb(self, aabb): |
| 158 | print("set net aabb:", aabb) |
| 159 | if not isinstance(aabb, torch.Tensor): |
| 160 | aabb = torch.tensor(aabb, dtype=torch.float32) |
| 161 | # self.register_buffer("aabb", aabb.to(self.encoder.weight.device)) |
| 162 | self.aabb = aabb.to(self.geo_encoder.weight.device) |
| 163 | |
| 164 | def encode(self, vol): |
| 165 | geo_feat = self.geo_encoder(vol[:, :1]) |
nothing calls this directly
no outgoing calls
no test coverage detected