search

traveller59 / second.pytorch

SECOND for KITTI/NuScenes object detection
MIT License
1.72k stars 721 forks source link

pointpillar multiclass object detection #169

Open kargarisaac opened 5 years ago

kargarisaac commented 5 years ago

Hi there, Is there any config file for multiclass object detection? why the config files of pointpillar are different in following two repos: https://github.com/nutonomy/second.pytorch https://github.com/traveller59/second.pytorch which one is more up-to-date?

JieLi1994 commented 5 years ago

mark

traveller59 commented 5 years ago

"https://github.com/nutonomy/second.pytorch" is pointpillars developed by nutonomy, their code is based on SECOND v1.0. "https://github.com/traveller59/second.pytorch" is SECOND developed by me. the pointpillars in this repo is an unofficial implementation, there may be some difference compare to original code. multi class config file is easy to implement, you just need to add some anchor generators and database samplers. there is an example for nuscenes dataset and an example for SECOND. I can't provide a config for kitti because I have no plan to train it.

ryontang commented 3 years ago

mark, @kargarisaac Have you solve this problem of training multiclass object detection by Pointpillars