rotivrotiv commited on
Commit
d4db961
·
verified ·
1 Parent(s): 58423b4

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +63 -3
README.md CHANGED
@@ -1,3 +1,63 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - forensic
5
+ - surveillance
6
+ - object-detection
7
+ - face-detection
8
+ - person-reid
9
+ - vehicle-reid
10
+ - vigia
11
+ language:
12
+ - pt
13
+ ---
14
+
15
+ # VigIA — Model Weights
16
+
17
+ Centralized model weights for [VigIA](https://github.com/vigia-project/vigia) forensic video analysis system.
18
+
19
+ ## Why This Repo Exists
20
+
21
+ VigIA uses multiple ML models from different sources (Google Drive, PyTorch Hub, insightface).
22
+ Many of these sources are unreliable for automated downloads (captchas, rate limits, broken URLs).
23
+ This repo consolidates all weights in one CDN-backed location for reliable deployment.
24
+
25
+ ## Models Included
26
+
27
+ | Subfolder | Model | Task | Original Source | License |
28
+ |-----------|-------|------|-----------------|---------|
29
+ | `osnet-ain/` | OSNet-AIN x1.0 | Person Re-ID | torchreid (Google Drive) | MIT |
30
+ | `solider-swin-s/` | SOLIDER Swin-S | Person Re-ID backbone | tinyvision (Google Drive) | Apache 2.0 |
31
+ | `solider-swin-b/` | SOLIDER Swin-B | Person Re-ID backbone | tinyvision (Google Drive) | Apache 2.0 |
32
+ | `transreid-vehicle/` | TransReID ViT | Vehicle Re-ID | damo-cv (Google Drive) | MIT |
33
+ | `resnet50-ibn/` | ResNet50-IBN | Vehicle Re-ID backbone | torchvision (PyTorch Hub) | BSD-3 |
34
+ | `scrfd-2.5g/` | SCRFD-2.5GF | Face Detection | insightface (buffalo_l.zip) | MIT |
35
+ | `scrfd-10g/` | SCRFD-10GF | Face Detection | insightface (buffalo_l.zip) | MIT |
36
+ | `scrfd-34g/` | SCRFD-34GF | Face Detection | insightface (OneDrive) | MIT |
37
+
38
+ ## Usage
39
+
40
+ ```python
41
+ from huggingface_hub import snapshot_download
42
+
43
+ # Download a specific model subfolder
44
+ snapshot_download(
45
+ repo_id='rotivrotiv/model-weights',
46
+ allow_patterns='solider-swin-b/**',
47
+ local_dir='./models/',
48
+ )
49
+ ```
50
+
51
+ ## License
52
+
53
+ Each model retains its original license (see table above).
54
+ All models are commercially compatible (Apache 2.0, MIT, BSD-3).
55
+ No model in this repo uses GPL, AGPL, or non-commercial licenses.
56
+
57
+ ## Credits
58
+
59
+ - [SOLIDER](https://github.com/tinyvision/SOLIDER-REID) — tinyvision, ICCV 2023
60
+ - [TransReID](https://github.com/damo-cv/TransReID) — DAMO Academy, ICCV 2021
61
+ - [OSNet](https://github.com/KaiyangZhou/deep-person-reid) — torchreid, MIT
62
+ - [SCRFD](https://github.com/deepinsight/insightface) — insightface, MIT
63
+ - [ResNet50](https://pytorch.org/vision/stable/models.html) — torchvision, BSD-3