加载已径训练好的模型(使用.pb格式),实现手写数字识别界面的设计,多一些功能的界面比如可以识别单张图片单个数字和单张图片多个数字的识别,显示出完整的代码

时间: 2023-12-11 19:04:59 浏览: 32
首先,我们需要加载已经训练好的模型文件,这里我们选择使用 TensorFlow 的模型加载方式。在加载模型之前,我们需要先定义输入和输出节点的名称,这个可以在训练模型时指定。 ```python import tensorflow as tf # 定义输入和输出节点的名称 input_tensor_name = 'input:0' output_tensor_name = 'output:0' # 加载模型文件 model_path = 'model.pb' with tf.gfile.GFile(model_path, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) # 导入模型 with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name='') # 获取输入和输出节点 input_tensor = graph.get_tensor_by_name(input_tensor_name) output_tensor = graph.get_tensor_by_name(output_tensor_name) ``` 接下来,我们设计一个简单的图形界面,使用 tkinter 库实现。在界面中,我们可以选择“手写数字输入”或者“图片输入”两种方式进行识别。对于“手写数字输入”,我们提供了一个小画板,可以让用户手写数字。对于“图片输入”,我们可以从本地文件中选择一张图片进行识别。 ```python import tkinter as tk from tkinter import filedialog from PIL import Image, ImageTk # 创建主窗口 root = tk.Tk() root.title('手写数字识别') # 创建画板 canvas = tk.Canvas(root, width=200, height=200, bg='white') canvas.pack() # 创建工具栏 toolbar = tk.Frame(root) toolbar.pack(side=tk.TOP) # 创建“手写数字输入”按钮 button_draw = tk.Button(toolbar, text='手写数字输入') button_draw.pack(side=tk.LEFT) # 创建“图片输入”按钮 button_image = tk.Button(toolbar, text='图片输入') button_image.pack(side=tk.LEFT) # 创建状态栏 statusbar = tk.Label(root, text='请在画板中手写数字', bd=1, relief=tk.SUNKEN, anchor=tk.W) statusbar.pack(side=tk.BOTTOM, fill=tk.X) # 显示主窗口 root.mainloop() ``` 接下来,我们需要实现“手写数字输入”和“图片输入”两种方式的识别功能。对于“手写数字输入”,我们可以使用鼠标事件来实现用户在画板上手写数字的过程,并将得到的图像数据传入模型进行识别。对于“图片输入”,我们可以使用文件对话框来让用户选择一张图片文件,并将其转换成模型可以处理的格式后进行识别。 ```python import numpy as np # 定义手写数字输入的事件处理函数 def draw(event): x, y = event.x, event.y r = 8 canvas.create_oval(x-r, y-r, x+r, y+r, fill='black') image = Image.new('L', (200, 200), 255) draw = ImageDraw.Draw(image) draw.ellipse((x-r, y-r, x+r, y+r), fill=0) data = np.array(image).reshape(-1, 28, 28, 1) result = session.run(output_tensor, feed_dict={input_tensor: data}) statusbar.config(text='识别结果为:%d' % result[0]) # 定义图片输入的事件处理函数 def choose_image(): # 打开文件对话框选择图片文件 file_path = filedialog.askopenfilename(filetypes=[('Image Files', '*.png;*.jpg;*.jpeg')]) if file_path: # 加载图片并进行预处理 image = Image.open(file_path).convert('L').resize((28, 28), Image.ANTIALIAS) data = np.array(image).reshape(-1, 28, 28, 1) # 进行识别并显示结果 result = session.run(output_tensor, feed_dict={input_tensor: data}) statusbar.config(text='识别结果为:%d' % result[0]) # 显示图片 image = ImageTk.PhotoImage(image) canvas.create_image(100, 100, image=image) canvas.image = image # 绑定鼠标事件 canvas.bind('<B1-Motion>', draw) # 绑定按钮事件 button_draw.config(command=lambda: statusbar.config(text='请在画板中手写数字')) button_image.config(command=choose_image) ``` 最后,我们需要在界面中添加更多的功能,比如支持单张图片多个数字的识别。具体实现可以根据需要进行扩展。 完整代码如下: ```python import tensorflow as tf import tkinter as tk from tkinter import filedialog from PIL import Image, ImageDraw, ImageTk import numpy as np # 定义输入和输出节点的名称 input_tensor_name = 'input:0' output_tensor_name = 'output:0' # 加载模型文件 model_path = 'model.pb' with tf.gfile.GFile(model_path, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) # 导入模型 with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name='') # 获取输入和输出节点 input_tensor = graph.get_tensor_by_name(input_tensor_name) output_tensor = graph.get_tensor_by_name(output_tensor_name) # 创建主窗口 root = tk.Tk() root.title('手写数字识别') # 创建画板 canvas = tk.Canvas(root, width=200, height=200, bg='white') canvas.pack() # 创建工具栏 toolbar = tk.Frame(root) toolbar.pack(side=tk.TOP) # 创建“手写数字输入”按钮 button_draw = tk.Button(toolbar, text='手写数字输入') button_draw.pack(side=tk.LEFT) # 创建“图片输入”按钮 button_image = tk.Button(toolbar, text='图片输入') button_image.pack(side=tk.LEFT) # 创建状态栏 statusbar = tk.Label(root, text='请在画板中手写数字', bd=1, relief=tk.SUNKEN, anchor=tk.W) statusbar.pack(side=tk.BOTTOM, fill=tk.X) # 定义手写数字输入的事件处理函数 def draw(event): x, y = event.x, event.y r = 8 canvas.create_oval(x-r, y-r, x+r, y+r, fill='black') image = Image.new('L', (200, 200), 255) draw = ImageDraw.Draw(image) draw.ellipse((x-r, y-r, x+r, y+r), fill=0) data = np.array(image).reshape(-1, 28, 28, 1) result = session.run(output_tensor, feed_dict={input_tensor: data}) statusbar.config(text='识别结果为:%d' % result[0]) # 定义图片输入的事件处理函数 def choose_image(): # 打开文件对话框选择图片文件 file_path = filedialog.askopenfilename(filetypes=[('Image Files', '*.png;*.jpg;*.jpeg')]) if file_path: # 加载图片并进行预处理 image = Image.open(file_path).convert('L').resize((28, 28), Image.ANTIALIAS) data = np.array(image).reshape(-1, 28, 28, 1) # 进行识别并显示结果 result = session.run(output_tensor, feed_dict={input_tensor: data}) statusbar.config(text='识别结果为:%d' % result[0]) # 显示图片 image = ImageTk.PhotoImage(image) canvas.create_image(100, 100, image=image) canvas.image = image # 绑定鼠标事件 canvas.bind('<B1-Motion>', draw) # 绑定按钮事件 button_draw.config(command=lambda: statusbar.config(text='请在画板中手写数字')) button_image.config(command=choose_image) # 创建会话 session = tf.Session(graph=graph) # 运行主窗口 root.mainloop() # 关闭会话 session.close() ```

