├── .gitignore ├── LICENSE ├── README.md ├── data ├── kitti │ └── ImageSets │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt ├── lyft │ └── ImageSets │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt └── waymo │ └── ImageSets │ ├── train.txt │ └── val.txt ├── docs ├── DEMO.md ├── GETTING_STARTED.md ├── INSTALL.md ├── dataset_vs_model.png ├── demo.png ├── framework.png ├── model_framework.png ├── multiple_models_demo.png └── open_mmlab.png ├── pcdet ├── __init__.py ├── config.py ├── datasets │ ├── __init__.py │ ├── augmentor │ │ ├── augmentor_utils.py │ │ ├── data_augmentor.py │ │ └── database_sampler.py │ ├── dataset.py │ ├── kitti │ │ ├── kitti_dataset.py │ │ ├── kitti_eval.py │ │ ├── kitti_object_eval_python │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── eval.py │ │ │ ├── evaluate.py │ │ │ ├── kitti_common.py │ │ │ └── rotate_iou.py │ │ └── kitti_utils.py │ ├── lyft │ │ ├── lyft_dataset.py │ │ └── lyft_utils.py │ ├── nuscenes │ │ ├── nuscenes_dataset.py │ │ └── nuscenes_utils.py │ ├── processor │ │ ├── data_processor.py │ │ └── point_feature_encoder.py │ └── waymo │ │ ├── waymo_dataset.py │ │ ├── waymo_eval.py │ │ └── waymo_utils.py ├── models │ ├── __init__.py │ ├── backbones_2d │ │ ├── __init__.py │ │ ├── base_bev_backbone.py │ │ └── map_to_bev │ │ │ ├── __init__.py │ │ │ ├── height_compression.py │ │ │ └── pointpillar_scatter.py │ ├── backbones_3d │ │ ├── __init__.py │ │ ├── pfe │ │ │ ├── __init__.py │ │ │ └── voxel_set_abstraction.py │ │ ├── pointnet2_backbone.py │ │ ├── spconv_backbone.py │ │ ├── spconv_unet.py │ │ └── vfe │ │ │ ├── __init__.py │ │ │ ├── mean_vfe.py │ │ │ ├── pillar_vfe.py │ │ │ └── vfe_template.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── anchor_head_multi.py │ │ ├── anchor_head_single.py │ │ ├── anchor_head_template.py │ │ ├── point_head_box.py │ │ ├── point_head_simple.py │ │ ├── point_head_template.py │ │ ├── point_intra_part_head.py │ │ └── target_assigner │ │ │ ├── anchor_generator.py │ │ │ ├── atss_target_assigner.py │ │ │ └── axis_aligned_target_assigner.py │ ├── detectors │ │ ├── PartA2_net.py │ │ ├── __init__.py │ │ ├── detector3d_template.py │ │ ├── point_rcnn.py │ │ ├── pointpillar.py │ │ ├── pv_rcnn.py │ │ ├── second_net.py │ │ └── second_net_iou.py │ ├── model_utils │ │ ├── dsnorm.py │ │ └── model_nms_utils.py │ └── roi_heads │ │ ├── __init__.py │ │ ├── partA2_head.py │ │ ├── pointrcnn_head.py │ │ ├── pvrcnn_head.py │ │ ├── roi_head_template.py │ │ ├── second_head.py │ │ └── target_assigner │ │ └── proposal_target_layer.py ├── ops │ ├── iou3d_nms │ │ ├── iou3d_nms_utils.py │ │ └── src │ │ │ ├── iou3d_cpu.cpp │ │ │ ├── iou3d_cpu.h │ │ │ ├── iou3d_nms.cpp │ │ │ ├── iou3d_nms.h │ │ │ ├── iou3d_nms_api.cpp │ │ │ └── iou3d_nms_kernel.cu │ ├── pointnet2 │ │ ├── pointnet2_batch │ │ │ ├── pointnet2_modules.py │ │ │ ├── pointnet2_utils.py │ │ │ └── src │ │ │ │ ├── ball_query.cpp │ │ │ │ ├── ball_query_gpu.cu │ │ │ │ ├── ball_query_gpu.h │ │ │ │ ├── cuda_utils.h │ │ │ │ ├── group_points.cpp │ │ │ │ ├── group_points_gpu.cu │ │ │ │ ├── group_points_gpu.h │ │ │ │ ├── interpolate.cpp │ │ │ │ ├── interpolate_gpu.cu │ │ │ │ ├── interpolate_gpu.h │ │ │ │ ├── pointnet2_api.cpp │ │ │ │ ├── sampling.cpp │ │ │ │ ├── sampling_gpu.cu │ │ │ │ └── sampling_gpu.h │ │ └── pointnet2_stack │ │ │ ├── pointnet2_modules.py │ │ │ ├── pointnet2_utils.py │ │ │ └── src │ │ │ ├── ball_query.cpp │ │ │ ├── ball_query_gpu.cu │ │ │ ├── ball_query_gpu.h │ │ │ ├── cuda_utils.h │ │ │ ├── group_points.cpp │ │ │ ├── group_points_gpu.cu │ │ │ ├── group_points_gpu.h │ │ │ ├── interpolate.cpp │ │ │ ├── interpolate_gpu.cu │ │ │ ├── interpolate_gpu.h │ │ │ ├── pointnet2_api.cpp │ │ │ ├── sampling.cpp │ │ │ ├── sampling_gpu.cu │ │ │ └── sampling_gpu.h │ ├── roiaware_pool3d │ │ ├── roiaware_pool3d_utils.py │ │ └── src │ │ │ ├── roiaware_pool3d.cpp │ │ │ └── roiaware_pool3d_kernel.cu │ └── roipoint_pool3d │ │ ├── roipoint_pool3d_utils.py │ │ └── src │ │ ├── roipoint_pool3d.cpp │ │ └── roipoint_pool3d_kernel.cu └── utils │ ├── __init__.py │ ├── box_coder_utils.py │ ├── box_utils.py │ ├── cal_quality_utils.py │ ├── calibration_kitti.py │ ├── common_utils.py │ ├── commu_utils.py │ ├── loss_utils.py │ ├── memory_ensemble_utils.py │ ├── object3d_kitti.py │ ├── self_training_utils.py │ └── spconv_utils.py ├── requirements.txt ├── setup.py └── tools ├── _init_path.py ├── cfgs ├── da-nuscenes-kitti_models │ ├── pointrcnn │ │ └── pointrcnn_ros.yaml │ ├── pointrcnn_st3d │ │ └── pointrcnn_st3d++_car.yaml │ ├── pvrcnn │ │ ├── pvrcnn_old_anchor.yaml │ │ ├── pvrcnn_old_anchor_ros.yaml │ │ └── pvrcnn_old_anchor_sn.yaml │ ├── pvrcnn_st3d │ │ └── pvrcnn_st3d.yaml │ ├── secondiou │ │ ├── secondiou_cyc.yaml │ │ ├── secondiou_old_anchor.yaml │ │ ├── secondiou_old_anchor_ros.yaml │ │ ├── secondiou_old_anchor_sn.yaml │ │ ├── secondiou_ped.yaml │ │ ├── secondiou_ros_cyc.yaml │ │ ├── secondiou_ros_ped.yaml │ │ ├── secondiou_sn_cyc.yaml │ │ └── secondiou_sn_ped.yaml │ └── secondiou_st3d │ │ ├── secondiou_st3d++_ros_car.yaml │ │ ├── secondiou_st3d++_ros_cyc.yaml │ │ ├── secondiou_st3d++_ros_ped.yaml │ │ ├── secondiou_st3d++_sn_car.yaml │ │ ├── secondiou_st3d++_sn_cyc.yaml │ │ ├── secondiou_st3d++_sn_ped.yaml │ │ └── secondiou_st3d_car.yaml ├── da-waymo-kitti_models │ ├── pvrcnn │ │ ├── pvrcnn_old_anchor.yaml │ │ ├── pvrcnn_old_anchor_ros.yaml │ │ └── pvrcnn_old_anchor_sn.yaml │ ├── pvrcnn_st3d │ │ └── pvrcnn_st3d.yaml │ ├── secondiou │ │ ├── secondiou_cyc.yaml │ │ ├── secondiou_cyc_ros.yaml │ │ ├── secondiou_cyc_sn.yaml │ │ ├── secondiou_old_anchor.yaml │ │ ├── secondiou_old_anchor_ros.yaml │ │ ├── secondiou_old_anchor_sn.yaml │ │ ├── secondiou_ped.yaml │ │ ├── secondiou_ped_ros.yaml │ │ └── secondiou_ped_sn.yaml │ └── secondiou_st3d │ │ ├── secondiou_st3d++_ros_car.yaml │ │ ├── secondiou_st3d++_ros_cyc.yaml │ │ ├── secondiou_st3d++_ros_ped.yaml │ │ ├── secondiou_st3d++_sn_car.yaml │ │ ├── secondiou_st3d++_sn_cyc.yaml │ │ ├── secondiou_st3d++_sn_ped.yaml │ │ └── secondiou_st3d_car.yaml ├── da-waymo-lyft_models │ ├── pvrcnn │ │ ├── pvrcnn.yaml │ │ └── pvrcnn_sn.yaml │ ├── pvrcnn_st3d │ │ └── pvrcnn_st3d.yaml │ ├── secondiou │ │ ├── secondiou.yaml │ │ └── secondiou_sn.yaml │ └── secondiou_st3d │ │ └── secondiou_st3d.yaml ├── da-waymo-nus_models │ ├── pvrcnn │ │ └── pvrcnn.yaml │ ├── pvrcnn_st3d │ │ └── pvrcnn_st3d.yaml │ ├── secondiou │ │ └── secondiou.yaml │ └── secondiou_st3d │ │ └── secondiou_st3d.yaml ├── dataset_configs │ ├── da-kitti_oracle_dataset.yaml │ ├── da_kitti_dataset.yaml │ ├── da_lyft_dataset.yaml │ ├── da_nuscenes_dataset.yaml │ ├── da_nuscenes_kitti_dataset.yaml │ ├── da_nuscenes_oracle_dataset.yaml │ ├── da_waymo_dataset.yaml │ ├── da_waymo_lyft_dataset.yaml │ ├── da_waymo_nuscenes_dataset.yaml │ ├── kitti_dataset.yaml │ ├── lyft_dataset.yaml │ ├── nuscenes_dataset.yaml │ ├── waymo_dataset.yaml │ └── waymo_dataset_no_gtsampling.yaml ├── kitti_models │ ├── PartA2.yaml │ ├── PartA2_free.yaml │ ├── pointpillar.yaml │ ├── pointrcnn.yaml │ ├── pointrcnn_iou.yaml │ ├── pv_rcnn.yaml │ ├── pvrcnn_oracle.yaml │ ├── second.yaml │ ├── second_iou.yaml │ ├── second_multihead.yaml │ └── secondiou_orcale.yaml ├── nuscenes_models │ ├── cbgs_pp_multihead.yaml │ ├── cbgs_second_multihead.yaml │ ├── pvrcnn │ │ └── pvrcnn_oracle.yaml │ ├── second_car.yaml │ ├── second_car_oracle.yaml │ └── secondiou_car_oracle.yaml └── waymo_models │ ├── PartA2.yaml │ ├── pv_rcnn.yaml │ ├── pvrcnn.yaml │ ├── pvrcnn_no_gtsampling.yaml │ ├── second.yaml │ └── second_no_gtsampling.yaml ├── eval_utils ├── eval_utils.py └── model_key_mapping.py ├── scripts ├── dist_test.sh ├── dist_train.sh ├── slurm_test_mgpu.sh ├── slurm_test_single.sh └── slurm_train.sh ├── test.py ├── train.py ├── train_utils ├── __init__.py ├── optimization │ ├── __init__.py │ ├── fastai_optim.py │ └── learning_schedules_fastai.py ├── train_st_utils.py └── train_utils.py └── visual_utils └── visualize_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/README.md -------------------------------------------------------------------------------- /data/kitti/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/kitti/ImageSets/test.txt -------------------------------------------------------------------------------- /data/kitti/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/kitti/ImageSets/train.txt -------------------------------------------------------------------------------- /data/kitti/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/kitti/ImageSets/val.txt -------------------------------------------------------------------------------- /data/lyft/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/lyft/ImageSets/test.txt -------------------------------------------------------------------------------- /data/lyft/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/lyft/ImageSets/train.txt -------------------------------------------------------------------------------- /data/lyft/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/lyft/ImageSets/val.txt -------------------------------------------------------------------------------- /data/waymo/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/waymo/ImageSets/train.txt -------------------------------------------------------------------------------- /data/waymo/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/data/waymo/ImageSets/val.txt -------------------------------------------------------------------------------- /docs/DEMO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/DEMO.md -------------------------------------------------------------------------------- /docs/GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/GETTING_STARTED.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/dataset_vs_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/dataset_vs_model.png -------------------------------------------------------------------------------- /docs/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/demo.png -------------------------------------------------------------------------------- /docs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/framework.png -------------------------------------------------------------------------------- /docs/model_framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/model_framework.png -------------------------------------------------------------------------------- /docs/multiple_models_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/multiple_models_demo.png -------------------------------------------------------------------------------- /docs/open_mmlab.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/docs/open_mmlab.png -------------------------------------------------------------------------------- /pcdet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/__init__.py -------------------------------------------------------------------------------- /pcdet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/config.py -------------------------------------------------------------------------------- /pcdet/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/__init__.py -------------------------------------------------------------------------------- /pcdet/datasets/augmentor/augmentor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/augmentor/augmentor_utils.py -------------------------------------------------------------------------------- /pcdet/datasets/augmentor/data_augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/augmentor/data_augmentor.py -------------------------------------------------------------------------------- /pcdet/datasets/augmentor/database_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/augmentor/database_sampler.py -------------------------------------------------------------------------------- /pcdet/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/dataset.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_dataset.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_eval.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/LICENSE -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/README.md -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/eval.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/evaluate.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py -------------------------------------------------------------------------------- /pcdet/datasets/kitti/kitti_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/kitti/kitti_utils.py -------------------------------------------------------------------------------- /pcdet/datasets/lyft/lyft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/lyft/lyft_dataset.py -------------------------------------------------------------------------------- /pcdet/datasets/lyft/lyft_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/lyft/lyft_utils.py -------------------------------------------------------------------------------- /pcdet/datasets/nuscenes/nuscenes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/nuscenes/nuscenes_dataset.py -------------------------------------------------------------------------------- /pcdet/datasets/nuscenes/nuscenes_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/nuscenes/nuscenes_utils.py -------------------------------------------------------------------------------- /pcdet/datasets/processor/data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/processor/data_processor.py -------------------------------------------------------------------------------- /pcdet/datasets/processor/point_feature_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/processor/point_feature_encoder.py -------------------------------------------------------------------------------- /pcdet/datasets/waymo/waymo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/waymo/waymo_dataset.py -------------------------------------------------------------------------------- /pcdet/datasets/waymo/waymo_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/waymo/waymo_eval.py -------------------------------------------------------------------------------- /pcdet/datasets/waymo/waymo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/datasets/waymo/waymo_utils.py -------------------------------------------------------------------------------- /pcdet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_2d/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_2d/base_bev_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_2d/base_bev_backbone.py -------------------------------------------------------------------------------- /pcdet/models/backbones_2d/map_to_bev/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_2d/map_to_bev/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_2d/map_to_bev/height_compression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_2d/map_to_bev/height_compression.py -------------------------------------------------------------------------------- /pcdet/models/backbones_2d/map_to_bev/pointpillar_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_2d/map_to_bev/pointpillar_scatter.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/pfe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/pfe/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/pfe/voxel_set_abstraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/pfe/voxel_set_abstraction.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/pointnet2_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/pointnet2_backbone.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/spconv_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/spconv_backbone.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/spconv_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/spconv_unet.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/vfe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/vfe/__init__.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/vfe/mean_vfe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/vfe/mean_vfe.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/vfe/pillar_vfe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/vfe/pillar_vfe.py -------------------------------------------------------------------------------- /pcdet/models/backbones_3d/vfe/vfe_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/backbones_3d/vfe/vfe_template.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/anchor_head_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/anchor_head_multi.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/anchor_head_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/anchor_head_single.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/anchor_head_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/anchor_head_template.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/point_head_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/point_head_box.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/point_head_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/point_head_simple.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/point_head_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/point_head_template.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/point_intra_part_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/point_intra_part_head.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/target_assigner/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/target_assigner/anchor_generator.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/target_assigner/atss_target_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/target_assigner/atss_target_assigner.py -------------------------------------------------------------------------------- /pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py -------------------------------------------------------------------------------- /pcdet/models/detectors/PartA2_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/PartA2_net.py -------------------------------------------------------------------------------- /pcdet/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/__init__.py -------------------------------------------------------------------------------- /pcdet/models/detectors/detector3d_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/detector3d_template.py -------------------------------------------------------------------------------- /pcdet/models/detectors/point_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/point_rcnn.py -------------------------------------------------------------------------------- /pcdet/models/detectors/pointpillar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/pointpillar.py -------------------------------------------------------------------------------- /pcdet/models/detectors/pv_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/pv_rcnn.py -------------------------------------------------------------------------------- /pcdet/models/detectors/second_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/second_net.py -------------------------------------------------------------------------------- /pcdet/models/detectors/second_net_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/detectors/second_net_iou.py -------------------------------------------------------------------------------- /pcdet/models/model_utils/dsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/model_utils/dsnorm.py -------------------------------------------------------------------------------- /pcdet/models/model_utils/model_nms_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/model_utils/model_nms_utils.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/__init__.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/partA2_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/partA2_head.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/pointrcnn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/pointrcnn_head.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/pvrcnn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/pvrcnn_head.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/roi_head_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/roi_head_template.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/second_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/second_head.py -------------------------------------------------------------------------------- /pcdet/models/roi_heads/target_assigner/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/models/roi_heads/target_assigner/proposal_target_layer.py -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/iou3d_nms_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/iou3d_nms_utils.py -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_cpu.h -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_nms.cpp -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_nms.h -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_nms_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_nms_api.cpp -------------------------------------------------------------------------------- /pcdet/ops/iou3d_nms/src/iou3d_nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/iou3d_nms/src/iou3d_nms_kernel.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_modules.py -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/ball_query.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/ball_query_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/ball_query_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/cuda_utils.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/group_points.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/group_points_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/group_points_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/interpolate.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/interpolate_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/interpolate_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/pointnet2_api.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/sampling.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_batch/src/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_batch/src/sampling_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/cuda_utils.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/group_points_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/group_points_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/pointnet2_api.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.cpp -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pcdet/ops/pointnet2/pointnet2_stack/src/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/pointnet2/pointnet2_stack/src/sampling_gpu.h -------------------------------------------------------------------------------- /pcdet/ops/roiaware_pool3d/roiaware_pool3d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roiaware_pool3d/roiaware_pool3d_utils.py -------------------------------------------------------------------------------- /pcdet/ops/roiaware_pool3d/src/roiaware_pool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roiaware_pool3d/src/roiaware_pool3d.cpp -------------------------------------------------------------------------------- /pcdet/ops/roiaware_pool3d/src/roiaware_pool3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roiaware_pool3d/src/roiaware_pool3d_kernel.cu -------------------------------------------------------------------------------- /pcdet/ops/roipoint_pool3d/roipoint_pool3d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roipoint_pool3d/roipoint_pool3d_utils.py -------------------------------------------------------------------------------- /pcdet/ops/roipoint_pool3d/src/roipoint_pool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roipoint_pool3d/src/roipoint_pool3d.cpp -------------------------------------------------------------------------------- /pcdet/ops/roipoint_pool3d/src/roipoint_pool3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/ops/roipoint_pool3d/src/roipoint_pool3d_kernel.cu -------------------------------------------------------------------------------- /pcdet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pcdet/utils/box_coder_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/box_coder_utils.py -------------------------------------------------------------------------------- /pcdet/utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/box_utils.py -------------------------------------------------------------------------------- /pcdet/utils/cal_quality_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/cal_quality_utils.py -------------------------------------------------------------------------------- /pcdet/utils/calibration_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/calibration_kitti.py -------------------------------------------------------------------------------- /pcdet/utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/common_utils.py -------------------------------------------------------------------------------- /pcdet/utils/commu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/commu_utils.py -------------------------------------------------------------------------------- /pcdet/utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/loss_utils.py -------------------------------------------------------------------------------- /pcdet/utils/memory_ensemble_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/memory_ensemble_utils.py -------------------------------------------------------------------------------- /pcdet/utils/object3d_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/object3d_kitti.py -------------------------------------------------------------------------------- /pcdet/utils/self_training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/self_training_utils.py -------------------------------------------------------------------------------- /pcdet/utils/spconv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/pcdet/utils/spconv_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/setup.py -------------------------------------------------------------------------------- /tools/_init_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/_init_path.py -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pointrcnn/pointrcnn_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pointrcnn/pointrcnn_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pointrcnn_st3d/pointrcnn_st3d++_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pointrcnn_st3d/pointrcnn_st3d++_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pvrcnn/pvrcnn_old_anchor_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/pvrcnn_st3d/pvrcnn_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/pvrcnn_st3d/pvrcnn_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_old_anchor_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ros_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ros_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ros_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_ros_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_sn_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_sn_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_sn_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou/secondiou_sn_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_ros_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d++_sn_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-nuscenes-kitti_models/secondiou_st3d/secondiou_st3d_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/pvrcnn/pvrcnn_old_anchor_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/pvrcnn_st3d/pvrcnn_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/pvrcnn_st3d/pvrcnn_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_cyc_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_old_anchor_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped_ros.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped_ros.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou/secondiou_ped_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_ros_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_cyc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_cyc.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d++_sn_ped.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-kitti_models/secondiou_st3d/secondiou_st3d_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/pvrcnn/pvrcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/pvrcnn/pvrcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/pvrcnn/pvrcnn_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/pvrcnn/pvrcnn_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/pvrcnn_st3d/pvrcnn_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/pvrcnn_st3d/pvrcnn_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/secondiou/secondiou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/secondiou/secondiou.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/secondiou/secondiou_sn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/secondiou/secondiou_sn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-lyft_models/secondiou_st3d/secondiou_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-lyft_models/secondiou_st3d/secondiou_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-nus_models/pvrcnn/pvrcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-nus_models/pvrcnn/pvrcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-nus_models/pvrcnn_st3d/pvrcnn_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-nus_models/pvrcnn_st3d/pvrcnn_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-nus_models/secondiou/secondiou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-nus_models/secondiou/secondiou.yaml -------------------------------------------------------------------------------- /tools/cfgs/da-waymo-nus_models/secondiou_st3d/secondiou_st3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/da-waymo-nus_models/secondiou_st3d/secondiou_st3d.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da-kitti_oracle_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da-kitti_oracle_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_kitti_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_kitti_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_lyft_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_lyft_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_nuscenes_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_nuscenes_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_nuscenes_kitti_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_nuscenes_kitti_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_nuscenes_oracle_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_nuscenes_oracle_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_waymo_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_waymo_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_waymo_lyft_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_waymo_lyft_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/da_waymo_nuscenes_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/da_waymo_nuscenes_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/kitti_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/kitti_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/lyft_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/lyft_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/nuscenes_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/nuscenes_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/waymo_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/waymo_dataset.yaml -------------------------------------------------------------------------------- /tools/cfgs/dataset_configs/waymo_dataset_no_gtsampling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/dataset_configs/waymo_dataset_no_gtsampling.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/PartA2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/PartA2.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/PartA2_free.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/PartA2_free.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/pointpillar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/pointpillar.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/pointrcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/pointrcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/pointrcnn_iou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/pointrcnn_iou.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/pv_rcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/pv_rcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/pvrcnn_oracle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/pvrcnn_oracle.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/second.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/second.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/second_iou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/second_iou.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/second_multihead.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/second_multihead.yaml -------------------------------------------------------------------------------- /tools/cfgs/kitti_models/secondiou_orcale.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/kitti_models/secondiou_orcale.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/cbgs_pp_multihead.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/cbgs_pp_multihead.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/cbgs_second_multihead.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/cbgs_second_multihead.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/pvrcnn/pvrcnn_oracle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/pvrcnn/pvrcnn_oracle.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/second_car.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/second_car.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/second_car_oracle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/second_car_oracle.yaml -------------------------------------------------------------------------------- /tools/cfgs/nuscenes_models/secondiou_car_oracle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/nuscenes_models/secondiou_car_oracle.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/PartA2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/PartA2.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/pv_rcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/pv_rcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/pvrcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/pvrcnn.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/pvrcnn_no_gtsampling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/pvrcnn_no_gtsampling.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/second.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/second.yaml -------------------------------------------------------------------------------- /tools/cfgs/waymo_models/second_no_gtsampling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/cfgs/waymo_models/second_no_gtsampling.yaml -------------------------------------------------------------------------------- /tools/eval_utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/eval_utils/eval_utils.py -------------------------------------------------------------------------------- /tools/eval_utils/model_key_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/eval_utils/model_key_mapping.py -------------------------------------------------------------------------------- /tools/scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/scripts/dist_test.sh -------------------------------------------------------------------------------- /tools/scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/scripts/dist_train.sh -------------------------------------------------------------------------------- /tools/scripts/slurm_test_mgpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/scripts/slurm_test_mgpu.sh -------------------------------------------------------------------------------- /tools/scripts/slurm_test_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/scripts/slurm_test_single.sh -------------------------------------------------------------------------------- /tools/scripts/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/scripts/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/train_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/train_utils/optimization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train_utils/optimization/__init__.py -------------------------------------------------------------------------------- /tools/train_utils/optimization/fastai_optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train_utils/optimization/fastai_optim.py -------------------------------------------------------------------------------- /tools/train_utils/optimization/learning_schedules_fastai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train_utils/optimization/learning_schedules_fastai.py -------------------------------------------------------------------------------- /tools/train_utils/train_st_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train_utils/train_st_utils.py -------------------------------------------------------------------------------- /tools/train_utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/train_utils/train_utils.py -------------------------------------------------------------------------------- /tools/visual_utils/visualize_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CVMI-Lab/ST3D/HEAD/tools/visual_utils/visualize_utils.py --------------------------------------------------------------------------------