M3DSSD: Monocular 3D Single Stage Object Detector
Setup
-
pytorch 0.4.1
-
Preparation
Download the full KITTI detection dataset. Then place a softlink (or the actual data) in M3DSSD/data/kitti*.
cd M3DSSD ln -s /path/to/kitti data/kitti
Then use the following scripts to extract the data splits, which use softlinks to the above directory for efficient storage.
# extract the data splits python data/kitti_split1/setup_split.py # build the KITTI devkit eval for each split. sh data/kitti_split1/devkit/cpp/build.sh
Build the nms modules
cd lib/nms make
Build the DCN modules
cd model/DCNv2 sh ./make.sh
Training
Review the configurations in scripts/config for details.
python scripts/train_rpn_3d.py --config=kitti_3d_base --exp_name base
- Tips: It is recommended to load a pre-trained model when training with feature alignment.
Testing
Modify the conf_path
and weights_path
to run test.
python scripts/test_rpn_3d.py
Acknowledgements
- Thanks Garrick Brazil for his great works.
- Thanks CharlesShang for his works.
- Thanks traveller59 for his works