Downloads · 30 days
184
8% of all-time downloads
convnext_nano.cosface_ms1mv3
gaunernst/convnext_nano.cosface_ms1mv3
convnext_nano.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.
ConvNeXt-Nano 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
2.3K
Public
Parameters
15.3M
61.2 MB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.safetensors61.2 MB · 100%
At a glance
- Task
- Image Feature Extraction
- Library
- timm
- Access
- Public
- Created
- Apr 25, 2024
- Updated
- Apr 25, 2024
- SHA
- 5a2926b0
Datasets
Model card
From the Hugging Face model README
Model card for gaunernst/convnext_nano.cosface_ms1mv3
ConvNeXt-Nano 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/convnext_nano.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 25, 2024
- Updated
- Apr 25, 2024