我在预测时,出现了报错,不知道怎么更改了?
报错信息:
import tkinter as tk
import tensorflow as tf
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
import numpy as np
from tkinter import *
import tkinter.filedialog # 注意次数要将文件对话框导入
from PIL import Image
from tensorflow.python.keras import models, layers
def ResNet_yuce():
global e1, pred, root, predict_class, k
name_dict = {"Uninfected": 0, "Parasitized": 1}
fig = plt.figure(figsize=(2, 2), dpi=100)
# 从本地选择一个文件,并返回文件的目录
# filenames = tkinter.filedialog.askopenfile()#形成一共io流 的文件名称
filename = tk.filedialog.askopenfilename(title=u'加载图片', initialdir=(())) # 只保留文件名和后缀
print(filename)
model_save_path = './ResNet50_test_tuxiangzengqiang_Skin_test.ckpt'
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(), # Flatten-变成一维数组
tf.keras.layers.Dense(128, activation='relu'), # relu函数可以解决 梯度消失问题 是线性的
tf.keras.layers.Dense(2, activation='softmax')]) # 2个类所以2个神经元 softmax只用在最后一层 可以让最后输出的概率在0-1之间
model.load_weights((model_save_path))#读取文件
# models = load_model("./ResNet50_test_tuxiangzengqiang_Skin_test.ckpt")
# 加载测试图片
img = Image.open(filename)
#
img = img.resize((128, 128), Image.ANTIALIAS)
# 将灰度图转化为RGB模式
img = img.convert("RGB")
plt.axis('off') # 去掉坐标轴
plt.imshow(img)
# 归一化
img1 = np.array(img) / 255.
# 将图片增加一个维度,目的是匹配网络模型
img1 = (np.expand_dims(img1, 0))
print(img1)
# 将预测结果转化为概率值
yuceceshi = model.predict(img1)
print(yuceceshi)
result = np.squeeze(yuceceshi)
print(result)
predict_class = np.argmax(result)
# pred = tf.argmax(result, axis=1)#求最大值得出最后识别出来的数
# print(inverse_dict[int(predict_class)],result[predict_class])
# 将预测的结果打印在图片上面
# plt.title([inverse_dict[int(predict_class)],result[predict_class]])
print(predict_class)
result1 = np.argmax(result) # 取出预测结果中概率最大的元素索引值
for k, v in name_dict.items(): # 将类别由数字转换为名称
if result1 == v: # 如果预测结果等于v, 打印出名称
print("预测结果:", k) # 打印出名称
pred = "预测结果:" + k
e1 = tk.Label(root, text=pred, font=('Arial', 16)) # 显示预测结果
e1.place(x=100, y=370)
canvas = FigureCanvasTkAgg(fig, master=root) # 渲染器 在画布上放置图片
canvas.draw()
canvas.get_tk_widget().place(x=100, y=150)
def picture():
global root
root = Tk()
root.wm_title("细胞图像疟疾检测")
root.geometry('1000x650')
bt = tk.Button(root, text="ResNet50", width=10, command=ResNet_yuce)
bt.place(x=210, y=90)
root.mainloop()
if __name__ == '__main__':
picture()