Skip to content

Commit

Permalink
add pretrained model links and performance
Browse files Browse the repository at this point in the history
  • Loading branch information
sshaoshuai committed Jun 24, 2020
1 parent c8e80b0 commit 98281d0
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 7 deletions.
13 changes: 8 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.

Expand All @@ -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

Expand Down
7 changes: 5 additions & 2 deletions docs/INSTALL.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -35,7 +37,8 @@ python setup.py develop
Currently we provide the dataloader of KITTI dataset, and the supporting of more datasets are on the way.

### 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):
* 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
Expand Down

0 comments on commit 98281d0

Please sign in to comment.