利用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
投稿猜你喜欢
- 本文实例讲述了pytorch制作自己的LMDB数据操作。分享给大家供大家参考，具体如下：前言记录下pytorch里如何使用lmdb的code
- 前言综合应用已经学习过的知识点：变量流程控制函数模块系统需求1.程序启动，显示名片管理系统欢迎界面，并显示功能菜单*************
目录一、简介思路高德地图API二、获取经纬度三、路线规划（四种方式）获取出行路线数据处理四、演示效果五、结尾一、简介路径规划中包括步行、公交- 这个阶段一直在做和梯度一类算法相关的东西，索性在这儿做个汇总：一、算 * 述梯度下降法(gradient descent)别名最速
- 一、生产者消费者模型介绍1.1 为什么需要使用生产者消费者模型生产者是指生产数据的任务，消费者是指消费数据的任务。当生产者的生产能力远大于消
- Python中str is not callable问题详解及解决办法问题提出： 在Python的代码，在运行过程中
- 如何用javascript来判定选择单选的radio哪个被选择,多选的checkbox有那几个被选中? <!DOCTYPE
- import urllib.parse,os.path,time,sysfrom http.client import HTTPSConne
- 冷备份是数据库文件的物理备份，通常在数据库通过一个shutdown normal或shutdown immediate 命令正常关闭后进行。
- pygame绘制机制简介 屏幕控制 pygame.display• 用来控制Pygame游戏的屏幕• Pygame有且只有一个屏幕
- 不用库，写了很久，一直出bug，到网上一搜，可以直接输入之后，eval(str)即可得到结果！eval程序如下：s=input("
- 本文实例讲述了js捐赠管理完整实现方法。分享给大家供大家参考。具体实现方法如下：index.html页面如下：<!DOCTYPE ht
- 一、Python下载安装1、Python官方下载地址：https://www.python.org/downloads官方下载速度太慢，你可
- MVVM模式不但可用于Form表单，在复杂的管理页面中也能大显身手。例如，分页显示Blog的功能，我们先把后端代码写出来：在apis.py中
- 前言今天呢，笔者想和大家聊聊python+pytest接口自动化中将代码进行封装，只有将测试代码进行封装，才能被测试框架识别执行。例如单个接
- 栅格就是你对页面版式的规划你日常所见的许多页面都有栅格存在。你可能注意不到，但它确实存在，并且支撑着设计内容，建立整体的架构，引导着页面的元
- 在实现鼠标停在某元素上，会出现提示信息，以前我们太多是采用javascript脚本代码来实现。看了怿飞的IE中伪类:hover的使用及BUG
- pandas可以对不同索引的对象进行算术运算，如果存在不同的索引对，结果的索引就是该索引对的并集。一、算术运算a、series的加法运算 &
- 这几个技术都不是很新了,现在还拿出来说也就是为了教教新手了呵呵,各位大虾多多指正.css之所以出现就是为了实现数据与数据的表现形式的分离,
- 1.self只有在类的方法中才会有，其他函数或方法是不必带self的。2.在调用时不必传入相应的参数。3.在类的方法中(如__init__)
