The Network

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