ResNet 进行图片识别 （基于tf.keras）

import tensorflow as tf

import matplotlib.pyplot as plt

from tensorflow.python.keras.applications.resnet50 import ResNet50

from tensorflow.python.keras.preprocessing import image

from tensorflow.python.keras.applications.resnet50 import preprocess_input, decode_predictions

inputs = tf.placeholder(tf.float32, (224, 224, 3))

tensorimg = tf.expand_dims(inputs, 0)

tensorimg =preprocess_input(tensorimg)

with tf.Session() as sess:

sess.run(tf.global_variables_initializer())

Reslayer = ResNet50(weights='imagenet')

logits = Reslayer(tensorimg)

img_path = 'hy.jpg'

img = image.load_img(img_path, target_size=(224, 224))

logitsv = sess.run(logits,feed_dict={inputs: img})

Pred =decode_predictions(logitsv, top=3)[0]

print('Predicted:', Pred,len(logitsv[0]))

fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(10, 8))

fig.sca(ax1)

ax1.imshow(img)

fig.sca(ax1)

barlist = ax2.bar(range(3), [ i[2] for i in Pred ])

barlist[0].set_color('g')

plt.sca(ax2)

plt.ylim([0, 1.1])

fig.subplots_adjust(bottom=0.2)

plt.show()

随便上网找一张图片，识别结果如下: