bg-cover

Develop a CNN to classify images

This is one of my machine learning projects that involves classifying photos as either containing a dog or cat. The dataset contains 24,998 images, split into 12,499 Cat images and 12,499 Dog images. The classification uses a Convolutional Neural Network architecture. Currently, The test loss and accuracy are 0.23407 and 90.13%. I will tune it, and it may have better accuracy in the futrue.

  • Machine Learning
  • Computer Vision
  • Python
  • Jupyter
  • Pytorch
Cats-and-Dogs-Classification-cover