https://github.com/lingfromsh/cat-and-dog-tutorials

Use alexnet to solving cat-and-dog classification
https://github.com/lingfromsh/cat-and-dog-tutorials

Last synced: 10 months ago
JSON representation

Use alexnet to solving cat-and-dog classification

• Host: GitHub
• URL: https://github.com/lingfromsh/cat-and-dog-tutorials
• Owner: lingfromSh
• Created: 2021-05-02T05:37:38.000Z (over 4 years ago)
• Default Branch: master
• Last Pushed: 2021-05-03T09:31:46.000Z (over 4 years ago)
• Last Synced: 2025-01-09T03:51:20.373Z (12 months ago)
• Language: Python
• Size: 2.93 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Cat and Dog

Use an alexnet to solve a bi-classification problem

## Training

You could run `train.py` to train the model.

After 100 epoches, this model could reach more than 90% to give a correct classification. 

## Predicting

You could run `predict.py` and set the input image by flag `-f`. After a few seconds, the model will give you the answer.  

## Testing

You could run `test.py` to check whether this model is overfitting training resources.

## About training and testing data

You could download cat&dog dataset from `kaggle` or elsewhere.

## Some advices

For having a more powerful ability, You could add some more preprocesses to generate nonstandard pics.