question_answer.py 文件源码

def extract_image_features(img_path):
    model = models.VGG_16('weights/vgg16_weights.h5')
    img = image.load_img(img_path,target_size=(224,224))
    x = image.img_to_array(img)
    x = np.expand_dims(x,axis=0)
    x = preprocess_input(x)
    last_layer_output = K.function([model.layers[0].input,K.learning_phase()],
        [model.layers[-1].input])
    features = last_layer_output([x,0])[0]
    return features