In this toy project, we aim to build a deep learning clssifier that predicts whether an image is cat or not.

Cat	Dog

Refer to this link to get the data.

In this project, we have two steps: training and predicting. In the predict step, you can upload any image from your laptop and predict it. Let's show you how to run the project.

If you do not have venv package, please refer to this link

$ python3 -m venv ENV_NAME

$ source ENV_NAME/bin/activate

To run this, make sure to install all the requirements by:

$ pip install -r requirements.txt 

$ python3 main.py --model MODEL_NAME --num_epochs

$ python3 main.py --model resnet --10

$ python3 main.py --model cnn --10

Resnet result

CNN result

$python3 predict.py --image_path "./data/Images/cat.0.jpg"

• Resnet
• Convolutional neural networks

Mr. A

Mr. B

Mm. K