Pytorch实现全连接神经网络模型和卷积神经网络训练MNIST数据集
Pytorch实现全连接神经网络模型/卷积神经网络训练MNIST数据集,并将训练好的模型在自己的手写图片数据集上测试
目录说明:
CNN文件夹是用来保存卷积神经网络模型代码,其中model.py,my_dataset.py是被自动调用的,都不需要运行
FC文件夹是用来保存全连接神经网络模型代码,其中model.py,my_dataset.py是被自动调用的,都不需要运行
dataset文件夹是保存MNIST官方数据集的文件夹,不需改动
images文件夹是用来保存REAEDME.md文件中引用的图片的,不需改动
my_mnist_dateset文件夹是用来保存自己手写数字图片与标签文件的,自己手写的图片请放在my_mnist_dateset/classify对应的文件夹中
make_ours_dataset.py文件是用来处理my_mnist_dateset文件夹下的图像并生成标签用的
requirements.txt文件是环境配置文件