
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
