(self, hexplane)
| 19 | self.hex_upsample_block = None # subclass |
| 20 | |
| 21 | def forward(self, hexplane): |
| 22 | hexplane = self.conv_block(hexplane) # increase number of channels |
| 23 | hexplane = [block(plane) for block, plane in zip(self.hex_upsample_block, hexplane)] |
| 24 | coords = [[2, 5], [2, 4], [2, 3], [4, 5], [3, 5], [3, 4]] |
| 25 | hexplane = [plane.unsqueeze(dim1).unsqueeze(dim2).permute(0, 2, 3, 4, 5, 1) for (dim1, dim2), plane in |
| 26 | zip(coords, hexplane)] # unsqueeze |
| 27 | voxel = self.op(hexplane) # broadcast and combine |
| 28 | out = self.forward_voxel(voxel) |
| 29 | return out |
| 30 | |
| 31 | @abstractmethod |
| 32 | def forward_voxel(self, voxel): |
nothing calls this directly
no test coverage detected