利用keras加载训练好的.H5文件,并实现预测图片
作者:Jaguar_95 时间:2021-01-18 21:18:57
我就废话不多说了,直接上代码吧!
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
标签:keras,H5文件,预测图片
0
投稿
猜你喜欢
Python操作Excel神器openpyxl使用教程(超详细!)
2021-02-28 21:58:35
python获取linux系统信息的三种方法
2022-04-13 06:27:55
Python xpath表达式如何实现数据处理
2021-09-02 22:07:15
python-web根据元素属性进行定位的方法
2021-05-04 04:22:35
JSP学生信息管理系统设计
2023-07-13 03:37:30
PHP的PDO预处理语句与存储过程
2023-06-17 11:49:59
网页布局设计基础
2008-10-16 13:58:00
Python pip 安装与使用(安装、更新、删除)
2022-07-30 01:58:19
python @classmethod 的使用场合详解
2023-08-02 20:50:35
Search File Contents PHP 搜索目录文本内容的代码
2023-11-24 08:09:40
ASP程序中调用函数Now()显示上午下午的问题
2009-08-27 13:09:00
讲解数据库管理系统必须提供的基本服务
2009-01-04 14:33:00
Dreamweaver MX 2004表格设计
2008-02-03 11:36:00
JavaScript中的私有成员 Javascript教程
2008-12-02 17:57:00
如何让python的运行速度得到提升
2023-04-26 21:48:35
python中heapq堆排算法的实现
2022-10-13 23:11:40
Python实战实现爬取天气数据并完成可视化分析详解
2022-01-04 09:33:44
自定义Django_rest_framework_jwt登陆错误返回的解决
2021-02-27 22:07:32
asp如何读取Access数据库的表名?
2009-11-18 20:48:00
Python中extend和append的区别讲解
2021-03-28 04:20:37