steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

requirements updated

Browse Source
This commit is contained in:
Si11ium 2020-06-26 15:09:27 +02:00
parent f942cac30e
commit cf99ac1aec
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main_pipeline.py
View File

@ -72,7 +72,9 @@ if __name__ == '__main__':
test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False, test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False,
refresh=True, transform=transforms) # , cluster_type='pc') refresh=True, transform=transforms) # , cluster_type='pc')
grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=8192)
# Here you can choose which dataset to load, in case there are more than one pointcloud files in raw/predict
grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=4096)
ps.init() ps.init()

BIN
requirements.txt
View File

Binary file not shown.