├── .gitignore ├── .gitmodules ├── .idea ├── .gitignore ├── PointRCNN-PyTorch.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.md ├── data └── KITTI │ └── ImageSets │ ├── test.txt │ ├── train.txt │ ├── trainval.txt │ └── val.txt ├── doc └── teaser.png ├── lib ├── config.py ├── datasets │ ├── kitti_dataset.py │ └── kitti_rcnn_dataset.py ├── net │ ├── point_rcnn.py │ ├── pointnet2_msg.py │ ├── rcnn_net.py │ ├── rpn.py │ └── train_functions.py ├── rpn │ ├── proposal_layer.py │ └── proposal_target_layer.py └── utils │ ├── bbox_transform.py │ ├── calibration.py │ ├── iou3d │ ├── iou3d_utils.py │ ├── setup.py │ └── src │ │ ├── iou3d.cpp │ │ └── iou3d_kernel.cu │ ├── kitti_utils.py │ ├── loss_utils.py │ ├── object3d.py │ └── roipool3d │ ├── roipool3d_utils.py │ ├── setup.py │ └── src │ ├── roipool3d.cpp │ └── roipool3d_kernel.cu └── tools ├── _init_path.py ├── cfgs └── default.yaml ├── eval_rcnn.py ├── generate_aug_scene.py ├── generate_gt_database.py ├── kitti_object_eval_python ├── LICENSE ├── README.md ├── eval.py ├── evaluate.py ├── kitti_common.py └── rotate_iou.py ├── script_train.sh ├── train_rcnn.py └── train_utils ├── fastai_optim.py ├── learning_schedules_fastai.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.gitmodules -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | # Default ignored files 3 | /workspace.xml -------------------------------------------------------------------------------- /.idea/PointRCNN-PyTorch.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.idea/PointRCNN-PyTorch.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /data/KITTI/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/data/KITTI/ImageSets/test.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/data/KITTI/ImageSets/train.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/data/KITTI/ImageSets/trainval.txt -------------------------------------------------------------------------------- /data/KITTI/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/data/KITTI/ImageSets/val.txt -------------------------------------------------------------------------------- /doc/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/doc/teaser.png -------------------------------------------------------------------------------- /lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/config.py -------------------------------------------------------------------------------- /lib/datasets/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/datasets/kitti_dataset.py -------------------------------------------------------------------------------- /lib/datasets/kitti_rcnn_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/datasets/kitti_rcnn_dataset.py -------------------------------------------------------------------------------- /lib/net/point_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/net/point_rcnn.py -------------------------------------------------------------------------------- /lib/net/pointnet2_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/net/pointnet2_msg.py -------------------------------------------------------------------------------- /lib/net/rcnn_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/net/rcnn_net.py -------------------------------------------------------------------------------- /lib/net/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/net/rpn.py -------------------------------------------------------------------------------- /lib/net/train_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/net/train_functions.py -------------------------------------------------------------------------------- /lib/rpn/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/rpn/proposal_layer.py -------------------------------------------------------------------------------- /lib/rpn/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/rpn/proposal_target_layer.py -------------------------------------------------------------------------------- /lib/utils/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/bbox_transform.py -------------------------------------------------------------------------------- /lib/utils/calibration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/calibration.py -------------------------------------------------------------------------------- /lib/utils/iou3d/iou3d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/iou3d/iou3d_utils.py -------------------------------------------------------------------------------- /lib/utils/iou3d/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/iou3d/setup.py -------------------------------------------------------------------------------- /lib/utils/iou3d/src/iou3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/iou3d/src/iou3d.cpp -------------------------------------------------------------------------------- /lib/utils/iou3d/src/iou3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/iou3d/src/iou3d_kernel.cu -------------------------------------------------------------------------------- /lib/utils/kitti_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/kitti_utils.py -------------------------------------------------------------------------------- /lib/utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/loss_utils.py -------------------------------------------------------------------------------- /lib/utils/object3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/object3d.py -------------------------------------------------------------------------------- /lib/utils/roipool3d/roipool3d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/roipool3d/roipool3d_utils.py -------------------------------------------------------------------------------- /lib/utils/roipool3d/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/roipool3d/setup.py -------------------------------------------------------------------------------- /lib/utils/roipool3d/src/roipool3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/roipool3d/src/roipool3d.cpp -------------------------------------------------------------------------------- /lib/utils/roipool3d/src/roipool3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/lib/utils/roipool3d/src/roipool3d_kernel.cu -------------------------------------------------------------------------------- /tools/_init_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/_init_path.py -------------------------------------------------------------------------------- /tools/cfgs/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/cfgs/default.yaml -------------------------------------------------------------------------------- /tools/eval_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/eval_rcnn.py -------------------------------------------------------------------------------- /tools/generate_aug_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/generate_aug_scene.py -------------------------------------------------------------------------------- /tools/generate_gt_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/generate_gt_database.py -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/LICENSE -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/README.md -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/eval.py -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/evaluate.py -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/kitti_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/kitti_common.py -------------------------------------------------------------------------------- /tools/kitti_object_eval_python/rotate_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/kitti_object_eval_python/rotate_iou.py -------------------------------------------------------------------------------- /tools/script_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/script_train.sh -------------------------------------------------------------------------------- /tools/train_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/train_rcnn.py -------------------------------------------------------------------------------- /tools/train_utils/fastai_optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/train_utils/fastai_optim.py -------------------------------------------------------------------------------- /tools/train_utils/learning_schedules_fastai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/train_utils/learning_schedules_fastai.py -------------------------------------------------------------------------------- /tools/train_utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelBunker/PointRCNN-PyTorch/HEAD/tools/train_utils/train_utils.py --------------------------------------------------------------------------------