pytorch建立自己的数据集（以mnist为例）

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

>>阅读原文<<