利用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字符串对象的实现
Dec 24 Python
Python实现抢购IPhone手机
Feb 07 Python
Python Unittest自动化单元测试框架详解
Apr 04 Python
python引入不同文件夹下的自定义模块方法
Oct 27 Python
python遍历小写英文字母的方法
Jan 02 Python
详解重置Django migration的常见方式
Feb 15 Python
django框架基于模板 生成 excel(xls) 文件操作示例
Jun 19 Python
Pandas透视表(pivot_table)详解
Jul 22 Python
Python如何避免文件同名产生覆盖
Jun 09 Python
Python进行统计建模
Aug 10 Python
如何使用Python对NetCDF数据做空间相关分析
Apr 21 Python
python基础入门之字典和集合
Jun 13 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 正则表达式常用函数
2014/08/17 PHP
PHP使用逆波兰式计算工资的方法
2015/07/29 PHP
php 一维数组的循环遍历实现代码
2017/04/10 PHP
DOM 基本方法
2009/07/18 Javascript
javascript 处理HTML元素必须避免使用的一种方法
2009/07/30 Javascript
JQuery 绑定事件时传递参数的实现方法
2009/10/13 Javascript
Javascript UrlDecode函数代码
2010/01/09 Javascript
jQuery中index()方法用法实例
2014/12/27 Javascript
jquery实现下拉框功能效果【实例代码】
2016/05/06 Javascript
KnockoutJS 3.X API 第四章之表单textInput、hasFocus、checked绑定
2016/10/11 Javascript
jQuery得到多个值只能用取Class ,不能用取ID的方法
2016/12/04 Javascript
JS中利用localStorage防止页面动态添加数据刷新后数据丢失
2017/03/10 Javascript
微信小程序之GET请求的实例详解
2017/09/29 Javascript
vue的keep-alive用法技巧
2019/08/15 Javascript
js脚本中执行java后台代码方法解析
2019/10/11 Javascript
解决父组件将子组件作为弹窗调用只执行一次created的问题
2020/07/24 Javascript
nuxt.js 在middleware(中间件)中实现路由鉴权操作
2020/11/06 Javascript
使用JS实现鼠标放上图片进行放大离开实现缩小功能
2021/01/27 Javascript
用Python实现服务器中只重载被修改的进程的方法
2015/04/30 Python
Python中字典的基础知识归纳小结
2015/08/19 Python
使用Python搭建虚拟环境的配置方法
2018/02/28 Python
对python中dict和json的区别详解
2018/12/18 Python
在python里协程使用同步锁Lock的实例
2019/02/19 Python
Python当中的array数组对象实例详解
2019/06/12 Python
python实现学生信息管理系统(精简版)
2020/11/27 Python
详解canvas多边形(蜘蛛图)的画法示例
2018/01/29 HTML / CSS
Spanx塑身衣官网:美国知名内衣品牌
2017/01/11 全球购物
医生自荐信
2013/10/11 职场文书
开办化妆品公司创业计划书
2013/12/26 职场文书
高校教师思想汇报
2014/01/11 职场文书
单位承诺书格式
2014/05/21 职场文书
党员创先争优心得体会
2014/09/11 职场文书
2014年英语教学工作总结
2014/12/17 职场文书
医药公司采购员岗位职责
2015/04/03 职场文书
开业典礼致辞
2015/07/29 职场文书
经典格言警句:没有热忱,世间便无进步
2019/11/13 职场文书