transfer-learning VGGNet16 class augmentation TO-DO Plot training/validation progress Add data augmentation Add dropout