利用keras加載訓練好的.H5文件,并實現(xiàn)預測圖片
更新時間：2020年01月24日 10:05:30 作者：Jaguar_95
今天小編就為大家分享一篇利用keras加載訓練好的.H5文件,并實現(xiàn)預測圖片，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了，直接上代碼吧！
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()
#規(guī)范化圖片大小和像素值
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)
#調(diào)用函數(shù)，規(guī)范化圖片
pre_x = get_inputs(images)
#預測
pre_y = model.predict(pre_x)
print(pre_y)
以上這篇利用keras加載訓練好的.H5文件,并實現(xiàn)預測圖片就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
在本篇文章里，小編給大家分享的是關于Python安裝第三方模塊的方法及實例代碼，需要的朋友們可以學習下。2020-05-05
python sklearn包——混淆矩陣、分類報告等自動生成方式
今天小編就為大家分享一篇python sklearn包——混淆矩陣、分類報告等自動生成方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
Flask處理POST請求的過程是一個相對直接且靈活的過程,它允許開發(fā)者根據(jù)需求接收、處理并響應客戶端發(fā)送的數(shù)據(jù),以下是對Flask如何處理POST請求的詳細簡述,需要的朋友可以參考下2024-10-10
這篇文章主要給大家介紹了關于Python開發(fā)入門之如何制作一個簡單的桌面應用的相關資料,我們不僅可以使用Python的圖像處理庫，如PIL等來實現(xiàn)圖片的處理和識別,同時你還可以設計和開發(fā)具有圖形界面的桌面應用程序,需要的朋友可以參考下2023-08-08
這篇文章主要介紹了Python設計模式之適配器模式原理與用法,結(jié)合實例形式分析了適配器模式的相關概念、原理及Python實現(xiàn)技巧,需要的朋友可以參考下2019-01-01
這篇文章主要給大家介紹了關于用python如何打印九九乘法表的相關資料,包括初級版和完整版,初級版通過雙層循環(huán)輸出每個乘法表的行,完整版通過調(diào)整內(nèi)層循環(huán)的范圍,使得每行的乘法表輸出更加規(guī)范,文章還提供了代碼解釋和一些小技巧,需要的朋友可以參考下2024-11-11
本文主要介紹了tensorflow中Dense函數(shù)的具體使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-03-03
在本篇文章里小編給大家分享的是關于ipad上運行python的方法步驟以及相關知識點，有需要的朋友們學習下。2019-10-10
字符串是python中最常見的數(shù)據(jù)類型,創(chuàng)建字符串只需用引號括起來分配給一個變量即可,這篇文章主要給大家介紹了關于Python字符串的基本操作方法的相關資料,文中通過圖文介紹的非常詳細,需要的朋友可以參考下2023-11-11
這篇文章主要介紹了Python文件去除注釋的方法,涉及Python正則表達式及文件操作的相關技巧,需要的朋友可以參考下2015-05-05