https://github.com/froukje/cnn-image-classification

This respository contains several examples of CNN image classification.
https://github.com/froukje/cnn-image-classification

cnn image-classification machine-learning pytorch transfer-learning

Last synced: about 1 month ago
JSON representation

This respository contains several examples of CNN image classification.

• Host: GitHub
• URL: https://github.com/froukje/cnn-image-classification
• Owner: froukje
• Created: 2022-04-29T08:48:22.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-12-02T16:41:33.000Z (about 1 year ago)
• Last Synced: 2023-12-02T17:31:44.676Z (about 1 year ago)
• Topics: cnn, image-classification, machine-learning, pytorch, transfer-learning
• Language: Jupyter Notebook
• Homepage:
• Size: 8.38 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Image Classification examples using CNNs (PyTorch)

* cats_dogs: binary classification

    * data: https://www.kaggle.com/datasets/tongpython/cat-and-dog 

    * classify cats and dogs using two different networks:

	1. self implemented CNN

	2. Transfer Learning using the VGG16 model

* birds species: multiclass classification

   * data: https://www.kaggle.com/datasets/gpiosenka/100-bird-species

   * classify 400 bird species with transfer learning using the VGG16 model

* crops: multiclass classification 

   * data: https://www.kaggle.com/aman2000jaiswal/agriculture-crop-images

   * The purpose of this exercise is to compare the workflow of Pytorch and Pytorch Lightning

* lego_figures: multiclass classification

   * data: https://www.kaggle.com/ihelon/lego-minifigures-classification

   * The purpose of this repository is to compare Tensorflow with PyTorch

* plant classification: multiclass classification

  * data: https://www.kaggle.com/datasets/vbookshelf/v2-plant-seedlings-dataset/code

  * An end-to-end notebook for image classification using PyTorch using transfer learning