利用keras加载训练好的.H5文件,并实现预测图片-创新互联

我就废话不多说了,直接上代码吧!

成都创新互联公司-专业网站定制、快速模板网站建设、高性价比轵城网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式轵城网站制作公司更省心,省钱,快速模板网站建设找我们,业务覆盖轵城地区。费用合理售后完善,10余年实体公司更值得信赖。
import matplotlib
matplotlib.use('Agg')
import os
from keras.models import load_model
import numpy as np
from PIL import Image
import cv2
#加载模型h6文件
model = load_model("C:\\python\\python3_projects\\cat_dog\\cats_dogs_fifty_thousand.h6")
model.summary()
#规范化图片大小和像素值
def get_inputs(src=[]):
  pre_x = []
  for s in src:
    input = cv2.imread(s)
    input = cv2.resize(input, (150, 150))
    input = cv2.cvtColor(input, cv2.COLOR_BGR2RGB)
    pre_x.append(input) # input一张图片
  pre_x = np.array(pre_x) / 255.0
  return pre_x
#要预测的图片保存在这里
predict_dir = 'C:\python\python3_projects\cat_dog\pics'
#这个路径下有两个文件,分别是cat和dog
test = os.listdir(predict_dir)
#打印后:['cat', 'dog']
print(test)
#新建一个列表保存预测图片的地址
images = []
#获取每张图片的地址,并保存在列表images中
for testpath in test:
  for fn in os.listdir(os.path.join(predict_dir, testpath)):
    if fn.endswith('jpg'):
      fd = os.path.join(predict_dir, testpath, fn)
      print(fd)
      images.append(fd)
#调用函数,规范化图片
pre_x = get_inputs(images)
#预测
pre_y = model.predict(pre_x)
print(pre_y)

标题名称:利用keras加载训练好的.H5文件,并实现预测图片-创新互联
分享路径:http://pwwzsj.com/article/psidj.html