/ Модель

Характеристики

Библиотека
timm
Опубликована
25 апр 2024

vit_tiny_patch8_112.adaface_ms1mv3

Эмбеддинги изображенийИзображение → embedding3 профессии

Vision Transformer, обученный на MS1MV3, извлекает 512-мерные векторы признаков из изображений лиц для задач распознавания и сопоставления.

359скачиваний за 30 дней2лайкавеса обновлены25 апр 2024

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

gaunernst/vit_tiny_patch8_112.adaface_ms1mv3 — модель Vision Transformer для распознавания лиц, обученная на наборе данных MS1MV3. Она совместима с библиотекой timm и выдаёт для изображения лица вектор признаков размерности 512; выходной вектор не нормализован, поэтому при необходимости его следует нормализовать отдельно. Модель предназначена для извлечения признаков лиц, которые можно использовать для сравнения и сопоставления изображений лиц.

Модель принимает изображение размером 112×112 пикселей и формирует эмбеддинг. В метаданных указаны задача извлечения признаков изображений и формат весов safetensors. Лицензия в предоставленных данных не обозначена.

Сценарии применения

  • извлечение эмбеддингов лиц из фотографий
  • сравнение двух изображений лиц по сходству
  • поиск совпадений лица в коллекции изображений

Ограничения

Выходные эмбеддинги не нормализованы: для сравнения с косинусной метрикой их следует нормализовать. В предоставленных данных условия лицензии не указаны.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Model card for gaunernst/vit_tiny_patch8_112.adaface_ms1mv3

A Vision Transformer (ViT) 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

import timm
import torch.nn.functional as F

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

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · timm · image-feature-extraction

Можно запустить

Подбираем GPU…

Ревизия весов
8f38db3

0 моделей