Vectorize voxel/sphere distance computation

neel-dey / AnyStar

[WACV 2024] AnyStar: Domain randomized universal star-convex 3D instance segmentation

https://arxiv.org/abs/2307.07044

MIT License

12 stars 3 forks source link

Vectorize voxel/sphere distance computation #2

Closed ckapoor7 closed 5 months ago

ckapoor7 commented 5 months ago

This PR vectorizes voxel-sphere distance computations. Earlier, the first step of label synthesis took a mean time of 7.7 seconds/label, which is now reduced to 2.38 seconds/label. I conducted these tests over synthesis of 20 labels, and have attached plots of each of these. Please let me know if this looks good to you, and is helpful!

Thank you, Chaitanya

vectorized non-vectorized

neel-dey commented 5 months ago

Very cool, thanks! I left a very minor comment re: the description on L32. Can merge right away after that :)

ckapoor7 commented 5 months ago

Awesome! I just pushed the changes -- let me know if anything else seems amiss :)

neel-dey commented 5 months ago

Merged, thanks again!