def convert(og_mesh, voxel): transforms = tfs.Compose([mesh_conversion, tfs.MeshLaplacianSmoothing(smoothing_iterations)]) new_mesh = transforms(voxel) new_mesh.vertices = pcfunc.realign(new_mesh.vertices, og_mesh.vertices) return {'vertices': new_mesh.vertices, 'faces': new_mesh.faces}
def test_ModelNetPointCloud(device): transform = tfs.Compose([ tfs.TriangleMeshToPointCloud(num_samples=32), tfs.NormalizePointCloud() ]) models = kal.datasets.ModelNet(basedir=MODELNET_ROOT, categories=['bathtub'], split='test', transform=transform) assert len(models) == 50 for obj, category in models: assert category.item() == 0 assert isinstance(obj, torch.Tensor) assert obj.size(0) == 32