Home

Awesome

MonoRCNN

MonoRCNN is a monocular 3D object detection method for autonomous driving, published at ICCV 2021 and WACV 2023. This project is an implementation of MonoRCNN.

Related Link

Visualization

<img src='images/KITTI_testset_000181.png' width=805> <p float="left"> <img src='images/WAYMO_valset_057159.png' width=400> <img src='images/WAYMO_valset_060794.png' width=400> </p>

Installation

Please use the Detectron2 included in this project. To ignore fully occluded objects during training, build.py, rpn.py, and roi_heads.py have been modified.

Dataset Preparation

Model & Log

Organize the downloaded files as follows:

├── projects
│   ├── MonoRCNN
│   │   ├── output
│   │   │   ├── model
│   │   │   ├── log.txt
│   │   │   ├── ...

Test

cd projects/MonoRCNN
./main.py --config-file config/MonoRCNN_KITTI.yaml --num-gpus 1 --resume --eval-only

Set VISUALIZE as True to visualize 3D object detection results (saved in output/evaluation/test/visualization).

Training

cd projects/MonoRCNN
./main.py --config-file config/MonoRCNN_KITTI.yaml --num-gpus 1

Citation

If you find this project useful in your research, please cite:

@inproceedings{MonoRCNN_ICCV21,
    title = {Geometry-based Distance Decomposition for Monocular 3D Object Detection},
    author = {Xuepeng Shi and Qi Ye and 
              Xiaozhi Chen and Chuangrong Chen and 
              Zhixiang Chen and Tae-Kyun Kim},
    booktitle = {ICCV},
    year = {2021},
}
@inproceedings{MonoRCNN_WACV23,
    title = {Multivariate Probabilistic Monocular 3D Object Detection},
    author = {Xuepeng Shi and Zhixiang Chen and Tae-Kyun Kim},
    booktitle = {WACV},
    year = {2023},
}

Acknowledgement