Vit Tiny Patch8 112.arcface Ms1mv3

🚀 gaunernst/vit_tiny_patch8_112.arcface_ms1mv3

這是一個用於人臉識別的視覺變換器（ViT）模型，在MS1MV3數據集上進行訓練。該模型使用此倉庫進行訓練：https://github.com/gau-nernst/timm-face ，並且與timm庫完全兼容。

🚀 快速開始

此模型是一個用於人臉識別的視覺變換器（ViT），在MS1MV3數據集上進行了訓練。它與timm庫完全兼容，可通過以下代碼使用。

💻 使用示例

基礎用法

import timm
import torch.nn.functional as F

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

📚 詳細文檔