Skip to content

Cats vs dogs classification using deep learning. Data augmentation and convolutional neural networks.

Notifications You must be signed in to change notification settings

RizwanMunawar/Cats-vs-dogs-classification-computer-vision-

Repository files navigation

CATS vs DOGS Classification using Convolutional Neural Networks and Data Augmentation

Dataset Details:
you can download dataset from google apis.

Dataset Description:
Dataset contain 3000 images of Cats and Dogs, we will train our model on 1700 images,710 images for validation and 604 images for testing.

Training Images of cats = 850
Training Images of dogs = 850

Validation Images of Cats = 352
Validation Images of Dogs = 358

Testing Images of Cats = 304
Testing Images of Dogs = 300

Overfiting and Underfitting aviodence Techniques Used:
1-Data Augmentation (zoom,horizontal_flip,rotation)
2-Dropout

Model Summary:
I used convolutional neural networks with 32, 64 and 128 layers.


Training and Validation Graph:

Results:
Achieved 84% Accuracy on Training data with epochs = 100
81% accuracy on validation data
80% accuracy on testing data