pytorch本地图片数据集加载成字典
from torchvision.transforms import transformsimport pandas as pdfile_path=os.path.join(r'E:\python存储\leaves\images')Image_list=[]labels_list=[]a={}for i in range(0,18352):image_path=os.path.join(r'E:\
·
from torchvision.transforms import transforms import pandas as pd file_path=os.path.join(r'E:\python存储\leaves\images') Image_list=[] labels_list=[] a={} for i in range(0,18352): image_path=os.path.join(r'E:\python存储\leaves\images','{}.jpg'.format(i)) image=Image.open(image_path) trans=transforms.ToTensor() Image_list.append(trans(image)) train_label_read=pd.read_csv(r'E:\python存储\leaves\train.csv') train_label=train_label_read['label'] dataset=dict(map(lambda x,y:[x,y],train_label,Image_list)) print(dataset)

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)