"Very Deep Convolutional Networks for Large-Scale Image Recognition" by Karen Simonyan and Andrew Zisserman.
Paper: https://arxiv.org/pdf/1409.1556.pdf
GPU POOR !!!
Didn't train cause I don't have a powerful GPU. But the architecture is there for playing.
Run script below to checkout the model informations
python info.py
Before running the script, place your data directory location for both train and test data in root_dir="{DIR}"
here at dataloader.py
python train.py --epochs 74 --vgg vgg16
@misc{simonyan2015deepconvolutionalnetworkslargescale,
title={Very Deep Convolutional Networks for Large-Scale Image Recognition},
author={Karen Simonyan and Andrew Zisserman},
year={2015},
eprint={1409.1556},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/1409.1556},
}