File size: 644 Bytes
7748717 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
---
datasets:
- gaunernst/ms1mv3-recordio
library_name: timm
tags:
- image-feature-extraction
- timm
---
# Model card for gaunernst/convnext_atto.cosface_ms1mv3
ConvNeXt-Atto 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
```python
import timm
import torch.nn.functional as F
model = timm.create_model("hf_hub:gaunernst/convnext_atto.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
```
|