├── .gitignore ├── README.md ├── center_head.py ├── centerpoint ├── dataset │ ├── centerpoint_dataloader.py │ └── point_cloud_dataset.py ├── models │ ├── base_detector.py │ ├── rpn.py │ ├── scn_backbone.py │ ├── single_stage.py │ ├── voxel_encoder.py │ └── voxelnet.py ├── nusc_common.py ├── nuscenes_dataset.py ├── pipelines │ └── __init__.py ├── registry.py └── utils │ ├── __init__.py │ ├── box_torch_ops.py │ ├── center_utils.py │ ├── centernet_loss.py │ ├── centerpoint_dataloader.py │ ├── checkpoint.py │ ├── circle_nms_jit.py │ ├── collate.py │ ├── compose.py │ ├── config.py │ ├── loading.py │ ├── norm_layers.py │ ├── point_cloud_ops.py │ ├── preprocess.py │ ├── sequential_utils.py │ ├── test_aug.py │ ├── voxel_generator.py │ └── weight_init.py ├── centerpoint_detection_inference.py ├── centerpoint_tracking_inference.py ├── configs └── nusc_centerpoint_voxelnet_dcn_0075voxel_flip_testset.py ├── dcn ├── __init__.py ├── deform_conv.py ├── setup.py └── src │ ├── deform_conv_cuda.cpp │ ├── deform_conv_cuda_kernel.cu │ ├── deform_pool_cuda.cpp │ └── deform_pool_cuda_kernel.cu ├── iou3d_nms ├── __init__.py ├── setup.py └── src │ ├── iou3d_nms.cpp │ └── iou3d_nms_kernel.cu ├── plot_centerpoint_detections.py ├── prepare_data_argoverse.py ├── prepare_data_nuscenes.py ├── pub_tracker.py ├── setup.py └── viz_aggregated_sweeps.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/README.md -------------------------------------------------------------------------------- /center_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/center_head.py -------------------------------------------------------------------------------- /centerpoint/dataset/centerpoint_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/dataset/centerpoint_dataloader.py -------------------------------------------------------------------------------- /centerpoint/dataset/point_cloud_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/dataset/point_cloud_dataset.py -------------------------------------------------------------------------------- /centerpoint/models/base_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/base_detector.py -------------------------------------------------------------------------------- /centerpoint/models/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/rpn.py -------------------------------------------------------------------------------- /centerpoint/models/scn_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/scn_backbone.py -------------------------------------------------------------------------------- /centerpoint/models/single_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/single_stage.py -------------------------------------------------------------------------------- /centerpoint/models/voxel_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/voxel_encoder.py -------------------------------------------------------------------------------- /centerpoint/models/voxelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/models/voxelnet.py -------------------------------------------------------------------------------- /centerpoint/nusc_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/nusc_common.py -------------------------------------------------------------------------------- /centerpoint/nuscenes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/nuscenes_dataset.py -------------------------------------------------------------------------------- /centerpoint/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /centerpoint/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/registry.py -------------------------------------------------------------------------------- /centerpoint/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /centerpoint/utils/box_torch_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/box_torch_ops.py -------------------------------------------------------------------------------- /centerpoint/utils/center_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/center_utils.py -------------------------------------------------------------------------------- /centerpoint/utils/centernet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/centernet_loss.py -------------------------------------------------------------------------------- /centerpoint/utils/centerpoint_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/centerpoint_dataloader.py -------------------------------------------------------------------------------- /centerpoint/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/checkpoint.py -------------------------------------------------------------------------------- /centerpoint/utils/circle_nms_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/circle_nms_jit.py -------------------------------------------------------------------------------- /centerpoint/utils/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/collate.py -------------------------------------------------------------------------------- /centerpoint/utils/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/compose.py -------------------------------------------------------------------------------- /centerpoint/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/config.py -------------------------------------------------------------------------------- /centerpoint/utils/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/loading.py -------------------------------------------------------------------------------- /centerpoint/utils/norm_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/norm_layers.py -------------------------------------------------------------------------------- /centerpoint/utils/point_cloud_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/point_cloud_ops.py -------------------------------------------------------------------------------- /centerpoint/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/preprocess.py -------------------------------------------------------------------------------- /centerpoint/utils/sequential_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/sequential_utils.py -------------------------------------------------------------------------------- /centerpoint/utils/test_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/test_aug.py -------------------------------------------------------------------------------- /centerpoint/utils/voxel_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/voxel_generator.py -------------------------------------------------------------------------------- /centerpoint/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint/utils/weight_init.py -------------------------------------------------------------------------------- /centerpoint_detection_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint_detection_inference.py -------------------------------------------------------------------------------- /centerpoint_tracking_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/centerpoint_tracking_inference.py -------------------------------------------------------------------------------- /configs/nusc_centerpoint_voxelnet_dcn_0075voxel_flip_testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/configs/nusc_centerpoint_voxelnet_dcn_0075voxel_flip_testset.py -------------------------------------------------------------------------------- /dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/__init__.py -------------------------------------------------------------------------------- /dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/deform_conv.py -------------------------------------------------------------------------------- /dcn/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/setup.py -------------------------------------------------------------------------------- /dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /dcn/src/deform_pool_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/src/deform_pool_cuda.cpp -------------------------------------------------------------------------------- /dcn/src/deform_pool_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/dcn/src/deform_pool_cuda_kernel.cu -------------------------------------------------------------------------------- /iou3d_nms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/iou3d_nms/__init__.py -------------------------------------------------------------------------------- /iou3d_nms/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/iou3d_nms/setup.py -------------------------------------------------------------------------------- /iou3d_nms/src/iou3d_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/iou3d_nms/src/iou3d_nms.cpp -------------------------------------------------------------------------------- /iou3d_nms/src/iou3d_nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/iou3d_nms/src/iou3d_nms_kernel.cu -------------------------------------------------------------------------------- /plot_centerpoint_detections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/plot_centerpoint_detections.py -------------------------------------------------------------------------------- /prepare_data_argoverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/prepare_data_argoverse.py -------------------------------------------------------------------------------- /prepare_data_nuscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/prepare_data_nuscenes.py -------------------------------------------------------------------------------- /pub_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/pub_tracker.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/setup.py -------------------------------------------------------------------------------- /viz_aggregated_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/johnwlambert/argoverse-centerpoint-simplified/HEAD/viz_aggregated_sweeps.py --------------------------------------------------------------------------------