Skip to content
main
Switch branches/tags
Code

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

choco-raisin

Chocolate chip vs raisin cookies image classification with fastai

Training notebook: Open In Colab

Inference webapp: Binder

Medium article in Thai

Our goal is to train a model that saves humanity from the repeated tragedy:

cookies

We train a chocalate chip cookies vs raisin cookies image classifier with data obtained from DuckDuckGo image search. We obtained over 90% accruacy in the validation set.

Repository Structure

- images
    - train
        -chocolate chip (572 images)
        - raisin (570 images)
    - valid
        -chocolate chip (144 images)
        - raisin (140 images)
- notebooks
    - train_image_classification.ipynb; train the model and basic inference notebook app
    - train_valid_splits.ipynb; splits and clean by hand
    - download_images.ipynb; download images

Authors

@ben-mj and @cstorm125

About

Chocolate chip vs raisin cookies image classification with fastai

Resources

License

Packages

No packages published