Skip to content

Kaggle competition dogs-vs-cats. Implemented Logistic Regression and InceptionV3 (transfer learning).

License

Notifications You must be signed in to change notification settings

yliang725/dogs-vs-cats

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

dogs-vs-cats

This program implemented Logistic Regression and InceptionV3 to classify the Kaggle competition dogs-vs-cats images (https://www.kaggle.com/c/dogs-vs-cats).

Data Preprocessing

Used torchvision transforms method to resize, center crop, to tensor and normalize the images.

Results

The testing accuracy for Logistic Regression is 69.70%.

The testing accuracy for InceptionV3 (transfer learning) is 99.15%.

The above results show that the InceptionV3 method had a much better performance compared to the Logistic Regression method.

Environment Settings

Both methods were implemented with PyTorch API.

The Jupyter nootbook was compiled on Anaconda, Python 3.7, and PyTorch 1.7.1.

About

Kaggle competition dogs-vs-cats. Implemented Logistic Regression and InceptionV3 (transfer learning).

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published