我们用的model是vgg16
vgg16的下载地址:
其结构如下,可以看到我们input这个模型中的图片大小应该为224*224的:
这是我们要用到的原图:
运行代码如下:
from keras.applications.vgg16 import VGG16
from keras import backend as K
K.clear_session()
model = VGG16(weights='./vgg16_weights_tf_dim_ordering_tf_kernels.h5')
model.summary()
from keras.preprocessing import image
from keras.applications.vgg16 import preprocess_input, decode_predictions
import numpy as np
# 此处是导入你想处理的照片的地址(如果照片和vgg16文件还有你写的代码文件在同一个存储路径的话,此处只需要写照片的名称)
img_path = 'creative_commons_elephant.jpg'
# 此处的target_size只能是224*224的,因为vgg16模型的结构中,规定的导入图片的大小就是224*224的
img = image.load_img(img_path, target_size=(224, 224))
# 将图片信息转换为数组的形式
x = image.img_to_array(img