--- license: apache-2.0 tags: - forensic - surveillance - object-detection - face-detection - person-reid - vehicle-reid - vigia language: - pt --- # VigIA — Model Weights Centralized model weights for [VigIA](https://github.com/vigia-project/vigia) forensic video analysis system. ## Why This Repo Exists VigIA uses multiple ML models from different sources (Google Drive, PyTorch Hub, insightface). Many of these sources are unreliable for automated downloads (captchas, rate limits, broken URLs). This repo consolidates all weights in one CDN-backed location for reliable deployment. ## Models Included | Subfolder | Model | Task | Original Source | License | |-----------|-------|------|-----------------|---------| | `osnet-ain/` | OSNet-AIN x1.0 | Person Re-ID | torchreid (Google Drive) | MIT | | `solider-swin-s/` | SOLIDER Swin-S | Person Re-ID backbone | tinyvision (Google Drive) | Apache 2.0 | | `solider-swin-b/` | SOLIDER Swin-B | Person Re-ID backbone | tinyvision (Google Drive) | Apache 2.0 | | `transreid-vehicle/` | TransReID ViT | Vehicle Re-ID | damo-cv (Google Drive) | MIT | | `resnet50-ibn/` | ResNet50-IBN | Vehicle Re-ID backbone | torchvision (PyTorch Hub) | BSD-3 | | `scrfd-2.5g/` | SCRFD-2.5GF | Face Detection | insightface (buffalo_l.zip) | MIT | | `scrfd-10g/` | SCRFD-10GF | Face Detection | insightface (buffalo_l.zip) | MIT | | `scrfd-34g/` | SCRFD-34GF | Face Detection | insightface (OneDrive) | MIT | ## Usage ```python from huggingface_hub import snapshot_download # Download a specific model subfolder snapshot_download( repo_id='rotivrotiv/model-weights', allow_patterns='solider-swin-b/**', local_dir='./models/', ) ``` ## License Each model retains its original license (see table above). All models are commercially compatible (Apache 2.0, MIT, BSD-3). No model in this repo uses GPL, AGPL, or non-commercial licenses. ## Credits - [SOLIDER](https://github.com/tinyvision/SOLIDER-REID) — tinyvision, ICCV 2023 - [TransReID](https://github.com/damo-cv/TransReID) — DAMO Academy, ICCV 2021 - [OSNet](https://github.com/KaiyangZhou/deep-person-reid) — torchreid, MIT - [SCRFD](https://github.com/deepinsight/insightface) — insightface, MIT - [ResNet50](https://pytorch.org/vision/stable/models.html) — torchvision, BSD-3