/ML-Image

An image classification project utilizing PyTorch and torchvision, featuring transfer learning with models like VGG, AlexNet, and ResNet. The project includes organized datasets and scripts for training and evaluation.

Primary LanguagePython

ML Image Classification Project - Udacity

Tech Leveraged in this image classification project:

  • Python
  • Pytorch
  • Torchvision
  • Transfer learning
  • VGG
  • AlexNet
  • RestNet
  • Tabulate
  • Colorama
  • Version control
  • CODES written with love and passion...