TensorFlow implementation of AlexNet and its training and testing on ImageNet ILSVRC 2012 dataset
Clone or download
Latest commit f3ae75e Feb 7, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
tf correct mismatched parenthesis Feb 7, 2018
tf_eager readmes edited Nov 20, 2017
LICENSE.md license added Feb 1, 2017
README.md classic tensorflow folder Nov 20, 2017
tsne-full.jpg tsne pic Nov 20, 2017

README.md

AlexNet training on ImageNet LSVRC 2012

alt text

This repository contains an implementation of AlexNet convolutional neural network and its training and testing procedures on the ILSVRC 2012 dataset, all using TensorFlow.

Folder tf contains code in the "classic TensorFlow" framework whereas code in the tf_eager directory has been developed with TensorFlow's new impearative style, TensorFlow eager.

The two implementations are independent and refer to the READMEs inside the folders for specific instruction on how to train and to test.

References

  • Alex Krizhevsky, Ilya Sutskever and Geoffrey E. Hinton. ImageNet Classification with Deep Convolutional Neural Networks. Advances in Neural Inforamtion Processing Systems 25, 2012.
  • Olga Russakovsky°, Jia Deng°, Hao Su, Jonathan Krause, Sanjeev Satheesh, Sean Ma, Zhiheng Huang, Andrej Karpathy, Aditya Khosla, Michael Bernstein, Alexander C. Berg and Li Fei-Fei. (° = equal contribution) ImageNet Large Scale Visual Recognition Challenge. IJCV, 2015