Определение типа галактик по изображениям

О сервисе

Что умеет эта модель? Данная модель классифицирует изображения галактик по трем распространенным типам: эллиптические (E), спиральные (S) и линзовидные спиральные (Sb). Достигнута взвешенная точность в 78%. Как обучена модель? Модель основана на архитектуре ViT (Vision Transformer) с базовой моделью google/vit-base-patch16-224-in21k . Она была дообучена на наборе данных, содержащем изображения галактик различных типов. Основные параметры конфигурации: Архитектура: ViTForImageClassification Размер изображения: 224x224 пикселя Размер патча: 16x16 пикселей Количество слоев: 12 Количество заголовков внимания: 12 Размер скрытого слоя: 768 Функция активации: GELU Отчет о классификации: precision recall f1-score support E 0.7656 0.8848 0.8209 13592 S 0.7526 0.6685 0.7081 13591 SB 0.8262 0.7900 0.8077 13591 accuracy 0.7811 40774 macro avg 0.7815 0.7811 0.7789 40774 weighted avg 0.7815 0.7811 0.7789 40774 Применение Модель может быть использована для автоматической каталогизации галактик в астрономических исследованиях, помогая ученым быстро сортировать и анализировать большие объемы изображений. Также может служить образовательным инструментом для демонстрации машинного обучения в астрономии. from transformers import pipeline classifier = pipeline("image-classification", model="dima806/galaxy_type_image_detection") # classifier("путь_к_вашему_изображению.jpg") Ссылки dima806/galaxy_type_image_detection Источник: https://qubu.ai/models/opredelenie-tipa-galaktik-po-izobrazheniyam License: MIT