利用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 调用DLL操作抄表机
Jan 12 Python
python基础教程之循环介绍
Aug 29 Python
pymongo实现多结果进行多列排序的方法
May 16 Python
Python操作使用MySQL数据库的实例代码
May 25 Python
Python参数解析模块sys、getopt、argparse使用与对比分析
Apr 02 Python
详解python 爬取12306验证码
May 10 Python
django 2.2和mysql使用的常见问题
Jul 18 Python
django 单表操作实例详解
Jul 30 Python
基于Python和PyYAML读取yaml配置文件数据
Jan 13 Python
解决keras GAN训练是loss不发生变化,accuracy一直为0.5的问题
Jul 02 Python
python如何建立全零数组
Jul 19 Python
Django url 路由匹配过程详解
Jan 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中设置、使用、删除Cookie的解决方法
2013/05/06 PHP
PHP入门之常量简介和系统常量
2014/05/12 PHP
PHP中imagick函数的中文解释
2015/01/21 PHP
2007/12/23更新创意无限,简单实用(javascript log)
2007/12/24 Javascript
web 页面分页打印的实现
2009/06/22 Javascript
javascript URL编码和解码使用说明
2010/04/12 Javascript
jquery的$(document).ready()和onload的加载顺序
2010/05/26 Javascript
extjs 如何给column 加上提示
2014/07/29 Javascript
JavaScript控制网页平滑滚动到指定元素位置的方法
2015/04/17 Javascript
js简单实现标签云效果实例
2015/08/06 Javascript
jquery插件之文字间歇自动向上滚动效果代码
2016/02/25 Javascript
对jquery的ajax进行二次封装以及ajax缓存代理组件:AjaxCache详解
2016/04/11 Javascript
老生常谈JavaScript 函数表达式
2016/09/01 Javascript
Angular1.x自定义指令实例详解
2017/03/01 Javascript
View.post() 不靠谱的地方你知道多少
2017/08/29 Javascript
基于匀速运动的实例讲解(侧边栏,淡入淡出)
2017/10/17 Javascript
基于Vue实现可以拖拽的树形表格实例详解
2018/10/18 Javascript
每个 JavaScript 工程师都应懂的33个概念
2018/10/22 Javascript
JS 验证码功能的三种实现方式
2018/11/26 Javascript
JavaScript中引用vs复制示例详析
2018/12/06 Javascript
javascript实现蒙版与禁止页面滚动
2020/01/11 Javascript
vue $mount 和 el的区别说明
2020/09/11 Javascript
Vue 使用iframe引用html页面实现vue和html页面方法的调用操作
2020/11/16 Javascript
对python PLT中的image和skimage处理图片方法详解
2019/01/10 Python
PyCharm中如何直接使用Anaconda已安装的库
2020/05/28 Python
python 用opencv实现霍夫线变换
2020/11/27 Python
css3动画过渡实现鼠标跟随导航效果
2018/02/08 HTML / CSS
澳洲本土太阳镜品牌:Quay Australia
2019/07/29 全球购物
远程学习的教学用品和家庭学习资源:Really Good Stuff
2020/04/27 全球购物
优秀管理者获奖感言
2014/02/17 职场文书
大学生村官演讲稿
2014/04/25 职场文书
运动会标语
2014/06/21 职场文书
春节超市活动方案
2014/08/14 职场文书
党在我心中的演讲稿
2014/09/13 职场文书
2016年国陪研修感言
2015/11/18 职场文书
2016教师节感恩话语
2015/12/09 职场文书