This repository contains an implementation of AlexNet(resnetv1 mobilenetv2) convolutional neural network.
python multigpu_train.py
if you want to use other train data instead of Dogs vs Cats(link: https://pan.baidu.com/s/1GoFCwLs8hetPdPdqdCnRPA passcode:pxkx), you may need to modify file: readdata.py
python test.py
- Alex Krizhevsky, Ilya Sutskever and Geoffrey E. Hinton. ImageNet Classification with Deep Convolutional Neural Networks. Advances in Neural Inforamtion Processing Systems 25, 2012.
- https://github.com/argman/EAST
- https://github.com/tensorflow/models/tree/master/research/slim