Add AGO nuScenes checkpoints

README.md

@@ -0,0 +1,50 @@
+---
+license: other
+license_name: nuscenes-terms-of-use
+license_link: https://www.nuscenes.org/terms-of-use
+tags:
+  - occupancy-prediction
+  - open-vocabulary
+  - autonomous-driving
+  - nuscenes
+---
+
+# AGO: Adaptive Grounding for Open World 3D Occupancy Prediction
+
+Pretrained weights for [AGO](https://github.com/EdwardLeeLPZ/AGO) (ICCV 2025) on Occ3D-nuScenes. Paper: [arXiv:2504.10117](https://arxiv.org/abs/2504.10117).
+
+## Weights
+
+Each file contains the model `state_dict`. The CLIP weights are not included and are loaded from FeatUp at runtime.
+
+| File | Config | Known mIoU / Unknown mIoU / mIoU |
+| :--- | :--- | :-: |
+| `ago_nuscenes_self_supervised.pth` | `ago_nuscenes_self_supervised.py` | - / - / 19.16 |
+| `ago_nuscenes_open_world_pretraining.pth` | `ago_nuscenes_open_world_pretraining.py` | 23.23 / 0.00 / 11.61 |
+| `ago_nuscenes_open_world_pretraining.pth` (zero-shot) | `ago_nuscenes_open_world_pretraining.py` | 33.95 / 0.42 / 7.13 |
+| `ago_nuscenes_open_world_finetuning.pth` | `ago_nuscenes_open_world_finetuning.py` | 36.88 / 8.78 / 14.40 |
+
+## Usage
+
+```shell
+hf download EdwardLeeLPZ/AGO --include "*.pth" --local-dir checkpoints
+bash tools/dist_test.sh projects/configs/ago_nuscenes_self_supervised.py checkpoints/ago_nuscenes_self_supervised.pth 8 --tag self_supervised
+```
+
+See the [code repository](https://github.com/EdwardLeeLPZ/AGO) for installation and data preparation.
+
+## License
+
+The weights are trained on nuScenes and subject to the [nuScenes terms of use](https://www.nuscenes.org/terms-of-use).
+
+## Citation
+
+```
+@inproceedings{li2025ago,
+  title={AGO: Adaptive Grounding for Open World 3D Occupancy Prediction},
+  author={Li, Peizheng and Ding, Shuxiao and Zhou, You and Zhang, Qingwen and Inak, Onat and Triess, Larissa and Hanselmann, Niklas and Cordts, Marius and Zell, Andreas},
+  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
+  pages={8645--8655},
+  year={2025}
+}
+```

SHA256SUMS

@@ -0,0 +1,3 @@
+110173bd60be38c2c192842d7faa5997cea8e51b78223e1bd55f02a24e1d9675  ago_nuscenes_open_world_finetuning.pth
+493906ad5690ce87792a4ecf26d492a384bc321e7b784a7633336ebb4e4b389c  ago_nuscenes_open_world_pretraining.pth
+968b8f6e8eeae8716a31f08b7ebb7a06c128fb41d1cf774175da11b599f4c1a0  ago_nuscenes_self_supervised.pth

ago_nuscenes_open_world_finetuning.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:110173bd60be38c2c192842d7faa5997cea8e51b78223e1bd55f02a24e1d9675
+size 272927269

ago_nuscenes_open_world_pretraining.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:493906ad5690ce87792a4ecf26d492a384bc321e7b784a7633336ebb4e4b389c
+size 272928200

ago_nuscenes_self_supervised.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:968b8f6e8eeae8716a31f08b7ebb7a06c128fb41d1cf774175da11b599f4c1a0
+size 272921683