Multi-Label-Image-Classification

Part 1A - Pre-defined Models

• In the notebook introduction.ipynb I
  • Train AlexNet (PyTorch built-in) from scratch.
  • Fine-tune AlexNet (PyTorch built-in), which is pretrained on ImageNet.
  • Train a simple network (defined in classifier.py from scratch.)

Part 1B - Self-designed Models:

• In the notebook Develop_Classifier.ipynb, I design my own model in details to solve this multi-label classification task.