Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/yuantianyuan01/StreamMapNet
https://github.com/yuantianyuan01/StreamMapNet
Last synced: 2 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/yuantianyuan01/StreamMapNet
- Owner: yuantianyuan01
- License: gpl-3.0
- Created: 2023-08-23T11:19:03.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-03-03T15:34:04.000Z (11 months ago)
- Last Synced: 2024-08-03T09:07:40.885Z (5 months ago)
- Language: Python
- Size: 426 KB
- Stars: 179
- Watchers: 9
- Forks: 15
- Open Issues: 19
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- awesome-hd-map-construction - [code
README
StreamMapNet
[WACV 2024] StreamMapNet: Streaming Mapping Network for Vectorized Online HD Map Construction
[](https://arxiv.org/abs/2308.12570)
## Introduction
This repository is an official implementation of StreamMapNet.
## Getting Started
### 1. Environment
**Step 1.** Create conda environment and activate it.
```
conda create --name streammapnet python=3.8 -y
conda activate streammapnet
```
**Step 2.** Install PyTorch.
```
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
```
**Step 3.** Install MMCV series.
```
# Install mmcv-series
pip install mmcv-full==1.6.0
pip install mmdet==2.28.2
pip install mmsegmentation==0.30.0
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
git checkout v1.0.0rc6
pip install -e .
```
**Step 4.** Install other requirements.
```
pip install -r requirements.txt
```
### 2. Data Preparation
**Step 1.** Download [NuScenes](https://www.nuscenes.org/download) dataset to `./datasets/nuScenes`.
**Step 2.** Download [Argoverse2 (sensor)](https://argoverse.github.io/user-guide/getting_started.html#download-the-datasets) dataset to `./datasets/av2`.
**Step 3.** Generate annotation files for NuScenes dataset.
```
python tools/nuscenes_converter.py --data-root ./datasets/nuScenes --newsplit
```
**Step 4.** Generate annotation files for Argoverse2 dataset.
```
python tools/argoverse_converter.py --data-root ./datasets/av2 --newsplit
```
### 3. Training and Validating
To train a model with 8 GPUs:
```
bash tools/dist_train.sh ${CONFIG} 8
```
To validate a model with 8 GPUs:
```
bash tools/dist_test.sh ${CONFIG} ${CEHCKPOINT} 8 --eval
```
To test a model's inference speed:
```
python tools/benchmark.py ${CONFIG} ${CEHCKPOINT}
```
## Results
### Results on Argoverse2 newsplit
| Range | $\mathrm{AP}_{ped}$ | $\mathrm{AP}_{div}$| $\mathrm{AP}_{bound}$ | $\mathrm{AP}$ | Config | Epoch | Checkpoint |
| :---: | :---: | :---: | :---: |:---:|:---: |:---: | :---: |
| $60\times 30\ m$ | 57.9 | 55.7| 61.3| 58.3| [Config](./plugin/configs/av2_newsplit_608_60x30_30e.py) | 30 | [ckpt](https://drive.google.com/file/d/1p6PZDGbVoxedU0YqEbvSBjCMkcTx91ld/view?usp=share_link)|
| $100\times 50\ m$ |60.0 | 45.9 | 48.9 | 51.6 | [Config](./plugin/configs/av2_newsplit_608_100x0_30e.py5) |30 | [ckpt](https://drive.google.com/file/d/1PkOiGFLGyQ7GUljeRS7REQS6Cv_pV1qx/view?usp=share_link)|
### Results on NuScenes newsplit
| Range | $\mathrm{AP}_{ped}$ | $\mathrm{AP}_{div}$| $\mathrm{AP}_{bound}$ | $\mathrm{AP}$ | Config | Epoch | Checkpoint |
| :---: | :---: | :---: | :---: |:---:|:---: |:---: | :---: |
| $60\times 30\ m$ | 32.2 | 29.3 | 40.8 | 34.1 | [Config](./plugin/configs/nusc_newsplit_480_60x30_24e.py) | 24| [ckpt](https://drive.google.com/file/d/1L9IRkd_Sg_hPu8SSagWBEZahUD_dvMeG/view?usp=share_link)|
| $100\times 50\ m$ | 25.6 | 17.4 | 24.3 | 22.4 | [Config](./plugin/configs/nusc_newsplit_480_100x50_24e.py)| 24 | [ckpt](https://drive.google.com/file/d/1nB4r108-rj87Ain7s8HHEo5hXvxZMMre/view?usp=share_link)|
### Results on NuScenes oldsplit
| Range | $\mathrm{AP}_{ped}$ | $\mathrm{AP}_{div}$| $\mathrm{AP}_{bound}$ | $\mathrm{AP}$ | Config | Epoch | Checkpoint |
| :---: | :---: | :---: | :---: |:---:|:---:|:---: | :---: |
| $60\times 30\ m$ | 61.7| 66.3 | 62.1 | 63.4 | [Config](./plugin/configs/nusc_baseline_480_60x30_30e.py) | 30| [ckpt](https://drive.google.com/file/d/1-n6DGu23KkSO8PFfJ01ofmtUed0zOMZ_/view?usp=share_link)|
## Citation
If you find our paper or codebase useful in your research, please give us a star and cite our paper.
```
@InProceedings{Yuan_2024_streammapnet,
author = {Yuan, Tianyuan and Liu, Yicheng and Wang, Yue and Wang, Yilun and Zhao, Hang},
title = {StreamMapNet: Streaming Mapping Network for Vectorized Online HD Map Construction},
booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
month = {January},
year = {2024},
pages = {7356-7365}
}
```