pytorch_cluster icon indicating copy to clipboard operation
pytorch_cluster copied to clipboard

voxel_grid enhancement -> Re-indexation from 0

Open tchaton opened this issue 5 years ago • 0 comments
trafficstars

Hey @rusty1s,

I am using voxel_grid, and on my 8000 point cloud, it creates a 5001 indexes with the highest being around 400k.

As a result, the pool_pos used to make the aggregation create a massive tensor, which isn't great for memory reduction.

The new features will add reindex=True, return new batch and reindexed indices from 0 to N.

Best. Thomas Chaton.

tchaton avatar Jan 13 '20 11:01 tchaton