Comparison of CNN Architectures for Image Classification

  • Tech Stack: Python, PyTorch, Jupyter, Matplotlib
  • Algos: Convolutional Neural Networks (CNN): LeNet, VGGNet and ResNet
  • Notebook URL: View on Github
  • Code URL: View on Github

The objective of this project is to demonstrate the creation and training of LeNet, VGGNet, and ResNet Convolutional Network models for image classification, and compare the results. These models were trained using Fashion MNIST, CIFAR-10, and EuroSAT datasets and have demonstrated their ability to classify various types of images, such as clothing items, objects, and land covers, respectively.