TurtleSmoke commited on
Commit
b2159b9
·
verified ·
1 Parent(s): 6a83582

docs: sync model card from GitHub 23f150e

Browse files
Files changed (1) hide show
  1. README.md +17 -19
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
- | File | Setting | Protocol 1 mAP / R1 | Protocol 2 mAP / R1 |
25
- |------|---------|---------------------|---------------------|
26
- | `agreid_aerial.pth` | AG-ReID, aerial training | A→G 52.09 / 66.20 | G→A 53.30 / 62.68 |
27
- | `agreid_ground.pth` | AG-ReID, ground training | A→G 55.99 / 67.13 | G→A 57.93 / 68.61 |
 
 
 
 
28
 
29
- The numbers are those obtained with these files and `scripts/eval.sh`. The checkpoints keep only the inference
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
- title = {3D-VITAL: Visibility-aware Identity Training from Artificial Liftings of 3D Reconstruction model for Cross-View Object Re-Identification},
54
- author = {Grolleau, William and Sabourin, Astrid and Lapouge, Guillaume and Achard, Catherine},
55
- booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
56
- year = {2026}
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
  ```