Skip to content

sauravshanu/mono3d_road

Repository files navigation

Monocular 3D Object Detection in Driving Scenarios using the Road Information:

This repo provides training and evaluation code for monocular 3D object detection. It generates road depth map using the semantic segmentation map and depth map of the scene and then it uses the road depth map to sample better 3D anchors and feeds it to the network for 3D bounding box prediction.

Major backbone of this code is borrowed from very well structed VisualDet3D code.

Here's the overview of this unique method.


Here are the predictions compared to the ground truth(in green).

Setup

pip3 install -r requirement.txt
# build ops (deform convs and iou3d), We will not install operations into the system environment
./make.sh

Training

By default trains for synthia. For KITTI change config/Road_synthia_train_full.py.

./launcher/det_precompute.sh config/Road_synthia_train_full.py road
./launcher/train.sh  --config/Road_synthia_train_full.py 0 mono3d_training_v1

About

Monocular 3D object detection using the road

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published