Semi-supervised 3D Object Detection with PatchTeacher and PillarMix (AAAI 2024)

This is the official implementation of Semi-supervised 3D Object Detection with PatchTeacher and PillarMix (AAAI 2024), built on OpenPCDet

Installation

Please refer to INSTALL.md for the installation of OpenPCDet.

Quick Demo

Please refer to DEMO.md for a quick demo to test with a pretrained model and visualize the predicted results on your custom data or the original KITTI data.

Getting Started

Please refer to GETTING_STARTED.md to learn more usage about this project.

Training PTPM on Waymo Open DataSet

Please refer to PTPM_WAYMO.md to train our method on Waymo Open Dataset.

Training PTPM on ONCE DataSet

Please refer to PTPM_ONCE.md to train our method on ONCE Dataset.

Citation

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

@inproceedings{wu2024semi,
  title={Semi-supervised 3D Object Detection with PatchTeacher and PillarMix},
  author={Wu, Xiaopei and Peng, Liang and Xie, Liang and Hou, Yuenan and Lin, Binbin and Huang, Xiaoshui and Liu, Haifeng and Cai, Deng and Ouyang, Wanli},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={38},
  number={6},
  pages={6153--6161},
  year={2024}
}