(self)
| 204 | self.assertEqual(ofield.F.size(1), (otensor.F.size(1) + tfield.F.size(1))) |
| 205 | |
| 206 | def stride_slice(self): |
| 207 | coords, colors, pcd = load_file("1.ply") |
| 208 | voxel_size = 0.02 |
| 209 | colors = torch.from_numpy(colors).float() |
| 210 | bcoords = batched_coordinates([coords / voxel_size], dtype=torch.float32) |
| 211 | tfield = TensorField(colors, bcoords) |
| 212 | |
| 213 | network = nn.Sequential( |
| 214 | MinkowskiToSparseTensor(), |
| 215 | MinkowskiConvolution(3, 8, kernel_size=3, stride=4, dimension=3), |
| 216 | MinkowskiReLU(), |
| 217 | MinkowskiConvolution(8, 16, kernel_size=3, stride=4, dimension=3), |
| 218 | ) |
| 219 | |
| 220 | otensor = network(tfield) |
| 221 | ofield = otensor.slice(tfield) |
| 222 | self.assertTrue(len(ofield) == len(tfield)) |
| 223 | |
| 224 | def field_to_sparse(self): |
| 225 | coords, colors, pcd = load_file("1.ply") |
nothing calls this directly
no test coverage detected