├── .gitignore ├── README.md ├── configs ├── nu_configs │ ├── cp_plus.yaml │ └── immortal.yaml ├── rmfp_configs │ ├── rmfp_lift01.yaml │ └── rmfp_test.yaml └── waymo_configs │ └── immortal_for_ctrl_keep10.yaml ├── data_loader ├── __init__.py ├── nuscenes_loader.py └── waymo_loader.py ├── evaluation ├── nuscenes │ ├── eval.sh │ ├── pipeline.sh │ ├── result_creation.py │ └── type_merge.py └── waymo │ ├── __init__.py │ ├── pred_bin.py │ ├── rm_fp_by_gt.py │ └── utils.py ├── main_nuscenes.py ├── main_waymo.py ├── mot_3d ├── __init__.py ├── association.py ├── data_protos │ ├── __init__.py │ ├── bbox.py │ └── validity.py ├── frame_data.py ├── life │ ├── __init__.py │ └── hit_manager.py ├── mot.py ├── motion_model │ ├── __init__.py │ └── kalman_filter.py ├── preprocessing │ ├── __init__.py │ ├── bbox_coarse_hash.py │ └── nms.py ├── tracklet │ ├── __init__.py │ ├── simple_tracklet.py │ └── tracklet.py ├── update_info_data.py ├── utils │ ├── __init__.py │ ├── cuda_ops.py │ ├── data_utils.py │ ├── geometry.py │ └── timer.py └── visualization │ ├── __init__.py │ └── visualizer2d.py ├── preparedata ├── nuscenes │ ├── detection.py │ ├── ego_pose.py │ ├── nu_convert_detection.sh │ ├── nu_preparedata.sh │ ├── time_stamp.py │ └── token_info.py └── waymo │ ├── detection.py │ ├── ego_info.py │ ├── time_stamp.py │ ├── waymo_convert_detection.sh │ └── waymo_preparedata.sh ├── preprocessing ├── detection_nms.py └── py_nms │ ├── __init__.py │ ├── bbox_coarse_hash.py │ └── nms.py ├── requirements.txt ├── run_mot.sh └── tools ├── iou_debug.py ├── len_bin.py ├── vis ├── show_bin.py ├── utils.py └── visualizer.py └── waymo_convert_detection.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/README.md -------------------------------------------------------------------------------- /configs/nu_configs/cp_plus.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/configs/nu_configs/cp_plus.yaml -------------------------------------------------------------------------------- /configs/nu_configs/immortal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/configs/nu_configs/immortal.yaml -------------------------------------------------------------------------------- /configs/rmfp_configs/rmfp_lift01.yaml: -------------------------------------------------------------------------------- 1 | bottom_lift: 0.1 2 | min_points: 0 -------------------------------------------------------------------------------- /configs/rmfp_configs/rmfp_test.yaml: -------------------------------------------------------------------------------- 1 | bottom_lift: 0 2 | min_points: 0 -------------------------------------------------------------------------------- /configs/waymo_configs/immortal_for_ctrl_keep10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/configs/waymo_configs/immortal_for_ctrl_keep10.yaml -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/nuscenes_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/data_loader/nuscenes_loader.py -------------------------------------------------------------------------------- /data_loader/waymo_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/data_loader/waymo_loader.py -------------------------------------------------------------------------------- /evaluation/nuscenes/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/nuscenes/eval.sh -------------------------------------------------------------------------------- /evaluation/nuscenes/pipeline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/nuscenes/pipeline.sh -------------------------------------------------------------------------------- /evaluation/nuscenes/result_creation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/nuscenes/result_creation.py -------------------------------------------------------------------------------- /evaluation/nuscenes/type_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/nuscenes/type_merge.py -------------------------------------------------------------------------------- /evaluation/waymo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluation/waymo/pred_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/waymo/pred_bin.py -------------------------------------------------------------------------------- /evaluation/waymo/rm_fp_by_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/waymo/rm_fp_by_gt.py -------------------------------------------------------------------------------- /evaluation/waymo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/evaluation/waymo/utils.py -------------------------------------------------------------------------------- /main_nuscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/main_nuscenes.py -------------------------------------------------------------------------------- /main_waymo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/main_waymo.py -------------------------------------------------------------------------------- /mot_3d/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mot_3d/association.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/association.py -------------------------------------------------------------------------------- /mot_3d/data_protos/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/data_protos/__init__.py -------------------------------------------------------------------------------- /mot_3d/data_protos/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/data_protos/bbox.py -------------------------------------------------------------------------------- /mot_3d/data_protos/validity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/data_protos/validity.py -------------------------------------------------------------------------------- /mot_3d/frame_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/frame_data.py -------------------------------------------------------------------------------- /mot_3d/life/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/life/__init__.py -------------------------------------------------------------------------------- /mot_3d/life/hit_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/life/hit_manager.py -------------------------------------------------------------------------------- /mot_3d/mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/mot.py -------------------------------------------------------------------------------- /mot_3d/motion_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/motion_model/__init__.py -------------------------------------------------------------------------------- /mot_3d/motion_model/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/motion_model/kalman_filter.py -------------------------------------------------------------------------------- /mot_3d/preprocessing/__init__.py: -------------------------------------------------------------------------------- 1 | from .nms import nms -------------------------------------------------------------------------------- /mot_3d/preprocessing/bbox_coarse_hash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/preprocessing/bbox_coarse_hash.py -------------------------------------------------------------------------------- /mot_3d/preprocessing/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/preprocessing/nms.py -------------------------------------------------------------------------------- /mot_3d/tracklet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/tracklet/__init__.py -------------------------------------------------------------------------------- /mot_3d/tracklet/simple_tracklet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/tracklet/simple_tracklet.py -------------------------------------------------------------------------------- /mot_3d/tracklet/tracklet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/tracklet/tracklet.py -------------------------------------------------------------------------------- /mot_3d/update_info_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/update_info_data.py -------------------------------------------------------------------------------- /mot_3d/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/utils/__init__.py -------------------------------------------------------------------------------- /mot_3d/utils/cuda_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/utils/cuda_ops.py -------------------------------------------------------------------------------- /mot_3d/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/utils/data_utils.py -------------------------------------------------------------------------------- /mot_3d/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/utils/geometry.py -------------------------------------------------------------------------------- /mot_3d/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/utils/timer.py -------------------------------------------------------------------------------- /mot_3d/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/visualization/__init__.py -------------------------------------------------------------------------------- /mot_3d/visualization/visualizer2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/mot_3d/visualization/visualizer2d.py -------------------------------------------------------------------------------- /preparedata/nuscenes/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/detection.py -------------------------------------------------------------------------------- /preparedata/nuscenes/ego_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/ego_pose.py -------------------------------------------------------------------------------- /preparedata/nuscenes/nu_convert_detection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/nu_convert_detection.sh -------------------------------------------------------------------------------- /preparedata/nuscenes/nu_preparedata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/nu_preparedata.sh -------------------------------------------------------------------------------- /preparedata/nuscenes/time_stamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/time_stamp.py -------------------------------------------------------------------------------- /preparedata/nuscenes/token_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/nuscenes/token_info.py -------------------------------------------------------------------------------- /preparedata/waymo/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/waymo/detection.py -------------------------------------------------------------------------------- /preparedata/waymo/ego_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/waymo/ego_info.py -------------------------------------------------------------------------------- /preparedata/waymo/time_stamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/waymo/time_stamp.py -------------------------------------------------------------------------------- /preparedata/waymo/waymo_convert_detection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/waymo/waymo_convert_detection.sh -------------------------------------------------------------------------------- /preparedata/waymo/waymo_preparedata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preparedata/waymo/waymo_preparedata.sh -------------------------------------------------------------------------------- /preprocessing/detection_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preprocessing/detection_nms.py -------------------------------------------------------------------------------- /preprocessing/py_nms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preprocessing/py_nms/__init__.py -------------------------------------------------------------------------------- /preprocessing/py_nms/bbox_coarse_hash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preprocessing/py_nms/bbox_coarse_hash.py -------------------------------------------------------------------------------- /preprocessing/py_nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/preprocessing/py_nms/nms.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_mot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/run_mot.sh -------------------------------------------------------------------------------- /tools/iou_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/iou_debug.py -------------------------------------------------------------------------------- /tools/len_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/len_bin.py -------------------------------------------------------------------------------- /tools/vis/show_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/vis/show_bin.py -------------------------------------------------------------------------------- /tools/vis/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/vis/utils.py -------------------------------------------------------------------------------- /tools/vis/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/vis/visualizer.py -------------------------------------------------------------------------------- /tools/waymo_convert_detection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Abyssaledge/ImmortalTracker-for-CTRL/HEAD/tools/waymo_convert_detection.sh --------------------------------------------------------------------------------