Skip to content

Image Classification with PyTorch on famous Cats vs Dogs dataset

Notifications You must be signed in to change notification settings

iskon4ik/cats_vs_dogs_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Cats vs Dogs Image Classification

Image Classification on famous Cats vs. Dogs dataset.
Dataset downloaded from Kaggle Competition: Dogs vs. Cats.
Model was based on pretrained ResNet34.

Stack used:

  • PyTorch - everything related to model training pipeline
  • sckit-learn - data split, metrics calculation
  • matplotlib - visualization
  • numpy - some minor data transformation during training/validation process
  • pandas - data exploration
  • Image.PIL - image processing and plotting
  • tqdm - training/validation progressbar

About

Image Classification with PyTorch on famous Cats vs Dogs dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published