[pytorch]自己定义数据集Dataset
reason:开始研究语义分割(Semantic Segmentation)了,并且要修改论文的源代码,把它转成自己需要的代码。所以从定义自己数据集开始。本文利用的数据集是MNIST,并且仿照着MNIST代码定义自己的数据集。1、平时使用MNIST数据集(1)创建数据集对象DOWNLOAD_MNIST = False # 如果没下载数据集就设置为truetrain_data = torchvision.datasets.MNIST( root='./mnist', train=T
原创
2020-10-26 15:57:22 ·
4506 阅读 ·
0 评论