站点图标 AI技术聚合

用Python实现BP神经网络识别MNIST手写数字数据集(带GUI)

用Python实现BP神经网络识别MNIST手写数字数据集(带GUI)

概述

计算机神经网络则是人工智能中最为基础的也是较为重要的部分,它使用深度学习的方式模拟了人的神经元的工作,是一种全新的计算方法。本文的目标就是通过学习神经网络的相关知识,了解并掌握BP神经网络的实现原理和构造方法,建立一个简单的BP神经网络,并用MNIST数据集训练该网络,使训练后的网络能够成功的分类出MNIST测试数据集上的数字,并能识别从文件中读入的图片上的数字。

开发环境

CPU:英特尔 Core i7-7700HQ
GPU :Nvidia GeForce GTX 1060
内存:16GB

操作系统:Windows 10 x64
开发环境:PyCharm Community Edition 2020.2 x64 + Python3.8

需要我们做的功能

总体目标:
构建一个简单的BP神经网络,让这个网络可以被训练,可以测试数据,并实现识别用户打开的图片中的数字
功能要求:

  1. 可以自己定义神经网络的输入层、隐藏层、学习率、训练生成等参数
  2. 网络可以训练和测试
  3. 可以从文件夹中读取图片并进行判断和分类

非功能性要求:

  1. 需要图形交互,使操作更容易
  2. GUI界面设计应该简单明了
  3. 测试数据的识别准确率更高

BP网络设计

反向传播算法,即Back Propagation是建立在梯度下降算法基础上,适用多层神经网络的参数训练方法。由于隐藏层节点的预测误差无法直接计算,因此,反向传播算法直接利用输出层节点的预测误差反向估计上一层隐藏节点的预测误差,即从后往前逐层从输出层把误差反向传播到输入层,从而实现对链接权重调整,这也是反向传播算法名称的由来。
一个典型的3层BP神经网络模型如下图所示:


Mnist数据集的测试图片像素是28X28的,所以输入节点的个数就是28X28=784;识别出的数字有0-9十个数字,所以输出的节点的个数设置为10个;因为输入层的节点较多,所以隐藏层的节点个数设置为100;
考虑到梯度下降算法能够较好的消除产生的误差,所以激活函数设置为sigmoid函数;学习率设置为0.2,太高或太低都会导致不同的问题(梯度爆炸、梯度消失);训练世代设置为5个世代
因此,网络设计的参数为:

输入层节点数:784;
隐藏层节点数:200;
输出层节点数:10
学习率:0.1;
训练世代:5;
激活函数:sigmoid函数

代码

import PIL
import numpy as np
import pandas as pd
import imageio
from PIL import Image, ImageTk # 导入图像处理函数库
import tkinter as tk
from tkinter import constants, ttk
from tkinter import filedialog   #导入文件对话框函数库


#————————————————————————神经网络构建,三层结构——————————————————#
#激活函数
def sigmoid(x):
    return 1.0 / (1.0 + np.exp(-x))
#定义神经网络函数
class neuralNetwork:
    #初始化神经网络
    def __init__(self,inputnodes,hiddennodes,outputnodes,learnrate):
        #设立每个神经网络的输入、隐藏、输出层的节点数
        self.inodes = inputnodes
        self.hnodes = hiddennodes
        self.onodes = outputnodes
        #设置学习率
        self.lrate = learnrate
        self.wi_h = (np.random.rand(self.hnodes,self.inodes)-0.5)
        self.wh_o = (np.random.rand(self.onodes,self.hnodes)-0.5)

        pass
    #训练神经网络
    def train(self,inputs_list,targets_list):
        #输入与标准结果
     inputs = np.array(inputs_list,ndmin=2).T
     targets= np.array(targets_list, ndmin=2).T
     #计算隐藏层的信号值
     hidden_inputs = np.dot(self.wi_h,inputs)
     hidden_outputs = sigmoid(hidden_inputs)
     #计算输出层的信号值
     outputs_inputs  = np.dot(self.wh_o, hidden_outputs)
     outputs_outputs = sigmoid(outputs_inputs)
    #计算误差:精确值-实际值
     output_errors = targets - outputs_outputs
     hidden_errors = np.dot(self.wh_o.T,output_errors)
     #根据公式得出的表达式,直接用
     self.wh_o += self.lrate * np.dot((output_errors*outputs_outputs*(1.0-outputs_outputs)),
                 np.transpose(hidden_outputs))
     self.wi_h += self.lrate * np.dot((hidden_errors * hidden_outputs * (1.0 - hidden_outputs)),
                 np.transpose (inputs))

    pass
        #接受输入,返回输出
        #将输出进行激活,归一化
    def query(self,input_list):
        inputs = np.array(input_list,ndmin=2).T
        hidden_inputs = np.dot(self.wi_h, inputs)
        hidden_outputs = sigmoid(hidden_inputs)
        outputs_inputs = np.dot(self.wh_o,hidden_outputs)
        outputs_outputs = sigmoid(outputs_inputs)
        return outputs_outputs

        pass
    pass


