原创 Pytorch學習筆記(二)自己加載單通道圖片用作數據集訓練

      作者:灰色橡皮擦      pytorch 在torchvision包裏面有很多的的打包好的數據集,例如minist,Imagenet-12,CIFAR10 和CIFAR100。在torchvision的dataset包裏面,