how to train multi-classes

sshaoshuai / PointRCNN

PointRCNN: 3D Object Proposal Generation and Detection from Point Cloud, CVPR 2019.

MIT License

1.72k stars 425 forks source link

how to train multi-classes #105

Open shengyuwoo opened 5 years ago

shengyuwoo commented 5 years ago

Hi, guys, thanks to your job. Now, I want to train a model of multi-classes, for example, pedestrian, car and cyclists. But I don't known how to modify your code, can you give me some advice? Which files need to be modified?

gakkiri commented 5 years ago

Also wondering, is it possible to just change the output of rcnn_cls?

direcf commented 3 years ago

https://github.com/direcf/PointRCNN_multiclass If any issues exist, please tell me. I will fix it. Maybe I will offer detail documentation and annotation untiil March.

Thx