这部分对BP神经网络类进行了参数的定义和对训练、激活函数进行了定义

#——————————————初始化GUI界面——————————————--#
window = tk.Tk()
window.title('神经网络识别MNIST数据集')
window.geometry('600x500')
global img_png  # 定义全局变量 图像的
var = tk.StringVar()  # 这时文字变量储存器
text = tk.Text(window,width=20,height=17)
text.pack(fill=tk.X,side=tk.BOTTOM)
text.insert(tk.END, '请输入相关数据,构建一个网络\n')

def craet_BPNN():
    global n
    global input_nodes
    global hidden_nodes
    global output_nodes
    global learning_rate
    global epochs
    global training_data_list
#—————————————创建神经网络对象并用数据集训练网络——————————#
# 输入、隐藏、输出节点数

    input_nodes =int(var_inputs.get())
    hidden_nodes= int(var_hidden.get())
    output_nodes = int( var_outputs.get())
    # 学习率
    learning_rate = float(var_lrate.get())
    epochs = int(var_epochs.get())
    # 创建神经网络对象
    n = neuralNetwork(input_nodes, hidden_nodes, output_nodes, learning_rate)
    text.insert(tk.END, 'BP网络构建成功!\n')
    text.insert(tk.END, '输入层节点数:'+var_inputs.get()+',隐藏层节点数:'+var_hidden.get()+',输出层节点数:'+var_outputs.get()+'\n')
    text.insert(tk.END, '学习率:' + var_lrate.get() + ',训练世代:' + var_epochs.get()+ '\n')
    text.insert(tk.END, '可以开始训练了!\n')
    #加载mnist数据集
    training_data_file = open("C:\\Users\\EASKWON\\Desktop\\mnist_train_100.csv", 'r')
    training_data_list = training_data_file.readlines()
    training_data_file.close()

上述代码的目的是使用函数创建神经网络对象并加载数据集

#开始训练函数,训练MNist数据集
def beg_train():
     for e in range(epochs):#   训练的世代,一次训练完成表示训练一个世代
            print("训练中,第", e, "个世代")
            text.insert(tk.END, '训练中,第' + str(e) + '个世代\n')
            text.update()
            t=0
            for record in training_data_list:
                t+=1
                print("已训练",t,"个数据")
            # 用”,“来区分数据
                all_values = record.split(',')
            # 将输入缩放和转换
                inputs = (np.asfarray(all_values[1:]) / 255.0 * 0.99) + 0.01
            # 将目标的输出值的0改为0.01,1改为0.99
                targets = np.zeros(output_nodes) + 0.01
                targets[int(all_values[0])] = 0.99
                n.train(inputs, targets)
                pass


     text.insert(tk.END, '训练完毕!\n')
     text.insert(tk.END, '可以开始测试你的网络了!\n')
pass

#       打开测试数据集MNIST-test

# 开始测试函数,遍历所有测试集中的测试数据,得出准确率

