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


Posted in Python onJanuary 24, 2020

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

import matplotlib
matplotlib.use('Agg')
import os
from keras.models import load_model
import numpy as np
from PIL import Image
import cv2
#加载模型h5文件
model = load_model("C:\\python\\python3_projects\\cat_dog\\cats_dogs_fifty_thousand.h5")
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文件,并实现预测图片就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
django启动uwsgi报错的解决方法
Apr 08 Python
Python实现的查询mysql数据库并通过邮件发送信息功能
May 17 Python
Python之用户输入的实例
Jun 22 Python
python3中zip()函数使用详解
Jun 29 Python
彻彻底底地理解Python中的编码问题
Oct 15 Python
Python实现的合并两个有序数组算法示例
Mar 04 Python
python3 json数据格式的转换(dumps/loads的使用、dict to str/str to dict、json字符串/字典的相互转换)
Apr 01 Python
几个适合python初学者的简单小程序,看完受益匪浅!(推荐)
Apr 16 Python
Python实现爬取亚马逊数据并打印出Excel文件操作示例
May 16 Python
Pytorch实现的手写数字mnist识别功能完整示例
Dec 13 Python
使用 Python 读取电子表格中的数据实例详解
Apr 17 Python
Python3自定义http/https请求拦截mitmproxy脚本实例
May 11 Python
keras模型可视化,层可视化及kernel可视化实例
Jan 24 #Python
keras 特征图可视化实例(中间层)
Jan 24 #Python
基于keras输出中间层结果的2种实现方式
Jan 24 #Python
tensorflow 保存模型和取出中间权重例子
Jan 24 #Python
tensorflow 模型权重导出实例
Jan 24 #Python
在Tensorflow中查看权重的实现
Jan 24 #Python
tensorflow求导和梯度计算实例
Jan 23 #Python
You might like
php为什么选mysql作为数据库? Mysql 创建用户方法
2007/07/02 PHP
php随机显示图片的简单示例
2014/02/15 PHP
prototype Element学习笔记(篇二)
2008/10/26 Javascript
javascript attachEvent绑定多个事件执行顺序问题
2010/10/20 Javascript
firefox浏览器不支持innerText的解决方法
2013/08/07 Javascript
JavaScript遍历table表格中的某行某列并打印其值
2014/07/08 Javascript
jquery中使用循环下拉菜单示例代码
2014/09/24 Javascript
JS中setTimeout的巧妙用法前端函数节流
2016/03/24 Javascript
详解node-ccap模块生成captcha验证码
2017/07/01 Javascript
详解TypeScript+Vue 插件 vue-class-component的使用总结
2019/02/18 Javascript
Vue 2.0双向绑定原理的实现方法
2019/10/23 Javascript
javascript Canvas动态粒子连线
2020/01/01 Javascript
vue项目打包之开发环境和部署环境的实现
2020/04/23 Javascript
[02:09]DOTA2辉夜杯 EHOME夺冠举杯现场
2015/12/28 DOTA
[42:52]IG vs VGJ.T 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
[30:37]【全国守擂赛】第三周擂主赛 Dark Knight vs. Leopard Gaming
2020/05/04 DOTA
python用来获得图片exif信息的库实例分析
2015/03/16 Python
合并百度影音的离线数据( with python 2.3)
2015/08/04 Python
python绘制直线的方法
2018/06/30 Python
使用Python轻松完成垃圾分类(基于图像识别)
2019/07/09 Python
python 协程中的迭代器,生成器原理及应用实例详解
2019/10/28 Python
python实现最速下降法
2020/03/24 Python
python如何进行矩阵运算
2020/06/05 Python
Python面向对象多态实现原理及代码实例
2020/09/16 Python
html5中localStorage本地存储的简单使用
2017/06/16 HTML / CSS
使用Html5 Stream开发实时监控系统
2020/06/02 HTML / CSS
德国高尔夫商店:Golfshop.de
2019/06/22 全球购物
写一个函数,要求输入一个字符串和一个字符长度,对该字符串进行分隔
2015/07/30 面试题
信息技术培训感言
2014/03/06 职场文书
我的长生果教学反思
2014/04/28 职场文书
安全环保标语
2014/06/09 职场文书
2014大四本科生自我鉴定总结
2014/10/04 职场文书
2015共产党员公开承诺书
2015/01/22 职场文书
杭州黄龙洞导游词
2015/02/10 职场文书
复兴之路观后感
2015/06/02 职场文书
mysql优化
2021/04/06 MySQL