Commit 98281d05 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

add pretrained model links and performance

parent c8e80b0c
......@@ -7,6 +7,8 @@ As of now, it mainly consists of `PCDet` toolbox for 3D object detection from po
### What does `PCDet` toolbox do?
Note that we have upgrated `PCDet` from `v0.1` to `v0.2` with prety new structures to support various datasets and models.
`PCDet` is a general PyTorch-based codebase for 3D object detection from point cloud.
It currently supports multiple state-of-the-art 3D object detection methods with highly refactored codes for both one-stage and two-stage 3D detection frameworks.
......@@ -32,13 +34,14 @@ Contributions are also welcomed.
Selected supported methods are shown in the below table. The results are the 3D detection performance of car class on the *val* set of KITTI dataset.
All models are trained with 8 GPUs and are available for download.
| | Batch Size | AP_Easy | AP_Mod. | AP_Hard | download |
| | Batch Size | AP_Easy | **AP_Moderate** | AP_Hard | download |
|---------------------------------------------|:----------:|:-------:|:-------:|:-------:|:---------:|
| [PointPillar](tools/cfgs/kitti_models/pointpillar.yaml) | 32 | - | - | - | [model]() |
| [SECOND](tools/cfgs/kitti_models/second.yaml) | 32 | - | - | - | [model]() |
| [Part-A^2](tools/cfgs/kitti_models/PartA2.yaml) | 32 | - | - | - | [model]() |
| [PV-RCNN](tools/cfgs/kitti_models/pv_rcnn.yaml) | 16 | - | - | - | [model]() |
| [PointPillar](tools/cfgs/kitti_models/pointpillar.yaml) | 32 | 86.46 | 77.28 | 74.65 | [model-18M](https://drive.google.com/file/d/1wMxWTpU1qUoY3DsCH31WJmvJxcjFXKlm/view?usp=sharing) |
| [SECOND](tools/cfgs/kitti_models/second.yaml) | 32 | 88.61 | 78.62| 77.22 | [model-20M](https://drive.google.com/file/d/1-01zsPOsqanZQqIIyy7FpNXStL3y4jdR/view?usp=sharing) |
| [Part-A^2](tools/cfgs/kitti_models/PartA2.yaml) | 32 | 89.55 | 79.40 | 78.84 | [model-244M](https://drive.google.com/file/d/10GK1aCkLqxGNeX3lVu8cLZyE0G8002hY/view?usp=sharing) |
| [PV-RCNN](tools/cfgs/kitti_models/pv_rcnn.yaml) | 16 | 89.34 | 83.69 | 78.70 | [model-50M](https://drive.google.com/file/d/1lIOq4Hxr0W3qsX83ilQv0nk1Cls6KAr-/view?usp=sharing) |
| [SECOND-MultiHead](tools/cfgs/kitti_models/second_multihead.yaml) | 32 | - | - | - | - |
| PointRCNN | - | - | - | - | Pending|
## Installation
......
......@@ -9,7 +9,9 @@ All the codes are tested in the following environment:
* `spconv v1.0` ([commit 8da6f96](https://github.com/traveller59/spconv/tree/8da6f967fb9a054d8870c3515b1b44eca2103634))
### Install `pcdet`
### Install `pcdet v0.2`
NOTE: Please re-install `pcdet v0.2` by running `python setup.py develop` if you have already installed `pcdet v0.1` previously.
a. Clone this repository.
```shell
git clone https://github.com/open-mmlab/OpenLidarPercept.git
......@@ -36,6 +38,7 @@ Currently we provide the dataloader of KITTI dataset, and the supporting of more
### KITTI Dataset
* Please download the official [KITTI 3D object detection](http://www.cvlibs.net/datasets/kitti/eval_object.php?obj_benchmark=3d) dataset and organize the downloaded files as follows (the road planes could be downloaded from [[road plane]](https://drive.google.com/file/d/1d5mq0RXRnvHPVeKx6Q612z0YRO1t2wAp/view?usp=sharing), which are optional for data augmentation in the training):
* NOTE: You need to create the infos and gt database again even you already have them from `pcdet v0.1`
```
PCDet
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment