利用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 相关文章推荐
Python中标准库OS的常用方法总结大全
Jul 19 Python
Python文件操作基本流程代码实例
Dec 11 Python
python的staticmethod与classmethod实现实例代码
Feb 11 Python
PyQt QCombobox设置行高的方法
Jun 20 Python
树莓派+摄像头实现对移动物体的检测
Jun 22 Python
python 三元运算符使用解析
Sep 16 Python
tensorflow的计算图总结
Jan 12 Python
python 给图像添加透明度(alpha通道)
Apr 09 Python
jupyter lab文件导出/下载方式
Apr 22 Python
Python计算矩阵的和积的实例详解
Sep 10 Python
Python爬虫之Selenium警告框(弹窗)处理
Dec 04 Python
粗暴解决CUDA out of memory的问题
May 22 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 date函数常用时间处理方法
2015/05/11 PHP
JavaScript 不只是脚本
2007/05/30 Javascript
javascript中的prototype属性实例分析说明
2010/08/09 Javascript
JS文本框追加多个下拉框的值的简单实例
2013/07/12 Javascript
jquery实现pager控件示例
2014/04/09 Javascript
举例讲解Node.js中的Writable对象
2015/07/29 Javascript
JS+CSS实现简单滑动门(滑动菜单)效果
2015/09/19 Javascript
JS实现自动阅读单词(有道单词本添加功能)
2016/11/14 Javascript
深入理解JS继承和原型链的问题
2016/12/17 Javascript
利用BootStrap的Carousel.js实现轮播图动画效果
2016/12/21 Javascript
在 Vue.js中优雅地使用全局事件的方法
2019/02/01 Javascript
Node.js 获取微信JS-SDK CONFIG的方法示例
2019/05/21 Javascript
微信小程序录音实现功能并上传(使用node解析接收)
2020/02/26 Javascript
Openlayers+EasyUI Tree动态实现图层控制
2020/09/28 Javascript
Python中decorator使用实例
2015/04/14 Python
Windows下python2.7.8安装图文教程
2016/05/26 Python
python如何在列表、字典中筛选数据
2018/03/19 Python
Python基于TCP实现会聊天的小机器人功能示例
2018/04/09 Python
Python简单定义与使用二叉树示例
2018/05/11 Python
python操作excel文件并输出txt文件的实例
2018/07/10 Python
python如何读取bin文件并下发串口
2019/07/05 Python
浅谈tensorflow 中的图片读取和裁剪方式
2020/06/30 Python
三星美国官网:Samsung美国
2017/02/06 全球购物
高中毕业生个人自我鉴定
2013/11/24 职场文书
我的大学生活职业生涯规划
2014/01/02 职场文书
食品销售计划书
2014/04/26 职场文书
小学校园广播稿集锦
2014/10/04 职场文书
2014年药剂科工作总结
2014/11/26 职场文书
党风廉政建设个人总结
2015/03/06 职场文书
2016年国庆节新闻稿范文
2015/11/25 职场文书
看古人们是如何赞美老师的?
2019/07/08 职场文书
导游词之西安大清真寺
2019/12/17 职场文书
2019关于垃圾分类处理的调查报告
2019/12/26 职场文书
mysql 如何获取两个集合的交集/差集/并集
2021/06/08 MySQL
如何利用React实现图片识别App
2022/02/18 Javascript
mysql的单列多值存储实例详解
2022/04/05 MySQL