/TransferLearningLab

Primary LanguageJupyter Notebook

ImageClassification - A Simple Deep Learning Network for Image Classification

Install & Run

  1. Python 3.
  2. NVIDIA cuDNN (Optional).
  3. pip install -r requirements.
  4. Get dataset. Note: Make sure to delete dataset\train\snake\25.png because it is not encoded in png format.
  5. Run image_classification.ipynb in Jupyter Notebook or any IDE with its extension.