利用keras加载训练好的.H5文件,并实现预测图片
作者：Jaguar_95 发布时间：2021-01-18 21:18:57
标签：keras,H5文件,预测图片
我就废话不多说了，直接上代码吧！
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)
来源：https://blog.csdn.net/Jaguar_95/article/details/82970754
0
投稿猜你喜欢
- 大家经常用的是Adodb.Stream，但这时就有个缺陷，就是不支持断点续传了。经常看到flashget中是红脸（即不支持断点续传）其实支持
一、获取安装包：Pycharm 官网 下载页面 ：点击打开Anconda 官网 下载页面 ：点击打开选择对应的系统和需要的版本进行下载，py- 概述Object.freeze(obj)可以冻结一个对象。一个被冻结的对象再也不能被修改；冻结了一个对象则不能向这个对象添加新的属性，不能删
- 前言：在motplotlib的学习过程中，我们使用最多的就是numpy模块。numpy 模块被称为 matplotlib 模块绘制图表伴侣。
- 需求：1. 可输入代码，并且代码语法高亮2. 支持编辑和不可编辑模式3. 提交到后端到代码内容为字符串格式实现在gitbug上找到vue-p
- 前言：线性回归模型属于经典的统计学模型，该模型的应用场景是根据已知的变量（即自变量）来预测某个连续的数值变量（即因变量）。例如餐厅根据媒体的
- （1）更有意义的搜索 数据可被XML唯一的标识。没有XML，搜索软件必须了解每个数据库是如何构建的。这实际上是不可能的，因为每个数据库描述数
- 本次薯片会一改以往低调、沉稳之作风，先瑜伽团美女show上阵，再是以臭为首的吃喝团上阵，两轮下来大家情绪Hight到了极点，自然讨论的时候思
- python脚本实现xlsx文件解析，供大家参考，具体内容如下环境配置：1.系统环境：Windows 7 64bit 2.编译环境：Pyth
- Python爬虫：一些常用的爬虫技巧总结爬虫在开发过程中也有很多复用的过程，这里总结一下，以后也能省些事情。1、基本抓取网页get方法imp
- 概述我们在使用mysql的时候，尤其是在创建数据库的时候，我们需要选择当前数据库的字符集和排序规则，这样你库里面的创建数据表的时候默认的编码
- 前言前几天逛github发现了一个有趣的并发库-conc，其目标是：更难出现goroutine泄漏处理panic更友好并发代码可读性高从简介
- 前言今天制作的这一款能在B站能指定直播间、自动发弹幕的功能的脚本，因为没做那么多的功能，所以代码很简单，适合刚入门的同学学习先打开一个直播间
- Mysql的分页的两个参数select * from user limit 1,21表示从第几条数据开始查(默认索引是0,如果写1,从第二条
- 简介Python免费调用百度AI实现图片上面的文字识别步骤安装百度AI库!pip install baidu-aip注册百度AI开放平台先注
- 简介CountVectorizer官方文档。将一个文档集合向量化为为一个计数矩阵。如果不提供一个先验字典，不使用分析器做某种特征选择，那么特
- 先上需要用到的全部代码片段（截取） MenuControl.prototype.boxDisplay = false;//是否显示图层选择菜
- 写完“无序列表”和“有序列表”之后已经有人和我说这两篇看得没什么意思。这两篇文章如果只以单向读取的形式阅读那么的确是没什么意思，但是这两篇重
- 初次安装完PyCharm后，新建项目时，遇到了No Python interpreter selected的问题。意思是说没有找到Pytho
- 下面是我已经证实可用的自动备份的方法. 1、打开企业管理器->管理->sql server代理 2、新建一个作业,作业名称随便取
