An open API service indexing awesome lists of open source software.

https://github.com/sovit-123/cats-vs.-dogs-image-classification


https://github.com/sovit-123/cats-vs.-dogs-image-classification

Last synced: 8 months ago
JSON representation

Awesome Lists containing this project

README

          

# Cats-vs.-Dogs-Image-Classification
This is an image classification project based on the past [Kaggle Competition]( https://www.kaggle.com/c/dogs-vs-cats/overview).

## Tech Stack
* Keras to build Neural Network Model.
* **Other required imports are very general to any Machine Learning project (like Numpy and Matplotlib). You can surely follow along with the code file.**

## Note
This an image classification project with almost 25000 images used for training. Consider using a GPU for training purpose. You can use Kaggle Kernels or Google Colaboratory as well.

The training has been done twice, once with just basic preprocessing (converting images to pixel tensors) and again with image augmentation. Try to reproduce the results by running in Jupyter Notebook. Upload your results in Kaggle and try to improve.

## Some Future Steps
* Try using a pre-trained network to extract the features and then your own dense layer for classification. Compare the results.