今天小编就为大家分享一篇pytorch实现建立自己的数据集(以mnist为例)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

本文将原始的numpy array数据在pytorch下封装为Dataset类的数据集，为后续深度网络训练提供数据。 加载并保存图像信息 首先导入需要的库，定义各种路径。 import os import matplotlib from keras.datasets import mnist import numpy as np from torch.utils.data.dataset import Dataset from PIL import Image import scipy.mis