(self, xyz)
| 75 | return sdf |
| 76 | |
| 77 | def get_values(self, xyz): |
| 78 | # xyz = (xyz - self.bound[:, 0].to(xyz.device)) / self.max_dis.to(xyz.device) |
| 79 | xyz = (xyz - self.bound[:, 0].to(xyz.device)) / self.bound_dis.to(xyz.device) |
| 80 | sdf = self.hash_sdf_out(xyz) |
| 81 | rgb = self.hash_color_out(xyz) |
| 82 | # outputs = torch.cat([rgb, sdf], dim=-1) |
| 83 | |
| 84 | return sdf, rgb |
| 85 | |
| 86 | def forward(self, xyz): |
| 87 | sdf, rgb = self.get_values(xyz) |