In this project we build a Convolutional Neural Network (CNN) to classify dogs and cats. The model has two versions, a single hidden layer model and a 4 hidden layers model. We also implemented a class for preprocessing/loading the data. The four layer model is as follows:
and its architecture is:
Open the requirements.txt to find out the packages required then install them before you proceed to running the code. For example, you will need to install pytorch. You may also need to download the data, then copy and paste the following link into your browser https://drive.google.com/drive/folders/1Wm2RcKtBSMipyjt8m6jg_tE04yGG8C-Z?usp=sharing
Put all Python files in the same folder together with the data, then run the main.py file in a terminal or any Python IDE
Jean N'dah Kouagou, AMMI-Gahna
email: [email protected]
This project is under the following License: GNU General Public License v2.0
When contributing to this repository, please first discuss the change you wish to make via issue, email, or any other method with the owner (s) of this repository before making a change.