MNIST System

Description/软件描述

An assignment for my school course software engineering.

This is a simple system that allows users to upload a digital picture and then classify it using a neural network. The GUI system is built using PyQt5. The neural network is built using PyTorch and is trained on the MNIST dataset.

Training hyperparameters and models are customizable. Currently, the models provided are MLP, AlexNet, ResNet, and ViT.

Using netron to visualize the model. But ViT is currently not supported for visualization. (Tranformer_encoder layer cannot be exported using torch.onnx.export directly)

MySQL database is used only to store the user's information. (Required by the experimental task)

Users can set their own username and password.

一个学校课程软件工程的实验作业。 使用PyQt5构建的一个简单的GUI，允许用户上传一个数字图片，然后使用神经网络对其进行分类。 神经网络使用PyTorch构建，并在MNIST数据集上进行训练。

训练超参数与模型是可以自定义的。 模型目前提供了MLP,AlexNet,ResNet和ViT四种模型。

使用netron来可视化模型。 但是ViT目前不支持可视化。(Tranformer_encoder层无法使用torch.onnx.export直接导出)

使用MySQL数据库，仅用于存储用户的信息。(实验任务要求)

用户可以设置自己的用户名和密码

Environment/环境

• MySQL
• Python==3.11.5
• PyQt5
• PyQt5_sip
• PyQtWebEngine
• PyQtChart
• mysql_connector
• netron
• numpy
• onnx
• onnxruntime
• Pillow==9.3.0
• torch
• torchvision

Tips:

• Higher version of Pillow may not support the image convert for PyQt5.
• PyQtWebEngine and PyQtChart need to be installed independently.

Installation/安装

Install the required packages by running the following command:

pip install -r requirements.txt

Usage/使用

Run mysql.sql first to create the database and table.

Be sure there is at least one user!

Then run the following command to start the system with your configuration:

python  main.py 
        --host {your_db_host} 
        --user {your_db_user} 
        --password {your_db_password} 
        --database db_mnist_exp
        --login_without_info

Arguments:

• --host: The host of the MySQL database. Default is 'localhost'.
• --user: The user of the MySQL database. Default is 'root'.
• --password: The password of the MySQL database. Default is '123456'
• --database: The database name. Default is 'db_mnist_exp', which is created by mysql.sql
• --login_without_info: Log in without checking the database and ignore the login dialog input. If you just want to experience the system, you can use this argument.

Software Snapshots/截图

Login Dialog:

System MainWindow: