Oxford VGGFace Implementation using Keras Functional Framework v2+
- This model is converted from original caffe network. (580 MB)
- It supports both Tensorflow and Theano backeds.
- You can also load only feature extraction layers with VGGFace(include_top=False) initiation (59MB).
- When you use it for the first time , weights are downloaded and stored in ~/.keras folder.
pip install keras_vggface
-
Label names are available now (Check the prediction code).
-
Project is now up-to-date with the new Keras version (2.0).
-
Old Implementation is still available at 'keras1' branch.
- Keras v2.0+
- Tensorflow 1.0+
- Warning: Tests are failing when backend selected as Theano!.
- Feature Extraction
from keras.engine import Model
from keras.layers import Input
from keras_vggface.vggface import VGGFace
# Convolution Features
vgg_model_conv = VGGFace(include_top=False, input_shape=(224, 224, 3), pooling='avg') # pooling: None, avg or max
# FC7 Features
vgg_model = VGGFace() # pooling: None, avg or max
out = vgg_model.get_layer('fc7').output
vgg_model_fc7 = Model(vgg_model.input, out)
# After this point you can use your models as usual for both.
# ...- Finetuning
from keras.engine import Model
from keras.layers import Flatten, Dense, Input
from keras_vggface.vggface import VGGFace
#custom parameters
nb_class = 2
hidden_dim = 512
vgg_model = VGGFace(include_top=False, input_shape=(224, 224, 3))
last_layer = vgg_model.get_layer('pool5').output
x = Flatten(name='flatten')(last_layer)
x = Dense(hidden_dim, activation='relu', name='fc6')(x)
x = Dense(hidden_dim, activation='relu', name='fc7')(x)
out = Dense(nb_class, activation='softmax', name='fc8')(x)
custom_vgg_model = Model(vgg_model.input, out)
# Train your model as usual.
# ...- Prediction
import numpy as np
from keras.preprocessing import image
from keras_vggface.vggface import VGGFace
from keras_vggface import utils
# tensorflow
model = VGGFace()
# Change the image path with yours.
img = image.load_img('../image/ak.jpg', target_size=(224, 224))
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = utils.preprocess_input(x)
preds = model.predict(x)
print('Predicted:', utils.decode_predictions(preds))Original models can be used for non-commercial research purposes under Creative Commons Attribution License.
The code that provided in this project is under MIT License.
If you find this project useful, please include reference link in your work.