This is a Convolutional Neural Network (CNN) model built using TensorFlow and Keras to classify images as either cats or dogs.
- Dataset used: Dogs vs Cats dataset from Kaggle
- Images are resized to 256x256 pixels
- Model architecture consists of 3 convolutional layers with batch normalization and max pooling, followed by fully connected dense layers with dropout
- Binary classification using sigmoid activation in the output layer
- Trained for 10 epochs with Adam optimizer and binary cross-entropy loss
- Python 3.x
- TensorFlow
- Keras
- OpenCV (
cv2) - Matplotlib
You can install the necessary packages with:
pip install tensorflow keras opencv-python matplotlib