Pics can be seen in the Jupyter notebook
Overall the results are pretty good for such a small amount of data and training. Also optimization of the model (loss function mostly) should improve the results if trained on full imagenet set etc
- Using Imagenet training set
- Using Resnet50 as the basic model for feature extraction