├── .gitignore ├── LICENSE ├── README.md ├── data └── KITTI │ └── ImageSets │ ├── test.txt │ ├── train.txt │ ├── trainval.txt │ └── val.txt ├── experiments └── example │ ├── clean.sh │ ├── kitti_example.yaml │ ├── slurm_train.sh │ └── train.sh ├── lib ├── backbones │ ├── dla.py │ ├── dlaup.py │ └── hourglass.py ├── datasets │ ├── kitti │ │ ├── kitti_dataset.py │ │ ├── kitti_eval_python │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── eval.py │ │ │ ├── evaluate.py │ │ │ ├── kitti_common.py │ │ │ └── rotate_iou.py │ │ └── kitti_utils.py │ └── utils.py ├── helpers │ ├── dataloader_helper.py │ ├── decode_helper.py │ ├── model_helper.py │ ├── optimizer_helper.py │ ├── save_helper.py │ ├── scheduler_helper.py │ ├── tester_helper.py │ ├── trainer_helper.py │ └── utils_helper.py ├── losses │ ├── centernet_loss.py │ ├── dim_aware_loss.py │ ├── focal_loss.py │ └── uncertainty_loss.py └── models │ └── centernet3d.py ├── requirements.txt ├── resources └── example.jpg └── tools ├── get_data_distribution.py ├── loc_error_by_shifting.py └── train_val.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/README.md -------------------------------------------------------------------------------- /data/KITTI/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/data/KITTI/ImageSets/test.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/data/KITTI/ImageSets/train.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/data/KITTI/ImageSets/trainval.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/data/KITTI/ImageSets/val.txt -------------------------------------------------------------------------------- /experiments/example/clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/experiments/example/clean.sh -------------------------------------------------------------------------------- /experiments/example/kitti_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/experiments/example/kitti_example.yaml -------------------------------------------------------------------------------- /experiments/example/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/experiments/example/slurm_train.sh -------------------------------------------------------------------------------- /experiments/example/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/experiments/example/train.sh -------------------------------------------------------------------------------- /lib/backbones/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/backbones/dla.py -------------------------------------------------------------------------------- /lib/backbones/dlaup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/backbones/dlaup.py -------------------------------------------------------------------------------- /lib/backbones/hourglass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/backbones/hourglass.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_dataset.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/LICENSE -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/README.md -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/eval.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/evaluate.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/kitti_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/kitti_common.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_eval_python/rotate_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_eval_python/rotate_iou.py -------------------------------------------------------------------------------- /lib/datasets/kitti/kitti_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/kitti/kitti_utils.py -------------------------------------------------------------------------------- /lib/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/datasets/utils.py -------------------------------------------------------------------------------- /lib/helpers/dataloader_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/dataloader_helper.py -------------------------------------------------------------------------------- /lib/helpers/decode_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/decode_helper.py -------------------------------------------------------------------------------- /lib/helpers/model_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/model_helper.py -------------------------------------------------------------------------------- /lib/helpers/optimizer_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/optimizer_helper.py -------------------------------------------------------------------------------- /lib/helpers/save_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/save_helper.py -------------------------------------------------------------------------------- /lib/helpers/scheduler_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/scheduler_helper.py -------------------------------------------------------------------------------- /lib/helpers/tester_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/tester_helper.py -------------------------------------------------------------------------------- /lib/helpers/trainer_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/trainer_helper.py -------------------------------------------------------------------------------- /lib/helpers/utils_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/helpers/utils_helper.py -------------------------------------------------------------------------------- /lib/losses/centernet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/losses/centernet_loss.py -------------------------------------------------------------------------------- /lib/losses/dim_aware_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/losses/dim_aware_loss.py -------------------------------------------------------------------------------- /lib/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/losses/focal_loss.py -------------------------------------------------------------------------------- /lib/losses/uncertainty_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/losses/uncertainty_loss.py -------------------------------------------------------------------------------- /lib/models/centernet3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/lib/models/centernet3d.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | fire 2 | scikit-image 3 | tqdm 4 | pyyaml 5 | numba -------------------------------------------------------------------------------- /resources/example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/resources/example.jpg -------------------------------------------------------------------------------- /tools/get_data_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/tools/get_data_distribution.py -------------------------------------------------------------------------------- /tools/loc_error_by_shifting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/tools/loc_error_by_shifting.py -------------------------------------------------------------------------------- /tools/train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinzhuma/monodle/HEAD/tools/train_val.py --------------------------------------------------------------------------------