open-mmlab / OpenPCDet

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
Apache License 2.0
4.72k stars 1.31k forks source link

Custom Dataset Labels #1674

Open preverte opened 3 weeks ago

preverte commented 3 weeks ago

Hello,

I've been recently trying to set up a custom dataset from pointclouds gathered from a RSHelios16P (similar pattern to VLP16). This is for an agricultural project, in which the goal is use the 3D clouds to detect trees.

As for what I understand from CUSTOM_DATASET_TUTORIAL.md, custom datasets are somehow treated as KITTI dataset, so this class mapping to KITTI classes needs to be done: image

Does this mean that only labels already defined in KITTI dataset can be used? What if I want to detect classes that are not specified in the KITTI dataset?

Also, is there any available custom dataset adapter following the tutorial online?

Thank you!

Petros626 commented 3 weeks ago

For my understanding the custom dataset template relies on the kittidataset, so the classes were limited to these three. In the official labels there were more labels available, but for different reasons only the classical classes are used. When you try to detect trees with different classes, than you have to change all the classes in custom template and kittidataset. You can also search through the issues some people created tutorials for your case.