Определение жанра по изображению обложки музыкального диска

Очевидно, что оформление музыкального альбома как-то связано с его содержанием. Но насколько связано? Как подтвердить это, опираясь на данные? И чем это может быть полезно?

Задача

Необходимо разработать модель, которая определит жанр музыкального альбома по изображению его обложки.

Инструменты

• pandas
• sklearn
• pytorch
• fastai
• resnet50

Отработанные методы и навыки

1. Загружены данные и проведена предобработка.
2. Построена сверточная нейронная сеть на основе предобученной модели ResNet-50 с фреймворком pytorch.

Итоги проекта

Samples avg (f1-score) = 0.82 средняя оценка, считая все категории равноправными. В целом модель обучилась хорошо, но из-за дисбаланса классов метрика хуже, чем могла бы быть. Рекомендацией является увеличение размера текущего набора данных, а точнее малочисленных класоов с f1-score < 0,6 : disco, jazz, pop. Это позволит улучшить метрику и качество работы модели.