利用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之父谈Python的未来形式
Jul 01 Python
Python对CSV、Excel、txt、dat文件的处理
Sep 18 Python
Python使用sqlalchemy模块连接数据库操作示例
Mar 13 Python
pyqt实现.ui文件批量转换为对应.py文件脚本
Jun 19 Python
django一对多模型以及如何在前端实现详解
Jul 24 Python
使用OpenCV-python3实现滑动条更新图像的Canny边缘检测功能
Dec 12 Python
python matplotlib包图像配色方案分享
Mar 14 Python
Python类的动态绑定实现原理
Mar 21 Python
详解查看Python解释器路径的两种方式
Oct 15 Python
Prometheus开发中间件Exporter过程详解
Nov 30 Python
matplotlib自定义鼠标光标坐标格式的实现
Jan 08 Python
Python实现我的世界小游戏源代码
Mar 02 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 读写json文件及修改json的方法
2018/03/07 PHP
PHP PDOStatement::bindValue讲解
2019/01/30 PHP
jquery blockUI 遮罩不能消失与不能提交的解决方法
2011/09/17 Javascript
JQuery入门—JQuery程序的代码风格详细介绍
2013/01/03 Javascript
Javascript跨域请求的4种解决方式
2013/03/17 Javascript
JS上传前预览图片实例
2013/03/25 Javascript
JavaScript对HTML DOM使用EventListener进行操作
2015/10/21 Javascript
理解javascript中的MVC模式
2016/01/28 Javascript
AngularJS自动表单验证
2016/02/01 Javascript
javascript实现数字倒计时特效
2016/03/30 Javascript
Node.js读写文件之批量替换图片的实现方法
2016/09/07 Javascript
微信小程序导航栏跟随滑动效果的实现代码
2019/05/14 Javascript
使用Vue实现简单计算器
2020/02/25 Javascript
使用python实现rsa算法代码
2016/02/17 Python
Python 闭包的使用方法
2017/09/07 Python
TensorFlow实现卷积神经网络CNN
2018/03/09 Python
tensorflow 输出权重到csv或txt的实例
2018/06/14 Python
Python 字符串与二进制串的相互转换示例
2018/07/23 Python
python图的深度优先和广度优先算法实例分析
2019/10/26 Python
Python调用.NET库的方法步骤
2019/12/27 Python
一款利用纯css3实现的win8加载动画的实例分析
2014/12/11 HTML / CSS
css 元素选择器的简单实例
2016/05/23 HTML / CSS
瑞典灯具和照明网上商店:Lamp24.se
2018/03/17 全球购物
Michael Kors英国官网:美国奢侈品品牌
2019/11/13 全球购物
惠普新加坡官方商店:HP Singapore
2020/04/17 全球购物
法学毕业生自我鉴定
2013/11/08 职场文书
初中三年学生的学习自我评价
2013/11/13 职场文书
财务人员求职自荐书范文
2014/02/10 职场文书
幼儿园元旦活动感言
2014/03/02 职场文书
会计专业应届生自荐信
2014/06/28 职场文书
毕业生代领毕业材料的授权委托书
2014/09/29 职场文书
运动会加油稿
2015/07/22 职场文书
python - timeit 时间模块
2021/04/06 Python
Mysql中 unique列插入重复值该怎么解决呢
2021/05/26 MySQL
SQL SERVER触发器详解
2022/02/24 SQL Server
Redis 的查询很快的原因解析及Redis 如何保证查询的高效
2022/03/16 Redis