rcmalli / keras-vggface

VGGFace implementation with Keras Framework
MIT License
928 stars 417 forks source link

The example on Feature extraction #7

Closed piggyandy closed 7 years ago

piggyandy commented 7 years ago

For feature extraction, wouldn't it simply be the code for "Prediction" except I would do:

model = VGGFace(include_top=False, pooling='avg')

(this is with reference to Keras' documentation on using their VGG16 model for feature extraction)

Moreover, I see that your code for pre-processing an image is similar to Keras' pre_processing function for imagenet_utils except I noticed that the values for zero-center by mean pixel are different, are they supposed to be?

Thanks!!

rcmalli commented 7 years ago

For the first question, it really depends on the layer you want to extract features. The information you get will be a bit different between "pooling", "fc6" and "fc7".

For the second one, this network trained on the different dataset (Check the references) so, mean pixel values are also not same as ImageNet.

piggyandy commented 7 years ago

Thank you!