Vit Tiny Patch8 112.cosface Ms1mv3

🚀 gaunernst/vit_tiny_patch8_112.cosface_ms1mv3

顔認識用のVision Transformer (ViT)モデルで、MS1MV3データセットで学習されています。このモデルはtimmと完全に互換性があります。

🚀 クイックスタート

このモデルは、顔認識に特化したVision Transformer (ViT)で、MS1MV3データセットを使用して学習されています。モデルの学習には、このリポジトリが使用されており、timmと完全に互換性があります。

💻 使用例

基本的な使用法

import timm
import torch.nn.functional as F

model = timm.create_model("hf_hub:gaunernst/vit_tiny_patch8_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

📚 詳細情報

データセット

ライブラリ

タグ

• 画像特徴抽出
• timm