Correct training provenance, add code link, config with loss-weight keys
Browse files- README.md +4 -3
- mapqr_mapgr_nusc_r50_24ep_config.py +4 -1
README.md
CHANGED
|
@@ -14,7 +14,7 @@ datasets:
|
|
| 14 |
|
| 15 |
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/)).
|
| 16 |
|
| 17 |
-
Training and evaluation code, installation and data preparation are in the GitHub repository (`mapqr/` sub-project).
|
| 18 |
|
| 19 |
| File | Description |
|
| 20 |
|---|---|
|
|
@@ -30,11 +30,12 @@ Training and evaluation code, installation and data preparation are in the GitHu
|
|
| 30 |
|
| 31 |
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.
|
| 32 |
|
| 33 |
-
This checkpoint
|
| 34 |
|
|
|
|
| 35 |
- Training: 8 GPUs × 4 samples, fp16.
|
| 36 |
- The numbers above come from evaluating this exact file with the released `tools/test.py`.
|
| 37 |
-
-
|
| 38 |
|
| 39 |
```shell
|
| 40 |
# inside mapqr/ of the code repository
|
|
|
|
| 14 |
|
| 15 |
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/)).
|
| 16 |
|
| 17 |
+
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).
|
| 18 |
|
| 19 |
| File | Description |
|
| 20 |
|---|---|
|
|
|
|
| 30 |
|
| 31 |
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.
|
| 32 |
|
| 33 |
+
This checkpoint is not the run reported in the paper.
|
| 34 |
|
| 35 |
+
- It was trained with the research code the GitHub repository was cleaned up from, using the same settings as the released config.
|
| 36 |
- Training: 8 GPUs × 4 samples, fp16.
|
| 37 |
- The numbers above come from evaluating this exact file with the released `tools/test.py`.
|
| 38 |
+
- Other 24-epoch runs of closely related configurations (different seeds, loss weights and query counts) reached 67.2–68.0 mAP<sub>2</sub>.
|
| 39 |
|
| 40 |
```shell
|
| 41 |
# inside mapqr/ of the code repository
|
mapqr_mapgr_nusc_r50_24ep_config.py
CHANGED
|
@@ -452,7 +452,10 @@ model = dict(
|
|
| 452 |
loss_pts=dict(type='PtsL1Loss', loss_weight=5.0),
|
| 453 |
loss_dir=dict(type='PtsDirCosLoss', loss_weight=0.005),
|
| 454 |
loss_seg=dict(type='SimpleLoss', pos_weight=4.0, loss_weight=1.0),
|
| 455 |
-
loss_pv_seg=dict(type='SimpleLoss', pos_weight=1.0, loss_weight=2.0)
|
|
|
|
|
|
|
|
|
|
| 456 |
train_cfg=dict(
|
| 457 |
pts=dict(
|
| 458 |
grid_size=[512, 512, 1],
|
|
|
|
| 452 |
loss_pts=dict(type='PtsL1Loss', loss_weight=5.0),
|
| 453 |
loss_dir=dict(type='PtsDirCosLoss', loss_weight=0.005),
|
| 454 |
loss_seg=dict(type='SimpleLoss', pos_weight=4.0, loss_weight=1.0),
|
| 455 |
+
loss_pv_seg=dict(type='SimpleLoss', pos_weight=1.0, loss_weight=2.0),
|
| 456 |
+
aux_loss_weights=[0.05, 0.05, 1.0, 1.0, 1.0],
|
| 457 |
+
aux_loss_weights_one2many=[1.0, 1.0, 1.0, 1.0, 1.0],
|
| 458 |
+
global_map_loss_weights=[1.0, 1.0, 0.0, 0.0, 0.0, 0.0]),
|
| 459 |
train_cfg=dict(
|
| 460 |
pts=dict(
|
| 461 |
grid_size=[512, 512, 1],
|