experiencor / image-to-3d-bbox

Build a CNN network to predict 3D bounding box of car from 2D image.

https://experiencor.github.io/sdc_3d.html

237 stars 59 forks source link

bounding-boxes cnn-keras convolutional-neural-networks deep-learning kitti-dataset regression self-driving-car

This is an implementation in Keras of the paper "3D Bounding Box Estimation Using Deep Learning and Geometry" (https://arxiv.org/abs/1612.00496).

Demo on a street video

"3D Box Regression.ipynb" => to construct and train the network to regress 3D bounding boxes.
"Final KITTI Evaluation.ipynb" => to draw 3D bounding boxes on 2D images.