├── .gitignore ├── LICENSE ├── README.md ├── evaluate.py ├── figures ├── img_3dbbox_000134.png ├── pc_pred_000134.png └── pytorch_trt.png ├── misc ├── log.md └── vis_data_gt.py ├── pointpillars ├── __init__.py ├── dataset │ ├── ImageSets │ │ ├── test.txt │ │ ├── train.txt │ │ ├── trainval.txt │ │ └── val.txt │ ├── __init__.py │ ├── data_aug.py │ ├── dataloader.py │ ├── demo_data │ │ ├── test │ │ │ ├── 000002.bin │ │ │ ├── 000002.png │ │ │ └── 000002.txt │ │ └── val │ │ │ ├── 000134.bin │ │ │ ├── 000134.png │ │ │ ├── 000134.txt │ │ │ └── 000134_gt.txt │ └── kitti.py ├── loss │ ├── __init__.py │ └── loss.py ├── model │ ├── __init__.py │ ├── anchors.py │ └── pointpillars.py ├── ops │ ├── __init__.py │ ├── iou3d │ │ ├── iou3d.cpp │ │ └── iou3d_kernel.cu │ ├── iou3d_module.py │ ├── voxel_module.py │ └── voxelization │ │ ├── voxelization.cpp │ │ ├── voxelization.h │ │ ├── voxelization_cpu.cpp │ │ └── voxelization_cuda.cu └── utils │ ├── __init__.py │ ├── io.py │ ├── process.py │ ├── viewpoint.json │ └── vis_o3d.py ├── pre_process_kitti.py ├── pretrained ├── epoch_160.pth └── summary │ └── events.out.tfevents.1650798663.VM-1-6-ubuntu.6731.0 ├── requirements.txt ├── setup.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/README.md -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/evaluate.py -------------------------------------------------------------------------------- /figures/img_3dbbox_000134.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/figures/img_3dbbox_000134.png -------------------------------------------------------------------------------- /figures/pc_pred_000134.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/figures/pc_pred_000134.png -------------------------------------------------------------------------------- /figures/pytorch_trt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/figures/pytorch_trt.png -------------------------------------------------------------------------------- /misc/log.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/misc/log.md -------------------------------------------------------------------------------- /misc/vis_data_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/misc/vis_data_gt.py -------------------------------------------------------------------------------- /pointpillars/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pointpillars/dataset/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/ImageSets/test.txt -------------------------------------------------------------------------------- /pointpillars/dataset/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/ImageSets/train.txt -------------------------------------------------------------------------------- /pointpillars/dataset/ImageSets/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/ImageSets/trainval.txt -------------------------------------------------------------------------------- /pointpillars/dataset/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/ImageSets/val.txt -------------------------------------------------------------------------------- /pointpillars/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/__init__.py -------------------------------------------------------------------------------- /pointpillars/dataset/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/data_aug.py -------------------------------------------------------------------------------- /pointpillars/dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/dataloader.py -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/test/000002.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/test/000002.bin -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/test/000002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/test/000002.png -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/test/000002.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/test/000002.txt -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/val/000134.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/val/000134.bin -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/val/000134.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/val/000134.png -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/val/000134.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/val/000134.txt -------------------------------------------------------------------------------- /pointpillars/dataset/demo_data/val/000134_gt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/demo_data/val/000134_gt.txt -------------------------------------------------------------------------------- /pointpillars/dataset/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/dataset/kitti.py -------------------------------------------------------------------------------- /pointpillars/loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .loss import Loss -------------------------------------------------------------------------------- /pointpillars/loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/loss/loss.py -------------------------------------------------------------------------------- /pointpillars/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/model/__init__.py -------------------------------------------------------------------------------- /pointpillars/model/anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/model/anchors.py -------------------------------------------------------------------------------- /pointpillars/model/pointpillars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/model/pointpillars.py -------------------------------------------------------------------------------- /pointpillars/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/__init__.py -------------------------------------------------------------------------------- /pointpillars/ops/iou3d/iou3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/iou3d/iou3d.cpp -------------------------------------------------------------------------------- /pointpillars/ops/iou3d/iou3d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/iou3d/iou3d_kernel.cu -------------------------------------------------------------------------------- /pointpillars/ops/iou3d_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/iou3d_module.py -------------------------------------------------------------------------------- /pointpillars/ops/voxel_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/voxel_module.py -------------------------------------------------------------------------------- /pointpillars/ops/voxelization/voxelization.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/voxelization/voxelization.cpp -------------------------------------------------------------------------------- /pointpillars/ops/voxelization/voxelization.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/voxelization/voxelization.h -------------------------------------------------------------------------------- /pointpillars/ops/voxelization/voxelization_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/voxelization/voxelization_cpu.cpp -------------------------------------------------------------------------------- /pointpillars/ops/voxelization/voxelization_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/ops/voxelization/voxelization_cuda.cu -------------------------------------------------------------------------------- /pointpillars/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/utils/__init__.py -------------------------------------------------------------------------------- /pointpillars/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/utils/io.py -------------------------------------------------------------------------------- /pointpillars/utils/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/utils/process.py -------------------------------------------------------------------------------- /pointpillars/utils/viewpoint.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/utils/viewpoint.json -------------------------------------------------------------------------------- /pointpillars/utils/vis_o3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pointpillars/utils/vis_o3d.py -------------------------------------------------------------------------------- /pre_process_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pre_process_kitti.py -------------------------------------------------------------------------------- /pretrained/epoch_160.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pretrained/epoch_160.pth -------------------------------------------------------------------------------- /pretrained/summary/events.out.tfevents.1650798663.VM-1-6-ubuntu.6731.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/pretrained/summary/events.out.tfevents.1650798663.VM-1-6-ubuntu.6731.0 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PointPillars/HEAD/train.py --------------------------------------------------------------------------------