├── .gitignore ├── INSTALL.md ├── README.md ├── data ├── __init__.py ├── kitti_dataset.py ├── kitti_gt_dataset.py ├── kitti_object_dataset.py ├── kitti_object_roi_dataset.py ├── kitti_prediction_dataset.py └── load_dataset.py ├── datasets └── KITTI_object │ ├── annotations │ ├── data_object_training_annotations.json │ ├── inference_annotations.json │ ├── object_inference_annotations.json │ ├── train_annotations.json │ └── val_annotations.json │ ├── test.txt │ ├── train.txt │ ├── training │ ├── depth │ ├── image_2 │ ├── label_2 │ └── planes │ ├── trainval.txt │ └── val.txt ├── experiments └── foresee │ ├── depth_normal_model.py │ ├── lateral_net.py │ ├── loss.py │ ├── pc │ ├── bin2obj.py │ ├── gen.sh │ ├── gen_colorps.sh │ ├── kitti_prediction.py │ └── kitti_prediction_colorps.py │ ├── test.sh │ ├── train.sh │ ├── train_kitti.py │ └── val_kitti.py ├── lib ├── __init__.py ├── core │ ├── __init__.py │ └── config.py ├── models │ ├── MobileNetV2.py │ ├── ResNeXt.py │ ├── __init__.py │ ├── image_transfer.py │ ├── lateral_net.py │ └── loss.py └── utils │ ├── __init__.py │ ├── bounding_box.py │ ├── chamfer_distance │ ├── __init__.py │ ├── chamfer_distance.cpp │ ├── chamfer_distance.cu │ └── chamfer_distance.py │ ├── collections.py │ ├── evaluate_depth_error.py │ ├── logging.py │ ├── misc.py │ ├── mobilenetv2_weight_helper.py │ ├── net_tools.py │ ├── obj_utils.py │ ├── resnext_weights_helper.py │ ├── timer.py │ └── training_stats.py ├── prepocessing ├── gen_depth.sh ├── generate_depth.py ├── generate_disp.py ├── generate_lidar.py ├── kitti_process_RANSAC.py └── kitti_util.py ├── pretrained_model └── ResNeXt_ImageNet └── tools ├── __init__.py ├── kitti_prediction.py ├── kitti_prediction_sample.py ├── kitti_prediction_sample_diff.py ├── parse_arg_base.py ├── parse_arg_test.py ├── parse_arg_train.py ├── parse_arg_val.py ├── test_kitti.py ├── train_kitti.py └── val_kitti.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pth 2 | *.pyc 3 | __pycache__/ 4 | 5 | -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/INSTALL.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/kitti_dataset.py -------------------------------------------------------------------------------- /data/kitti_gt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/kitti_gt_dataset.py -------------------------------------------------------------------------------- /data/kitti_object_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/kitti_object_dataset.py -------------------------------------------------------------------------------- /data/kitti_object_roi_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/kitti_object_roi_dataset.py -------------------------------------------------------------------------------- /data/kitti_prediction_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/kitti_prediction_dataset.py -------------------------------------------------------------------------------- /data/load_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/data/load_dataset.py -------------------------------------------------------------------------------- /datasets/KITTI_object/annotations/data_object_training_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/annotations/data_object_training_annotations.json -------------------------------------------------------------------------------- /datasets/KITTI_object/annotations/inference_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/annotations/inference_annotations.json -------------------------------------------------------------------------------- /datasets/KITTI_object/annotations/object_inference_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/annotations/object_inference_annotations.json -------------------------------------------------------------------------------- /datasets/KITTI_object/annotations/train_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/annotations/train_annotations.json -------------------------------------------------------------------------------- /datasets/KITTI_object/annotations/val_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/annotations/val_annotations.json -------------------------------------------------------------------------------- /datasets/KITTI_object/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/test.txt -------------------------------------------------------------------------------- /datasets/KITTI_object/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/train.txt -------------------------------------------------------------------------------- /datasets/KITTI_object/training/depth: -------------------------------------------------------------------------------- 1 | /mnt/cephfs/common/lab/wangxinlong/data/KITTI/Kitti/object/training/depth -------------------------------------------------------------------------------- /datasets/KITTI_object/training/image_2: -------------------------------------------------------------------------------- 1 | /mnt/cephfs/common/lab/wangxinlong/data/KITTI/data_object_image_2/training/image_2 -------------------------------------------------------------------------------- /datasets/KITTI_object/training/label_2: -------------------------------------------------------------------------------- 1 | /mnt/cephfs/common/lab/wangxinlong/data/KITTI/data_object_label_2/training/label_2 -------------------------------------------------------------------------------- /datasets/KITTI_object/training/planes: -------------------------------------------------------------------------------- 1 | /mnt/cephfs/common/lab/wangxinlong/data/KITTI/planes -------------------------------------------------------------------------------- /datasets/KITTI_object/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/trainval.txt -------------------------------------------------------------------------------- /datasets/KITTI_object/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/datasets/KITTI_object/val.txt -------------------------------------------------------------------------------- /experiments/foresee/depth_normal_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/depth_normal_model.py -------------------------------------------------------------------------------- /experiments/foresee/lateral_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/lateral_net.py -------------------------------------------------------------------------------- /experiments/foresee/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/loss.py -------------------------------------------------------------------------------- /experiments/foresee/pc/bin2obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/pc/bin2obj.py -------------------------------------------------------------------------------- /experiments/foresee/pc/gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/pc/gen.sh -------------------------------------------------------------------------------- /experiments/foresee/pc/gen_colorps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/pc/gen_colorps.sh -------------------------------------------------------------------------------- /experiments/foresee/pc/kitti_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/pc/kitti_prediction.py -------------------------------------------------------------------------------- /experiments/foresee/pc/kitti_prediction_colorps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/pc/kitti_prediction_colorps.py -------------------------------------------------------------------------------- /experiments/foresee/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/test.sh -------------------------------------------------------------------------------- /experiments/foresee/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/train.sh -------------------------------------------------------------------------------- /experiments/foresee/train_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/train_kitti.py -------------------------------------------------------------------------------- /experiments/foresee/val_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/experiments/foresee/val_kitti.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/core/config.py -------------------------------------------------------------------------------- /lib/models/MobileNetV2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/models/MobileNetV2.py -------------------------------------------------------------------------------- /lib/models/ResNeXt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/models/ResNeXt.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/image_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/models/image_transfer.py -------------------------------------------------------------------------------- /lib/models/lateral_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/models/lateral_net.py -------------------------------------------------------------------------------- /lib/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/models/loss.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/bounding_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/bounding_box.py -------------------------------------------------------------------------------- /lib/utils/chamfer_distance/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/chamfer_distance/__init__.py -------------------------------------------------------------------------------- /lib/utils/chamfer_distance/chamfer_distance.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/chamfer_distance/chamfer_distance.cpp -------------------------------------------------------------------------------- /lib/utils/chamfer_distance/chamfer_distance.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/chamfer_distance/chamfer_distance.cu -------------------------------------------------------------------------------- /lib/utils/chamfer_distance/chamfer_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/chamfer_distance/chamfer_distance.py -------------------------------------------------------------------------------- /lib/utils/collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/collections.py -------------------------------------------------------------------------------- /lib/utils/evaluate_depth_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/evaluate_depth_error.py -------------------------------------------------------------------------------- /lib/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/logging.py -------------------------------------------------------------------------------- /lib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/misc.py -------------------------------------------------------------------------------- /lib/utils/mobilenetv2_weight_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/mobilenetv2_weight_helper.py -------------------------------------------------------------------------------- /lib/utils/net_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/net_tools.py -------------------------------------------------------------------------------- /lib/utils/obj_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/obj_utils.py -------------------------------------------------------------------------------- /lib/utils/resnext_weights_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/resnext_weights_helper.py -------------------------------------------------------------------------------- /lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/timer.py -------------------------------------------------------------------------------- /lib/utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/lib/utils/training_stats.py -------------------------------------------------------------------------------- /prepocessing/gen_depth.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/gen_depth.sh -------------------------------------------------------------------------------- /prepocessing/generate_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/generate_depth.py -------------------------------------------------------------------------------- /prepocessing/generate_disp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/generate_disp.py -------------------------------------------------------------------------------- /prepocessing/generate_lidar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/generate_lidar.py -------------------------------------------------------------------------------- /prepocessing/kitti_process_RANSAC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/kitti_process_RANSAC.py -------------------------------------------------------------------------------- /prepocessing/kitti_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/prepocessing/kitti_util.py -------------------------------------------------------------------------------- /pretrained_model/ResNeXt_ImageNet: -------------------------------------------------------------------------------- 1 | /data00/home/wangxinlong/experiments/VNL_mono_depth/datasets/pretrained_model/ResNeXt_ImageNet -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/kitti_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/kitti_prediction.py -------------------------------------------------------------------------------- /tools/kitti_prediction_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/kitti_prediction_sample.py -------------------------------------------------------------------------------- /tools/kitti_prediction_sample_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/kitti_prediction_sample_diff.py -------------------------------------------------------------------------------- /tools/parse_arg_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/parse_arg_base.py -------------------------------------------------------------------------------- /tools/parse_arg_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/parse_arg_test.py -------------------------------------------------------------------------------- /tools/parse_arg_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/parse_arg_train.py -------------------------------------------------------------------------------- /tools/parse_arg_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/parse_arg_val.py -------------------------------------------------------------------------------- /tools/test_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/test_kitti.py -------------------------------------------------------------------------------- /tools/train_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/train_kitti.py -------------------------------------------------------------------------------- /tools/val_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WXinlong/ForeSeE/HEAD/tools/val_kitti.py --------------------------------------------------------------------------------