#————————————————————————测试MNIst数据集————————————————————#
def beg_test():
    global test_data_list
   #数据集的文件路径由自己定义,这是我自己的路径
    test_data_file = open("C:\\Users\\EASKWON\\PycharmProjects\\BpNetwork\\mnist_test_10.csv", 'r')
    test_data_list = test_data_file.readlines()
    test_data_file.close()
    all_values = test_data_list[0].split(',')
    print(all_values[1])
    n.query((np.asfarray(all_values[1:])/255.0*0.99)+0.01)
    #用来存放分数,即正确率
    scorecard = []
    text.insert(tk.END, '开始测试MNIST数据集.....\n')
    for record in test_data_list:
                #用”,“号分开数据
                all_values = record.split(',')
                # 用准确值标签记录数字准确值
                correct_label = int(all_values[0])
                print("---------")
                print("正确结果",correct_label)
                # 缩放
                inputs = (np.asfarray(all_values[1:]) / 255.0 * 0.99) + 0.01
                # 计算输出
                outputs = n.query(inputs)
                # 输出的最大值即为判断值
                label = np.argmax(outputs)
                print( "神经网络判断",label)
                # 将正确和错误的判断形成一个列表
                if (label == correct_label):
                    # 正确为1
                    scorecard
                    scorecard.append(1)
                else:
                    # 错误为0
                    scorecard
                    scorecard.append(0)
    print(scorecard)
    scorecard_array = np.asarray(scorecard)
    #正确率
    right_rate = (scorecard_array.sum() / scorecard_array.size) * 100

    text.insert(tk.END, '数据测试完毕\n')
    text.insert(tk.END, '正确率='+str(right_rate)+'%\n')
    text.update()
    print("正确率= ", right_rate, "%")
    pass

上述代码的作用是实现数据集的测试功能。它可以遍历测试集中的所有测试图片,得到最终的正确率。
至此,其实我们已经实现了识别手写数字的功能,我们的目标也就完成了。下一步是改进它并使其更实用。

#打开图片的函数,并尝试识别自己的图片
#编写GUI,让其更容易交互
def Open_Img():
        global img_png
        global path
        global label_Img
        OpenFile = tk.Tk()  # 创建新窗口
        OpenFile.withdraw()
        file_path = filedialog.askopenfilename()
        print("训练已结束,开始测试图片")
        text.insert(tk.END, '开始测试图片\n')
        path=file_path
        Img =Image.open(file_path)
        img_png = ImageTk.PhotoImage(Img)
        label_Img = tk.Label(window, image=img_png)
        Label_Show = tk.Label(window, image=img_png,
                              # 使用 textvariable 替换 text, 因为这个可以变化
                              bg='white', font=('Arial', 12), width=60, height=60)
        Label_Show.place(x=80, y=80)
        var.set('图像已打开')
        #自己图片的数据存放在这里
        our_own_dataset = []
        image_file_name = path
        print("加载中 ... ", image_file_name)
        text.insert(tk.END, '加载中....'+image_file_name+'\n')
        # 用文件名来设置准确值标签
        label = int(image_file_name[20])
        #将图片转换为数组
        img_array = imageio.imread(image_file_name, as_gray=True)
        # 将图片从28X28的数组转换成长为784的array
        img_data = img_array.reshape(784)
        # 缩放灰度值为0-1范围内
        img_data = (img_data / 255.0 * 0.99) + 0.01
        print("图像最小值为",np.min(img_data))
        print("图像最大值为",np.max(img_data))
        # 将标签值放到数组第一个
        record = np.append(label, img_data)
        our_own_dataset.append(record)
        item = 0
        correct_label = our_own_dataset[item][0]
        # 将转换值作为输入
        inputs = our_own_dataset[item][1:]
        # 计算网络的输出
        outputs = n.query(inputs)
        print("输出节点的输出为:",outputs)
        text.insert(tk.END, '输出节点的输出为\n'+str(outputs)+'\n')
        text.update()
        # 最高输出值所在的数字作为识别标签
        label = np.argmax(outputs)
        print("神经网络说:“它是", label, "”")
        text.insert(tk.END, '神经网络认为图中的数字是' + str(label) + '\n')
        text.see(tk.END)
        if (label == correct_label):
            print("恭喜你,匹配成功!")
            text.insert(tk.END, '恭喜你,识别成功了!\n')
        else:
            print("很遗憾,识别失败了")
            text.insert(tk.END, '很遗憾,识别失败了!再试一次吧\n')
        pass

#显示图片的函数
def SHOW():
    global img_png
    Label_Show = tk.Label(window, image=img_png,
                          # 使用 textvariable 替换 text, 因为这个可以变化
                          bg='white', font=('Arial', 12), width=60, height=60)
    Label_Show.place(x=80, y=80)

pass


img_frame = tk.LabelFrame(window, text='图像显示', padx=10, pady=10,
                       width=120,height=120)
