Skip to content

Обучение классификаторов на основе сверточных сетей

Notifications You must be signed in to change notification settings

ivantipow/Simpsons-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 

Repository files navigation

Simpsons Classification

В данном ноутбуке рассматривается решение задачи классификации изображений в рамках учебного Kaggle-соревнования Journey to Springfield от Deep Learning School. Для решения используется техника Transfer Learning/Fine tuning над нейронными сетями Efficientnet_B4 и ResNet 18, предобученными на ImageNet. В этом блокноте приведён основной training loop, выполняется аугментация изображений, в качестве оптимизатора используется AdamW. Для работы с learning rate используется scheduler StepLR. Для борьбы с дисбалансом классов используется oversampling.

Результат на Kaggle: 0.99362 F1-Score.

About

Обучение классификаторов на основе сверточных сетей

Topics

Resources

Stars

Watchers

Forks