Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangyp15 authored Jun 14, 2021
1 parent ac09880 commit a624da8
Showing 1 changed file with 17 additions and 2 deletions.
19 changes: 17 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,20 +38,21 @@ Please download [KITTI dataset](http://www.cvlibs.net/datasets/kitti/eval_object
#ROOT
|data/
|KITTI/
|ImageSets/ [already provided in this repo]
|object/
|training/
|calib/
|image_2/
|label/
|ImageSets/
|testing/
|calib/
|image_2/
|ImageSets/
```

Then modify the paths in config/paths_catalog.py according to your data path.

### Training & Evaluation
## Training & Evaluation

Training with one GPU: (TODO: multi-GPU training will be further tested)

Expand All @@ -67,6 +68,20 @@ CUDA_VISIBLE_DEVICES=0 python tools/plain_train_net.py --config runs/monoflex.ya

You can also specify --vis when evaluation to visualize the predicted heatmap and 3D bounding boxes.

## Citation

If you find our work useful in your research, please consider citing:

```latex
@InProceedings{MonoFlex,
author = {Zhang, Yunpeng and Lu, Jiwen and Zhou, Jie},
title = {Objects Are Different: Flexible Monocular 3D Object Detection},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2021},
pages = {3289-3298}
}
```

## Acknowlegment

Expand Down

0 comments on commit a624da8

Please sign in to comment.