This project implements a convolutional neural network (CNN) for image classification of cats and dogs. The CNN consists of four convolutional layers three max pooling layers and two dense layers. It is trained using the CIFAR-10 dataset which contains 60000 32x32 color images divided into 10 classes. The model is compiled with the Adam optimizer sparse categorical crossentropy loss function and accuracy metric. It is trained for 10 epochs and achieves an accuracy of around 90% on the test set. The trained model can be saved and loaded for future use. Additionally the project includes code for visualizing sample images and their corresponding predictions. This project showcases the implementation of a CNN for image classification using TensorFlow and demonstrates its effectiveness in distinguishing between cats and dogs.
-
Notifications
You must be signed in to change notification settings - Fork 0
Vaishakgkumar/Implementation-of-simple-CNN
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published