insaiyancvk/catdogvision

Cats vs Dogs classifier based on CNNs implemented using PyTorch

Cats and Dogs classifier

Cats vs Dogs classifier trained on MobileNetV2. Trained and implemented using PyTorch.

Setup

Clone the repository:

git clone https://github.com/insaiyancvk/catdogvision

Navigate into the folder:

cd catdogvision

For running CLI app:

Install dependencies:

pip install torch torchvision pillow requests

Run the CLI app:

python cli.py

For running the web app:

Install dependencies:

pip install -r requirements.txt

Run the app:

python app.py