The model was created in CreateML using the dataset https://github.com/garythung/trashnet
Download the model here
- 80% of the images from each class used for training, 20% for testing
- Flip and rotate data augmentations applied
- Training accuracy = 87%. Validation accuracy = 88%