Skip to content

My project to classify cat and dog using pytorch very easy way!

Notifications You must be signed in to change notification settings

vovod/dog-cat-classify

Repository files navigation

DOG vs CAT classify

My project to classify cat and dog using pytorch!

DOG CAT

Requirements

CUDA, conda installed.

conda create --name torch-cuda
conda activate torch-cuda
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Dataset

KAGGLE: Dogs vs. Cats Redux: Kernels Edition.
The train folder contains 25,000 images of dogs and cats.
The test folder contains 12,500 images of dogs and cats - non classify.

Let's start!

Preprocessing data:

preprocess.py

Training with custom model:

Before training, create a folder name weights to save model.

train.py

And predicting:

predict.py

This is the end!

The model from Juan Zamora.

Thank you for stopping by!

About

My project to classify cat and dog using pytorch very easy way!

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages