Problem understanding: Cat vs. Classifier

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.

Run the project

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

Create virtual environment

$ python3 -m venv ENV_NAME

Activate your environment

$ source ENV_NAME/bin/activate

Requirement installations

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

$ pip install -r requirements.txt 

Training the model

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

Example of running models

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

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

Results Presentation

Resnet result

CNN result

Make prediction

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

Example: Result

{'class': 'cat', 'confidence': '0.5970718264579773'}

---

---

---

Related Papers

• Resnet
• Convolutional neural networks

Contributors

Mr. A

Mr. B

Mm. K