mmheydari97 / cat-vs-dog

Solution to cats vs. dogs kaggle competition using CNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cats vs. Dogs

This repository contains a solution to cats vs dogs kaggle competition. The dataset contains about 25,000 pictures of cats and dogs and an algorithm to be developed should classify them correctly. I used a CNN architecture similar to the AlexNet with minor adjustments. One of the challenges is to feed the network because probably your machine is not able to load the dataset entirely. Furthermore, it is not efficient at all. so I used a data generator with python to move only small batches of the dataset to the ram each time. The accuracy was confidently more than %90 on both training and test sets.

note

I was running my code with a batch size equal to 64 for a long time. I observed that the accuracy bounces up and down
in each epoch and its trend is too noisy. I solved this issue by doubling the batch size to 128. You may need this conclusion
in similar problems so I decided to note it.

License

This project is licensed under the Apache License 2.0 - see the LICENSE file for details

About

Solution to cats vs. dogs kaggle competition using CNN

License:Apache License 2.0


Languages

Language:Jupyter Notebook 68.5%Language:Python 31.5%