Downloads · 30 days
374
7% of all-time downloads
vit_small_patch8_gap_112.cosface_ms1mv3
gaunernst/vit_small_patch8_gap_112.cosface_ms1mv3
vit_small_patch8_gap_112.cosface_ms1mv3 is a image feature extraction model from gaunernst. Use it for the image feature extraction task on the model card, and read the license before you ship it in a product. It is set up for timm.
A Vision Transformer (ViT) for face recognition, trained on MS1MV3 dataset. The model was trained using this repo: https://github.com/gau-nernst/timm-face. It is fully compatible with timm.
All-time downloads
5.1K
Public
Parameters
21.6M
86.6 MB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.safetensors86.6 MB · 100%
At a glance
- Task
- Image Feature Extraction
- Library
- timm
- Access
- Public
- Created
- Apr 29, 2024
- Updated
- Apr 29, 2024
- SHA
- c8f63e46
Datasets
Model card
From the Hugging Face model README
Model card for gaunernst/vit_small_patch8_gap_112.cosface_ms1mv3
A Vision Transformer (ViT) for face recognition, trained on MS1MV3 dataset. The model was trained using this repo: https://github.com/gau-nernst/timm-face. It is fully compatible with timm.
Usage
import timm
import torch.nn.functional as F
model = timm.create_model("hf_hub:gaunernst/vit_small_patch8_gap_112.cosface_ms1mv3", pretrained=True).eval()
embs = model(torch.randn(1, 3, 112, 112)) # output shape (1, 512)
embs = F.normalize(embs, dim=1) # model output is not normalized
- Task
- Image Feature Extraction
- Library
- timm
- Created
- Apr 29, 2024
- Updated
- Apr 29, 2024