img_frame.place(x=55,y=50)
# 创建文本窗口,显示当前操作8状态
in_lable=tk.Label(window,text='输入层节点数:')
in_lable.pack()
in_lable.place(x=300,y=40)

var_inputs=tk.StringVar()
var_inputs.set('784')
entry_inputs=tk.Entry(window,textvariable=var_inputs,width=10)
entry_inputs.place(x=380,y=40)

hi_lable=tk.Label(window,text='隐藏层节点数:')
hi_lable.pack()
hi_lable.place(x=300,y=70)

var_hidden=tk.StringVar()
var_hidden.set('50')
entry_hidden=tk.Entry(window,textvariable=var_hidden,width=10)
entry_hidden.place(x=380,y=70)

out_lable=tk.Label(window,text='输出层节点数:')
out_lable.pack()
out_lable.place(x=300,y=100)

var_outputs=tk.StringVar()
var_outputs.set('10')
entry_outputs=tk.Entry(window,textvariable=var_outputs,width=10)
entry_outputs.place(x=380,y=100)

rate_lable=tk.Label(window,text='学习率:')
rate_lable.pack()
rate_lable.place(x=300,y=130)

var_lrate=tk.StringVar()
var_lrate.set('0.1')
entry_lrate=tk.Entry(window,textvariable=var_lrate,width=10)
entry_lrate.place(x=380,y=130)

epochs_lable=tk.Label(window,text='训练世代:')
epochs_lable.pack()
epochs_lable.place(x=300,y=160)

var_epochs=tk.StringVar()
var_epochs.set('5')
entry_epochs=tk.Entry(window,textvariable=var_epochs,width=10)
entry_epochs.place(x=380,y=160)

#训练数据集按钮
btn_train = tk.Button(window,text='构建网络',width=15, height=2,
                         command=craet_BPNN)
btn_train.pack()
btn_train.place(x=30,y = 210)
#测试数据集按钮
btn_test = tk.Button(window,text='训练数据集',width=15, height=2,
                         command=beg_train)
btn_test.pack()
btn_test.place(x=170,y=210)
# 创建打开图像按钮
btn_Open = tk.Button(window,
                         text='测试数据集',  # 显示在按钮上的文字
                         width=15, height=2,
                         command=beg_test)  # 点击按钮式执行的命令
btn_Open.pack()
# 按钮位置
btn_Open.place(x=310,y=210)
# 创建显示图像按钮
btn_Show = tk.Button(window,
                         text='打开测试图片',  # 显示在按钮上的文字
                         width=15, height=2,
                         command=Open_Img)  # 点击按钮式执行的命令

btn_Show.pack()
# 按钮位置
btn_Show.place(x=450,y=210)
# 运行整体窗口
window.mainloop()
pass

以上代码实现了GUI的编写,让用户自行从文件中读取图片并识别,并由网络给出识别结果。因为GUI比较繁琐,所以代码看起来偏长。不过和最终效果比起来,这点付出是值得的。我们的功能也已经实现完毕了,接下来就看一下具体效果了。

达到效果


这是程序运行时的界面,可以自定义参数的输入

在Pycharm的控制台界面也能看到每张图片的识别结果,1是识别正确,0是识别错误。可以看到识别的正确率还是挺高的,由97%。

这里是用户自行选择图片进行识别,可以自己写然后进行识别,但前提是图像尺寸必须是28X28,如果尺寸打了就必须对其进行池化到28X28的大小,否则就会导致输入参数量巨大(几万乃至几十万个输入参数),那就不是BP神经网络可以解决的问题了,就必须要用到卷积深度神经网络进行特征提取再来分类了。

可以看出,我们搭建的神经网络已经成功识别了我们的手写数字,我们的目标也成功了!
至此,我们已经成功建立了神经网络,可以说实现了最基本的人工智能。但要明白,这只是人工智能最基础的部分,要实现真正强大的人工智能,我们还有很长的路要走。但现阶段,无论是复杂的还是简单的神经网络都是从这个网络衍生出来的,所谓的千变万化是分不开的。万丈高楼从地面出发,一步一个脚印,总有一天会到达我们心中的目标。

版权声明:本文为博主宋昆昆昆伟easkwon原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/chook_you_too_b/article/details/123030895

退出移动版