Chocolate chip vs raisin cookies image classification with fastai
Our goal is to train a model that saves humanity from the repeated tragedy:
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.
- 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
@ben-mj and @cstorm125