ranamostafaabdelmohsen / painting

Implementation of PointPainting

PointPainting

An implementation of PointPainting (https://arxiv.org/abs/1911.10150), an image-lidar fusion algorithm for 3D object detection. Code is partly based on the Pointpillars repo as well as this object detection repo. The algorithm involves performing a semantic segmentation of the image, projecting the pointcloud to the resulting segmentation map to give each lidar point its class score, and running the augmented pointcloud through a 2D object detector (SSD) in BEV. Some changes to the algorithm have been made including removal of orientation estimation, and changes to the feature extractor. Predictions are made for the Car class on the KITTI dataset.

Above shows predicted (green) bounding boxes for an image and BEV map as well as ground truths (blue)

Installation and Training steps

Clone the repo:

git clone https://github.com/rshilliday/painting.git
cd painting
pip install -r requirements.txt

Download and unzip the KITTI Dataset (images, lidar, labels, calib) Also download and save weights for the semantic segmentation network

Train and evaluate the network:

python train.py
python eval.py

Future Improvements:

• Add data augmentation
• Add Focal Loss
• Add rotation prediction