Haiyang-W / CAGroup3D

[NeurIPS2022] This is the official code of "CAGroup3D: Class-Aware Grouping for 3D Object Detection on Point Clouds".
92 stars 9 forks source link

How can I get the number of points for each point cloud in a batch? #14

Closed Daybreak-Zheng closed 1 year ago

Daybreak-Zheng commented 1 year ago

Thank you very much for your excellent work。 How can I get the number of points for each point cloud in a batch? The batch_dict contains the points of all the point clouds in a batch. What do I do if I want to get a point cloud in a batch? Looking forward to your reply!

Haiyang-W commented 1 year ago

Please refer this line. The first dim of points is batch id.