/Transfer_Learning_Comparision

Comparison of CNN Models by performing classification on Medical MNIST data-set via Transfer Learning.

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

Transfer_Learning_Comparision

Medical MNIST Classifcation

  • Applied transfer learning to classify medical imaging dataset using tensorflow-keras in google colab.
  • Implemented CNN architectures such as Vgg19, Resnet50, Xception, MobileNet and EfficentNet using functional Api.

References:

This project was based off the transfer learning tutorial from Tensorflow guide: https://www.tensorflow.org/tutorials/images/transfer_learning

Dataset Used: Medical Mnist

Url: https://www.kaggle.com/andrewmvd/medical-mnist

Accuracy on Train & Valid Sets

alt text

Loss on Train & Valid Sets

alt text

Accuracy & Loss on Test Set

alt text