上传者: 46846685
|
上传时间: 2026-01-17 13:40:36
|
文件大小: 5KB
|
文件类型: PY
本资源为手写数字识别分类的入门级实战代码,代码使用pytorch架构编写,并且无需显卡,只通过CPU进行训练。
代码编写了一个简单的卷积神经网络,输入为单通道的28×28图片,输出是一个10维向量。
数据集的格式应在代码文件同目录下包含两个文件夹,分别为训练文件夹和测试文件夹,训练和测试文件夹下各包含10个以0~9数字命名的文件夹,文件夹中包含了对应的若干张图片文件。
代码在每轮训练结束后会输出训练集分类正确率和测试集分类正确率,并且记录在txt文件中。