本文主要是介绍mninst数据集图片下载,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
//不需要在官网下载,直接通过python代码,利用pytorch下载即可
from icecream import ic
from torchvision import datasets
from tqdm import tqdm
import ostrain_data = datasets.MNIST(root="./data/", train=True, download=True)
test_data = datasets.MNIST(root="./data/", train=False, download=True)
saveDirTrain = './DataImages-Train'
saveDirTest = './DataImages-Test'if not os.path.exists(saveDirTrain):os.mkdir(saveDirTrain)
if not os.path.exists(saveDirTest):os.mkdir(saveDirTest)ic(len(train_data), len(test_data))
ic(train_data[0])
ic(train_data[0][0])def save_img(data, save_path):for i in tqdm(range(len(data))):img, label = data[i]img.save(os.path.join(save_path, str(i) + '-label-' + str(label) + '.png'))save_img(train_data, saveDirTrain)
save_img(test_data, saveDirTest)
这篇关于mninst数据集图片下载的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!