File size: 3,282 Bytes
08ed2ee dcaf240 08ed2ee dcaf240 08ed2ee dcaf240 08ed2ee dcaf240 08ed2ee 22494c7 08ed2ee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | ---
license: mit
tags:
- autonomous-driving
- hd-map
- vectorized-map
- nuscenes
- mmdetection3d
datasets:
- nuscenes
---
# MapGR: MapQR + MapGR checkpoint (nuScenes, ResNet-50, 24 epochs)
This repository hosts the released weights for **Learning Global Representation from Queries for Vectorized HD Map Construction** (ICML 2026, [arXiv:2510.06969](https://arxiv.org/abs/2510.06969), [project page](https://xr-lee.github.io/mapgr-demo/)).
Training and evaluation code, installation and data preparation are in the GitHub repository [XR-Lee/MapGR](https://github.com/XR-Lee/MapGR) (`mapqr/` sub-project).
| File | Description |
|---|---|
| `mapqr_mapgr_nusc_r50_24ep.pth` | Model weights: `state_dict` plus minimal `meta`, no optimizer state. |
| `mapqr_mapgr_nusc_r50_24ep_config.py` | Fully resolved config, for reference. Train and evaluate with `mapqr/projects/configs/mapgr/mapqr_mapgr_nusc_r50_24ep.py` from the code repository. |
## Results (nuScenes val)
| | mAP<sub>1</sub> | mAP<sub>2</sub> | AP<sub>div</sub> / AP<sub>ped</sub> / AP<sub>bou</sub> (mAP<sub>2</sub> thresholds) |
|---|---|---|---|
| Paper (MapQR + MapGR, 24 ep) | 45.3 | 68.1 | 69.8 / 65.8 / 68.8 |
| This checkpoint | 44.87 | 67.96 | 69.35 / 64.95 / 69.57 |
mAP<sub>2</sub> averages Chamfer AP over the {0.5, 1.0, 1.5} m thresholds; mAP<sub>1</sub> averages over {0.2, 0.5, 1.0} m.
This checkpoint is not the run reported in the paper.
- It was trained with the research code the GitHub repository was cleaned up from, using the same settings as the released config.
- Training: 8 GPUs × 4 samples, fp16.
- The numbers above come from evaluating this exact file with the released `tools/test.py`.
- Other 24-epoch runs of closely related configurations (different seeds, loss weights and query counts) reached 67.2–68.0 mAP<sub>2</sub>.
```shell
# inside mapqr/ of the code repository
wget -P ckpts https://huggingface.co/Lee1413/MapGR/resolve/main/mapqr_mapgr_nusc_r50_24ep.pth
./tools/dist_test_map.sh projects/configs/mapgr/mapqr_mapgr_nusc_r50_24ep.py ckpts/mapqr_mapgr_nusc_r50_24ep.pth 1
```
SHA-256 of `mapqr_mapgr_nusc_r50_24ep.pth`: `d4f868312234806b5c0a97ea2c7edf71173de3fac1d510742a99317e1c6f6916`
## Training logs
mmdet3d logs of the two 24-epoch nuScenes runs. User names, host names and internal path prefixes are replaced with `<user>`, `<host>` and `<workspace>`. Nothing else is changed.
| Run | Files | Setup | mAP<sub>2</sub> at epoch 12 / 24 |
|---|---|---|---|
| MapQR + MapGR (this checkpoint) | [`20260329_014512.log`](logs/mapqr_mapgr_nusc_r50_24ep/20260329_014512.log), [`.log.json`](logs/mapqr_mapgr_nusc_r50_24ep/20260329_014512.log.json) | 8 GPUs × 4 samples, fp16, PyTorch 2.0.1 | 59.90 / 67.96 |
| MapTRv2 + MapGR (the paper's run; weights not released) | [`train.273710.log`](logs/maptrv2_mapgr_nusc_r50_24ep/train.273710.log) | 4 GPUs × 4 samples, fp16, PyTorch 1.9.1 | 56.14 / 64.98 |
## Citation
```bibtex
@inproceedings{qiu2026mapgr,
title = {Learning Global Representation from Queries for Vectorized {HD} Map Construction},
author = {Qiu, Shoumeng and Li, Xinrun and Long, Yang and Xue, Xiangyang and Ojha, Varun and Pu, Jian},
booktitle = {International Conference on Machine Learning (ICML)},
year = {2026}
}
```
|