/MNIST-on-QT--for-teaching-demo

This project uses pyside6 to package a QT program for pytorch-based MNIST handwritten digit classification, which is suitable for demonstrating convolutional neural networks in a cognitive course

Primary LanguagePython

MNIST on QT, for teaching demo

This project uses pyside6 to package a QT program for pytorch-based MNIST handwritten digit classification, which is suitable for demonstrating convolutional neural networks in a cognitive course.

该项目使用pyside6打包了一个QT程序，用于基于pytorch的MNIST手写数字分类，适合在基础课程中演示卷积神经网络。

Program Performance

image

image

CNN model

The training file of the model is saved in . /res/CNN mnist.ipynb, and you can modify the model to improve the model classification results. The parameters of the trained model are saved in . /res/mnist_cnn.pth.

模型的训练文件保存在./res/CNN mnist.ipynb，你可以修改模型来改善模型分类效果。已训练的模型参数保存在./res/mnist_cnn.pth。