Skip to content

# This is a image classification by VGG16 pre-trained model.#

Notifications You must be signed in to change notification settings

luntai/VGG16_Keras_TensorFlow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

VGG16_Keras_TensorFlow

This is a image classification by VGG16 pre-trained model.

Envoriment

Python3.5
Keras2.0
TensorFlow
Windows10
Most py-package (such like numpy or cv2...) you can install by pip install xx.whl from here.

Steps

Download the weight file 'vgg16_weights_tf_dim_ordering_tf_kernels.h5'. In here and download it.
Put the weight file into data/model

Used Files

model/synset_words.txt : map class id (which is equal to line id) to synset_id and human words. For example, If you get the prediction id is 281, then you should out put the 281th content in synset_words.txt. model/vgg16_weights_tf_dim_ordering_tf_kernels.h5: special VGG16 pre-trained weight file for Keras2.

Hint

Mind the path which is used in windows. If you are UNIX player, should change the format.
Any other errors happen you can check my blog.
You should be careful with the Keras's backend (th:theano or tf:tensorflow), because they have different mat size format.

Resource

About model you can find in Github-Glist.
About VGG16 network structure find in here.

Please

Give me a star. Thks.

About

# This is a image classification by VGG16 pre-trained model.#

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages