建设治安总队网站目的,电子商务网站建设及维护管理ppt,网站建设的种类,如何运行asp网站在查看pytorch官方文档的时候#xff0c;在这里链接中https://pytorch.org/tutorials/beginner/basics/data_tutorial.html的Creating a Custom Dataset for your files章节#xff0c;有提到要自定义数据集#xff0c;需要用到实际的图片和标签。
在网上找了半天没找到在这里链接中https://pytorch.org/tutorials/beginner/basics/data_tutorial.html的Creating a Custom Dataset for your files章节有提到要自定义数据集需要用到实际的图片和标签。
在网上找了半天没找到写了一个脚本将图片和标签文本下载到本地。
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor# Download training data from open datasets.
training_data datasets.FashionMNIST(rootdata,trainTrue,downloadTrue,transformToTensor(),
)# Download test data from open datasets.
test_data datasets.FashionMNIST(rootdata,trainFalse,downloadTrue,transformToTensor(),
)# 写入到本地
count0
for index,x in test_data:print(index.size(),x)countcount1classes [T-shirttop,Trouser,Pullover,Dress,Coat,Sandal,Shirt,Sneaker,Bag,Ankleboot,]import torchfrom torchvision.utils import save_imagefolder_path ./data/imageandlableTest # 替换为你的文件夹路径filename {}{}.jpg.format(classes[x],count) # 图片文件名# 确保文件夹存在import osif not os.path.exists(folder_path):os.makedirs(folder_path)# 保存图片save_path os.path.join(folder_path, filename)save_image(index, save_path)with open(./data/imageandlableTest/output.txt, a) as f: f.write({},{}\n.format(filename,x))print(count)
