Convnext Nano.cosface Ms1mv3

🚀 gaunernst/convnext_nano.cosface_ms1mv3

ConvNeXt-Nano 是用於人臉識別的模型，在 MS1MV3 數據集上進行訓練。該模型使用此倉庫進行訓練：https://github.com/gau-nernst/timm-face，並且與 timm 庫完全兼容。

🚀 快速開始

ConvNeXt-Nano 專為人臉識別任務設計，在 MS1MV3 數據集上進行了訓練。它藉助 timm 庫，能夠方便地在項目中使用。

✨ 主要特性

• 數據集：使用 gaunernst/ms1mv3-recordio 數據集進行訓練。
• 庫兼容性：與 timm 庫完全兼容，便於集成到現有項目中。
• 應用場景：適用於圖像特徵提取任務。

💻 使用示例

基礎用法

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