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

Что умеет эта модель?

Данная модель классифицирует изображения галактик по трем распространенным типам: эллиптические (E), спиральные (S) и линзовидные спиральные (Sb). Достигнута взвешенная точность в 78%.

Как обучена модель?

Модель основана на архитектуре ViT (Vision Transformer) с базовой моделью google/vit-base-patch16-224-in21k. Она была дообучена на наборе данных, содержащем изображения галактик различных типов. Основные параметры конфигурации:

• Архитектура: ViTForImageClassification
• Размер изображения: 224x224 пикселя
• Размер патча: 16x16 пикселей
• Количество слоев: 12
• Количество заголовков внимания: 12
• Размер скрытого слоя: 768
• Функция активации: GELU

Отчет о классификации:

Применение

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

from transformers import pipeline
classifier = pipeline("image-classification", model="dima806/galaxy_type_image_detection")
# classifier("путь_к_вашему_изображению.jpg")

Ссылки