QMNIST数据集运用

1、准备数据集github   ：https://github.com/facebookresearch/qmnist（如果下载的压缩包，自行解压）这里面只包含一个qmnist.py的文件，只是方便用户使用数据集，而没有训练模型。

qmnist-master文件

2import时遇到一个问题，貌似只有同级文件之间可以import，因此需要将qmnist.py与你的原来的mnist程序放在同一个文件夹之下。（我的是main.py）

在程序开头需要导入from qmnist import QMNIST

---

github官网也给出了使用部分方法：

from qmnist import QMNIST

qtrain=QMNIST('_qmnist',train=True,download=True)

qtest=QMNIST('_qmnist',train=False)

qtest10k=QMNIST('_qmnist',what='test10k',compat=False,download='True')

qall=QMNIST('_qmnist',what='nist',compat=False)

---

原来main主函数的数据加载器部分函数

直接修改数据加载器部分可以少修改参数，直接从源头替换数据集。

修改的部分：datasets去掉，MNIST改为QMNIST，后面文件的路径。

其中torch.utils.data.DataLoader为torch的数据批量抽取，其参数有：数据集对象（Dataset）、batc_size、shuffle（设置为真不会按数据顺序输出，一般在训练数据中使用）、num_workers（设置多少个子进程可以使用，设置0表示在主进程中使用）

本文参考https://blog.csdn.net/geter_CS/article/details/83378786