A deep learning model to classify dogs and cats
This repository contains a deep learning model using CNN to classify between Cats and Dogs. The dataset is downloaded from the Kaggle.
Libraries Used are:
TensorFlow
Keras
matplotlib
numpy
pandas
scikit-learn
openCV
keras.utils.image_dataset_from_directory is used for data preparation. It simplifies the process of loading and preprocessing image data for training machine learning models. The function takes care of reading images from the specified directory, resizing them, and converting them to a format suitable for training.
Model used three convolution layers, three max pooling layers, one flatten layer and 3 fully connected layers.
Results:
Train accuracy: 99%
Validation accuracy: 80%
shubhammachal / cat_vs_dogcnn Goto Github PK
View Code? Open in Web Editor NEWA deep learning model to classify dogs and cats