Convnext Nano.cosface Ms1mv3

🚀 gaunernst/convnext_nano.cosface_ms1mv3

このモデルは、MS1MV3データセットで学習された顔認識用のConvNeXt-Nanoモデルです。このモデルは、https://github.com/gau-nernst/timm-face のリポジトリを使用して学習され、timmと完全に互換性があります。

🚀 クイックスタート

このモデルは、顔認識タスクに使用できるConvNeXt-Nanoモデルです。MS1MV3データセットで学習されており、timmライブラリと互換性があります。

✨ 主な機能

• 顔認識に特化したConvNeXt-Nanoモデル
• MS1MV3データセットで学習
• timmライブラリとの完全互換性

📦 インストール

このモデルを使用するには、timmライブラリが必要です。以下のコマンドでインストールできます。

pip install 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

📚 ドキュメント

• データセット：gaunernst/ms1mv3-recordio
• ライブラリ：timm
• タグ：image-feature-extraction、timm