https://github.com/abudubai16/cnn-using-transfer-learning

https://github.com/abudubai16/cnn-using-transfer-learning

Science Score: 13.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
  • DOI references
  • Academic publication links
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (4.7%) to scientific vocabulary
Last synced: 6 months ago · JSON representation

Repository

Basic Info
  • Host: GitHub
  • Owner: abudubai16
  • Language: Jupyter Notebook
  • Default Branch: main
  • Size: 101 KB
Statistics
  • Stars: 1
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Created almost 2 years ago · Last pushed over 1 year ago
Metadata Files
Readme

README.md

CNN-using-transfer-learning

This model implements the Inceptionet model for feature extraction, that was originally created by Google for use in GoogLeNet. This is the backbone for the entire model, the first few Inceptionet blocks are frozen during training so that the parameters for those layers do not change, the last block is made trainable, and a fully connected layer is implemented for a simple MNIST purpose. Ther MNIST has 5 classes and looks into classification of the types of diseases in a certain plant, the dataset is downloaded from Kaggle using opendatasets. Aux outputs from Inceptionet are voided for this particular model.

The best accuracy for this particular CNN was between 75-80%, I would like to save the parameters of the model for every epoch in training where the validation accuracy is better than the current saved model, and load those parameters for testing purposes allowing the best version of the model being saved. Since this is my first model using pytorch I will be looking into such things later.

Owner

  • Login: abudubai16
  • Kind: user

GitHub Events

Total
Last Year