/Pytorch-Transfer-learning-Multi-Label-image-Classification

This is my sample kernel for the kaggle competition iMet Collection 2019 - FGVC6 (Recognize artwork attributes from The Metropolitan Museum of Art)

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Pytorch Transfer Learning image Classification

This is my sample kernel for the kaggle competition iMet Collection 2019 - FGVC6 (Recognize artwork attributes from The Metropolitan Museum of Art)

Highlights of this project:

  • Pytorch
  • Pytorch custom data class
  • Transfer learning (Resnet50)
  • Multi label classification
  • 1103 categories of labels