/-MNIST-

基于卷积神经网络的Mnist数字手写数据集识别,并且支持手写输入进行推理

Primary LanguagePython

手写输入识别

基于pytorch

requirement

pip install -r requirements.txt

使用部分

在train.py部分实现了训练MNIST数据集,并且将训练的模型参数保存在本地

在inference.py部分实现了在手写板上绘制图像交由神经网络进行识别

(如果实在不想训练可以直接下载我这个参数,然后改一下路径运行inference.py进行推理