/ Модель

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

Параметры
113 млн
Лицензия
Apache 2.0
Архитектура
vit · dense
Базовая модель
vit-base-patch16-224-in21k
Опубликована
24 фев 2024

skin_types_image_detection

Классификация изображенийИзображение → label3 профессии

Дообученная модель ViT классифицирует тип кожи на фотографии лица как сухой, нормальный или жирный.

50 236скачиваний за 30 дней17лайковвеса обновлены19 окт 2024

Бенчмарки

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

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

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

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

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

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

Модель dima806/skin_types_image_detection разработана на основе Vision Transformer google/vit-base-patch16-224-in21k и дообучена для классификации типов кожи по фотографии лица. Она относит изображение к одному из трёх классов: сухая, нормальная или жирная кожа; архитектура ViTForImageClassification содержит 113,2 млн параметров.

Модель предназначена для автоматической предварительной оценки типа кожи по портретному изображению, например, как часть приложения или сервиса по уходу за кожей. В отчёте на 1529 примерах указана точность 65,34%; качество различается между классами, поэтому результат следует считать ориентировочным. Модель принимает изображения и выдаёт метку класса.

Репозиторий распространяется по лицензии Apache-2.0. В README не приведены сведения о происхождении и репрезентативности данных, поэтому точность на других типах изображений и условиях съёмки может отличаться.

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

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

Ограничения

В отчёте README точность составляет 65,34%, поэтому предсказания могут быть ошибочными и не заменяют профессиональную оценку. Репрезентативность данных и устойчивость к различным условиям съёмки не описаны.

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

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

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

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

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

Открыть на Hugging Face

Detects the skin type (dry, normal, oily) based on facial image.

See https://www.kaggle.com/code/dima806/skin-types-image-detection-vit for details.

image/png

Classification report:

              precision    recall  f1-score   support

         dry     0.6829    0.6346    0.6578       509
      normal     0.6414    0.6314    0.6364       510
        oily     0.6390    0.6941    0.6654       510

    accuracy                         0.6534      1529
   macro avg     0.6544    0.6534    0.6532      1529
weighted avg     0.6544    0.6534    0.6532      1529

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
ViTForImageClassification
Ревизия весов
5aace4f

0 моделей