├── README.md ├── dataset ├── data_conf.py ├── data_util.py ├── dense_pointcloud.py ├── frustum_dataset.py ├── rcnn_dataset.py └── rpn_dataset.py ├── images ├── 001101.png ├── 001138.png └── architecture2.png ├── kitti ├── kitti_object.py └── kitti_util.py ├── models ├── frustum_model_util.py ├── frustum_pointnets_v2.py ├── img_feature_extractor.py ├── img_seg_net.py ├── img_seg_net_deeplab.py ├── img_vgg_pyramid.py ├── model_util.py ├── pointnet_util.py ├── projection.py ├── rcnn.py ├── rpn.py ├── rpn_img.py ├── tf_ops │ ├── 3d_interpolation │ │ ├── interpolate.cpp │ │ ├── tf_interpolate.cpp │ │ ├── tf_interpolate.py │ │ ├── tf_interpolate_compile.sh │ │ ├── tf_interpolate_op_test.py │ │ └── visu_interpolation.py │ ├── grouping │ │ ├── compile.sh │ │ ├── query_ball_point.cpp │ │ ├── query_ball_point.cu │ │ ├── query_ball_point_block.cu │ │ ├── query_ball_point_grid.cu │ │ ├── selection_sort.cpp │ │ ├── selection_sort.cu │ │ ├── selection_sort_const.cu │ │ ├── test_knn.py │ │ ├── tf_grouping.cpp │ │ ├── tf_grouping.py │ │ ├── tf_grouping_compile.sh │ │ ├── tf_grouping_g.cu │ │ └── tf_grouping_op_test.py │ ├── pointSIFT_op │ │ ├── __init__.py │ │ ├── main.cpp │ │ ├── pointSIFT.cu │ │ ├── pointSIFT_op.py │ │ └── tf_pointSIFT_compile.sh │ └── sampling │ │ ├── tf_sampling.cpp │ │ ├── tf_sampling.py │ │ ├── tf_sampling_compile.sh │ │ └── tf_sampling_g.cu └── tf_util.py ├── test.sh ├── test ├── detect.py ├── test_frustum.py ├── test_rcnn.py └── test_rpn.py ├── test_frustum.sh ├── test_rcnn.sh ├── train ├── provider.py ├── stat_proposal.py ├── test.py ├── test_frustum.py ├── test_rcnn.py ├── train.py ├── train_frustum.py ├── train_img_seg.py ├── train_rcnn.py ├── train_rpn.py ├── train_rpn_img.py └── train_util.py ├── train_frustum.sh ├── train_rcnn.sh ├── train_rpn.sh ├── utils ├── box_encoder.py ├── box_util.py ├── nms_rotate.py ├── parameterize.py └── provider.py └── visualize ├── images_to_video.py ├── img_seg.py ├── mayavi └── mayavi_utils.py ├── obj_detection ├── __init__.py ├── calib_utils.py ├── evaluation.py ├── evaluation_test.py ├── obj_utils.py └── obj_utils_test.py ├── vis_utils.py └── viz_detection.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/README.md -------------------------------------------------------------------------------- /dataset/data_conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/data_conf.py -------------------------------------------------------------------------------- /dataset/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/data_util.py -------------------------------------------------------------------------------- /dataset/dense_pointcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/dense_pointcloud.py -------------------------------------------------------------------------------- /dataset/frustum_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/frustum_dataset.py -------------------------------------------------------------------------------- /dataset/rcnn_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/rcnn_dataset.py -------------------------------------------------------------------------------- /dataset/rpn_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/dataset/rpn_dataset.py -------------------------------------------------------------------------------- /images/001101.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/images/001101.png -------------------------------------------------------------------------------- /images/001138.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/images/001138.png -------------------------------------------------------------------------------- /images/architecture2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/images/architecture2.png -------------------------------------------------------------------------------- /kitti/kitti_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/kitti/kitti_object.py -------------------------------------------------------------------------------- /kitti/kitti_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/kitti/kitti_util.py -------------------------------------------------------------------------------- /models/frustum_model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/frustum_model_util.py -------------------------------------------------------------------------------- /models/frustum_pointnets_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/frustum_pointnets_v2.py -------------------------------------------------------------------------------- /models/img_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/img_feature_extractor.py -------------------------------------------------------------------------------- /models/img_seg_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/img_seg_net.py -------------------------------------------------------------------------------- /models/img_seg_net_deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/img_seg_net_deeplab.py -------------------------------------------------------------------------------- /models/img_vgg_pyramid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/img_vgg_pyramid.py -------------------------------------------------------------------------------- /models/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/model_util.py -------------------------------------------------------------------------------- /models/pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/pointnet_util.py -------------------------------------------------------------------------------- /models/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/projection.py -------------------------------------------------------------------------------- /models/rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/rcnn.py -------------------------------------------------------------------------------- /models/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/rpn.py -------------------------------------------------------------------------------- /models/rpn_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/rpn_img.py -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/interpolate.cpp -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/tf_interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/tf_interpolate.cpp -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/tf_interpolate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/tf_interpolate.py -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/tf_interpolate_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/tf_interpolate_compile.sh -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/tf_interpolate_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/tf_interpolate_op_test.py -------------------------------------------------------------------------------- /models/tf_ops/3d_interpolation/visu_interpolation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/3d_interpolation/visu_interpolation.py -------------------------------------------------------------------------------- /models/tf_ops/grouping/compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/compile.sh -------------------------------------------------------------------------------- /models/tf_ops/grouping/query_ball_point.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/query_ball_point.cpp -------------------------------------------------------------------------------- /models/tf_ops/grouping/query_ball_point.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/query_ball_point.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/query_ball_point_block.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/query_ball_point_block.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/query_ball_point_grid.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/query_ball_point_grid.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/selection_sort.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/selection_sort.cpp -------------------------------------------------------------------------------- /models/tf_ops/grouping/selection_sort.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/selection_sort.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/selection_sort_const.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/selection_sort_const.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/test_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/test_knn.py -------------------------------------------------------------------------------- /models/tf_ops/grouping/tf_grouping.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/tf_grouping.cpp -------------------------------------------------------------------------------- /models/tf_ops/grouping/tf_grouping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/tf_grouping.py -------------------------------------------------------------------------------- /models/tf_ops/grouping/tf_grouping_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/tf_grouping_compile.sh -------------------------------------------------------------------------------- /models/tf_ops/grouping/tf_grouping_g.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/tf_grouping_g.cu -------------------------------------------------------------------------------- /models/tf_ops/grouping/tf_grouping_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/grouping/tf_grouping_op_test.py -------------------------------------------------------------------------------- /models/tf_ops/pointSIFT_op/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/tf_ops/pointSIFT_op/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/pointSIFT_op/main.cpp -------------------------------------------------------------------------------- /models/tf_ops/pointSIFT_op/pointSIFT.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/pointSIFT_op/pointSIFT.cu -------------------------------------------------------------------------------- /models/tf_ops/pointSIFT_op/pointSIFT_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/pointSIFT_op/pointSIFT_op.py -------------------------------------------------------------------------------- /models/tf_ops/pointSIFT_op/tf_pointSIFT_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/pointSIFT_op/tf_pointSIFT_compile.sh -------------------------------------------------------------------------------- /models/tf_ops/sampling/tf_sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/sampling/tf_sampling.cpp -------------------------------------------------------------------------------- /models/tf_ops/sampling/tf_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/sampling/tf_sampling.py -------------------------------------------------------------------------------- /models/tf_ops/sampling/tf_sampling_compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/sampling/tf_sampling_compile.sh -------------------------------------------------------------------------------- /models/tf_ops/sampling/tf_sampling_g.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_ops/sampling/tf_sampling_g.cu -------------------------------------------------------------------------------- /models/tf_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/models/tf_util.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test.sh -------------------------------------------------------------------------------- /test/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test/detect.py -------------------------------------------------------------------------------- /test/test_frustum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test/test_frustum.py -------------------------------------------------------------------------------- /test/test_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test/test_rcnn.py -------------------------------------------------------------------------------- /test/test_rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test/test_rpn.py -------------------------------------------------------------------------------- /test_frustum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test_frustum.sh -------------------------------------------------------------------------------- /test_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/test_rcnn.sh -------------------------------------------------------------------------------- /train/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/provider.py -------------------------------------------------------------------------------- /train/stat_proposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/stat_proposal.py -------------------------------------------------------------------------------- /train/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/test.py -------------------------------------------------------------------------------- /train/test_frustum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/test_frustum.py -------------------------------------------------------------------------------- /train/test_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/test_rcnn.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train.py -------------------------------------------------------------------------------- /train/train_frustum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_frustum.py -------------------------------------------------------------------------------- /train/train_img_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_img_seg.py -------------------------------------------------------------------------------- /train/train_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_rcnn.py -------------------------------------------------------------------------------- /train/train_rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_rpn.py -------------------------------------------------------------------------------- /train/train_rpn_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_rpn_img.py -------------------------------------------------------------------------------- /train/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train/train_util.py -------------------------------------------------------------------------------- /train_frustum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train_frustum.sh -------------------------------------------------------------------------------- /train_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train_rcnn.sh -------------------------------------------------------------------------------- /train_rpn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/train_rpn.sh -------------------------------------------------------------------------------- /utils/box_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/utils/box_encoder.py -------------------------------------------------------------------------------- /utils/box_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/utils/box_util.py -------------------------------------------------------------------------------- /utils/nms_rotate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/utils/nms_rotate.py -------------------------------------------------------------------------------- /utils/parameterize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/utils/parameterize.py -------------------------------------------------------------------------------- /utils/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/utils/provider.py -------------------------------------------------------------------------------- /visualize/images_to_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/images_to_video.py -------------------------------------------------------------------------------- /visualize/img_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/img_seg.py -------------------------------------------------------------------------------- /visualize/mayavi/mayavi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/mayavi/mayavi_utils.py -------------------------------------------------------------------------------- /visualize/obj_detection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /visualize/obj_detection/calib_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/obj_detection/calib_utils.py -------------------------------------------------------------------------------- /visualize/obj_detection/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/obj_detection/evaluation.py -------------------------------------------------------------------------------- /visualize/obj_detection/evaluation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/obj_detection/evaluation_test.py -------------------------------------------------------------------------------- /visualize/obj_detection/obj_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/obj_detection/obj_utils.py -------------------------------------------------------------------------------- /visualize/obj_detection/obj_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/obj_detection/obj_utils_test.py -------------------------------------------------------------------------------- /visualize/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/vis_utils.py -------------------------------------------------------------------------------- /visualize/viz_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JenningsL/PointRCNN/HEAD/visualize/viz_detection.py --------------------------------------------------------------------------------