Skip to content

Commit 4d8624f

Browse files
authored
Include Argoverse2 data preprocessing (open-mmlab#1321)
1 parent a68aaa6 commit 4d8624f

File tree

2 files changed

+279
-24
lines changed

2 files changed

+279
-24
lines changed

docs/GETTING_STARTED.md

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,37 @@ python -m pcdet.datasets.waymo.waymo_dataset --func create_waymo_infos \
108108

109109
Note that you do not need to install `waymo-open-dataset` if you have already processed the data before and do not need to evaluate with official Waymo Metrics.
110110

111+
### Argoverse2 Dataset
112+
* Download the **Argoverse 2 Sensor Dataset** from the [official website](https://www.argoverse.org/av2.html#download-link), and then extract them.
113+
* Install the official API of Argoverse 2
114+
```shell
115+
pip install av2==0.2.0
116+
```
117+
* Generate info files for `train` and `val`.
118+
```python
119+
python -m pcdet.datasets.argo2.argo2_dataset --root_path data/argo2/sensor --output_dir data/argo2
120+
```
121+
- Note that this [issue](https://github.com/argoverse/av2-api/issues/102) from the argo2 api might be noticed.
122+
- If the CPU memory of your machine is limited, you can set `--workers=0` in the training script.
123+
- The organized files are as follows:
124+
```
125+
OpenPCDet
126+
├── data
127+
│ ├── argo2
128+
│ │ │── ImageSets
129+
│ │ │ ├──train.txt & val.txt
130+
│ │ │── training
131+
│ │ │ ├──velodyne
132+
│ │ │── sensor
133+
│ │ │ ├──val
134+
│ │ │── argo2_infos_train.pkl
135+
│ │ │── argo2_infos_val.pkl
136+
│ │ │── val_anno.feather
137+
138+
├── pcdet
139+
├── tools
140+
```
141+
111142

112143
### ONCE Dataset
113144
* Please download train/val/test of the official [ONCE Dataset](https://once-for-auto-driving.github.io/download.html#downloads) and

0 commit comments

Comments
 (0)