相关推荐

最新推荐

recommend-type

tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例

升级到tf 2.0后, 训练的模型想转成1.x版本的.pb模型, 但之前提供的通过ckpt转pb模型的方法都不可用(因为保存的ckpt不再有.meta)文件, 尝试了好久, 终于找到了一个方法可以迂回转到1.x版本的pb模型. Note: 本方法首先...
recommend-type

如何使用C#将Tensorflow训练的.pb文件用在生产环境详解

主要给大家介绍了关于如何使用C#将Tensorflow训练的.pb文件用在生产环境的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考借鉴,下面随着小编来一起学习学习吧
recommend-type

将keras的h5模型转换为tensorflow的pb模型操作

主要介绍了将keras的h5模型转换为tensorflow的pb模型操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
recommend-type

tensorflow模型文件(ckpt)转pb文件的方法(不知道输出节点名)

主要介绍了tensorflow模型文件(ckpt)转pb文件(不知道输出节点名),本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
recommend-type

将tensorflow模型打包成PB文件及PB文件读取方式

今天小编就为大家分享一篇将tensorflow模型打包成PB文件及PB文件读取方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

spring添加xml配置文件

1. 创建一个新的Spring配置文件,例如"applicationContext.xml"。 2. 在文件头部添加XML命名空间和schema定义,如下所示: ``` <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.springframework.org/schema/beans
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。