An open API service indexing awesome lists of open source software.

https://github.com/sirius-mhlee/image-classification-using-tensorflow-vgg16

Tensorflow implementation of Very Deep Convolutional Networks for Large-Scale Image Recognition.
https://github.com/sirius-mhlee/image-classification-using-tensorflow-vgg16

deep-learning image-classification paper-implementation python tensorflow vgg16

Last synced: 2 months ago
JSON representation

Tensorflow implementation of Very Deep Convolutional Networks for Large-Scale Image Recognition.

Awesome Lists containing this project

README

          

Image Classification using Tensorflow VGG16
===========================================

Tensorflow implementation of the model described in the paper Very Deep Convolutional Networks for Large-Scale Image Recognition.
http://www.robots.ox.ac.uk/~vgg/research/very_deep/

Very Deep Convolutional Networks for Large-Scale Image Recognition
K. Simonyan, A. Zisserman
arXiv Technical report, 2014
ImageNet Large Scale Visual Recognition Competition 2014.

Training Usage
-----

python TrainingModel.py train_data_file max_epoch batch_size output_npy_file output_mean_file

ex)
python TrainingModel.py ./data/train_data.txt 3 3 ./vgg16/save_model.npy ./vgg16/save_mean.txt

Training Result
------

![lena_train_result.jpg](./example/lena_train_result.jpg)

Prediction Usage
-----

python PredictionImage.py npy_file mean_file imagenet_label_file input_image_file

ex)
python PredictionImage.py ./vgg16/vgg16.npy ./vgg16/vgg16_mean.txt ./vgg16/imagenet_classes.txt ./example/lena.jpg

Prediction Result
------

![lena.jpg](./example/lena.jpg) ![lena_result.jpg](./example/lena_result.jpg)