利用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的__builtin__模块中的一些要点知识
May 02 Python
Python的Twisted框架上手前所必须了解的异步编程思想
May 25 Python
Python获取SQLite查询结果表列名的方法
Jun 21 Python
微信跳一跳辅助python代码实现
Jan 05 Python
ubuntu17.4下为python和python3装上pip的方法
Jun 12 Python
Tensorflow加载预训练模型和保存模型的实例
Jul 27 Python
Windows系统下PhantomJS的安装和基本用法
Oct 21 Python
Python3将数据保存为txt文件的方法
Sep 12 Python
python 操作hive pyhs2方式
Dec 21 Python
python 实现百度网盘非会员上传超过500个文件的方法
Jan 07 Python
python数字转对应中文的方法总结
Aug 02 Python
Python字符串格式化方式
Apr 07 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 函数学习简单小结
2010/07/08 PHP
探讨php中防止SQL注入最好的方法是什么
2013/06/10 PHP
linux下编译安装memcached服务
2014/08/03 PHP
thinkphp循环结构用法实例
2014/11/24 PHP
使用php自动备份数据库表的实现方法
2017/07/28 PHP
php+js实现点赞功能的示例详解
2020/08/07 PHP
在修改准备发的批量美化select+可修改select时,在非IE下发现了几个问题
2007/01/09 Javascript
jquery+ashx无刷新GridView数据显示插件(实现分页、排序、过滤功能)
2010/04/25 Javascript
十个迅速提升JQuery性能让你的JQuery跑得更快
2012/12/10 Javascript
简单实现JS倒计时效果
2016/12/23 Javascript
promise处理多个相互依赖的异步请求(实例讲解)
2017/08/03 Javascript
bootstrap table实现x-editable的行单元格编辑及解决数据Empty和支持多样式问题
2017/08/10 Javascript
移动设备手势事件库Touch.js使用详解
2017/08/18 Javascript
解决Vue不能检测数组或对象变动的问题
2018/02/24 Javascript
20道JS原理题助你面试一臂之力(必看)
2019/07/22 Javascript
如何使用proxy实现一个简单完整的MVVM库的示例代码
2019/09/17 Javascript
Node.js利用Express实现用户注册登陆功能(推荐)
2020/10/26 Javascript
vue脚手架项目创建步骤详解
2021/03/02 Vue.js
[50:58]2018DOTA2亚洲邀请赛3月29日 小组赛A组OpTic VS Newbee
2018/03/30 DOTA
Python中序列的修改、散列与切片详解
2017/08/27 Python
使用pygame模块编写贪吃蛇的实例讲解
2018/02/05 Python
python用BeautifulSoup库简单爬虫实例分析
2018/07/30 Python
python计算阶乘和的方法(1!+2!+3!+...+n!)
2019/02/01 Python
Python循环结构的应用场景详解
2019/07/11 Python
Django实现发送邮件功能
2019/07/18 Python
Python socket连接中的粘包、精确传输问题实例分析
2020/03/24 Python
Django利用elasticsearch(搜索引擎)实现搜索功能
2020/11/26 Python
Python爬虫自动化爬取b站实时弹幕实例方法
2021/01/26 Python
Html5移动端适配IphoneX等机型的方法
2019/06/25 HTML / CSS
WatchShop法国:英国排名第一的独立手表零售商
2020/02/17 全球购物
数百万免费的图形资源:Freepik
2020/09/21 全球购物
Bloomingdale’s阿联酋:选购奢华时尚、美容及更多
2020/09/22 全球购物
Linux的文件类型
2012/03/07 面试题
CAD制图人员的自荐信
2014/02/07 职场文书
股权转让协议书范本
2014/04/12 职场文书
培训心得体会怎么写
2016/01/25 职场文书