docs: sync model card from GitHub 23f150e
Browse files
README.md
CHANGED
|
@@ -3,10 +3,10 @@ license: cc-by-nc-4.0
|
|
| 3 |
library_name: pytorch
|
| 4 |
pipeline_tag: image-feature-extraction
|
| 5 |
tags:
|
| 6 |
-
- person-re-identification
|
| 7 |
-
- aerial-ground
|
| 8 |
-
- vision-transformer
|
| 9 |
-
- neurips2026
|
| 10 |
---
|
| 11 |
|
| 12 |
# 3D-VITAL
|
|
@@ -16,18 +16,18 @@ Cross-View Object Re-Identification** (NeurIPS 2026): [arXiv:2610.16692](https:/
|
|
| 16 |
[poster](https://neurips.cc/virtual/2026/loc/paris/poster/150244).
|
| 17 |
Code, training and evaluation are at <https://github.com/TurtleSmoke/3D-VITAL>.
|
| 18 |
|
| 19 |
-
Each checkpoint is a ViT-B/16 + BNNeck trained on one camera type of AG-ReID, together with its 3D-VITAL synthetic
|
| 20 |
-
views. Retrieval uses the cosine distance on the BNNeck feature, with a single forward pass.
|
| 21 |
-
|
| 22 |
## Checkpoints
|
| 23 |
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
-
|
| 30 |
-
parameters (backbone and BNNeck) and are stored in fp16, so they can differ from the paper by up to 0.1 point.
|
| 31 |
|
| 32 |
## Usage
|
| 33 |
|
|
@@ -37,8 +37,6 @@ hf download TurtleSmoke/3D-VITAL --include "*.pth" --local-dir checkpoints
|
|
| 37 |
bash scripts/eval.sh agreid_aerial $DATA checkpoints/agreid_aerial.pth
|
| 38 |
```
|
| 39 |
|
| 40 |
-
`$DATA` holds the real datasets, see the [GitHub README](https://github.com/TurtleSmoke/3D-VITAL#data).
|
| 41 |
-
|
| 42 |
## License
|
| 43 |
|
| 44 |
The weights are released for non-commercial research under CC BY-NC 4.0. They were trained on
|
|
@@ -50,9 +48,9 @@ respective terms also apply. The code is under the MIT license.
|
|
| 50 |
|
| 51 |
```bibtex
|
| 52 |
@inproceedings{grolleau2026vital,
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
}
|
| 58 |
```
|
|
|
|
| 3 |
library_name: pytorch
|
| 4 |
pipeline_tag: image-feature-extraction
|
| 5 |
tags:
|
| 6 |
+
- person-re-identification
|
| 7 |
+
- aerial-ground
|
| 8 |
+
- vision-transformer
|
| 9 |
+
- neurips2026
|
| 10 |
---
|
| 11 |
|
| 12 |
# 3D-VITAL
|
|
|
|
| 16 |
[poster](https://neurips.cc/virtual/2026/loc/paris/poster/150244).
|
| 17 |
Code, training and evaluation are at <https://github.com/TurtleSmoke/3D-VITAL>.
|
| 18 |
|
|
|
|
|
|
|
|
|
|
| 19 |
## Checkpoints
|
| 20 |
|
| 21 |
+
Each checkpoint is trained on one camera type of AG-ReID, given by its name.
|
| 22 |
+
|
| 23 |
+
| Checkpoint | Query → Gallery | mAP | Rank-1 |
|
| 24 |
+
|---------------------|-----------------|-------|--------|
|
| 25 |
+
| `agreid_aerial.pth` | A → G | 52.09 | 66.20 |
|
| 26 |
+
| | G → A | 53.30 | 62.68 |
|
| 27 |
+
| `agreid_ground.pth` | A → G | 55.99 | 67.13 |
|
| 28 |
+
| | G → A | 57.93 | 68.61 |
|
| 29 |
|
| 30 |
+
A = aerial, G = ground. The checkpoints are stored in fp16, so they can differ slightly from the paper.
|
|
|
|
| 31 |
|
| 32 |
## Usage
|
| 33 |
|
|
|
|
| 37 |
bash scripts/eval.sh agreid_aerial $DATA checkpoints/agreid_aerial.pth
|
| 38 |
```
|
| 39 |
|
|
|
|
|
|
|
| 40 |
## License
|
| 41 |
|
| 42 |
The weights are released for non-commercial research under CC BY-NC 4.0. They were trained on
|
|
|
|
| 48 |
|
| 49 |
```bibtex
|
| 50 |
@inproceedings{grolleau2026vital,
|
| 51 |
+
title = {3D-VITAL: Visibility-aware Identity Training from Artificial Liftings of 3D Reconstruction model for Cross-View Object Re-Identification},
|
| 52 |
+
author = {Grolleau, William and Sabourin, Astrid and Lapouge, Guillaume and Achard, Catherine},
|
| 53 |
+
booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
|
| 54 |
+
year = {2026}
|
| 55 |
}
|
| 56 |
```
|