小土堆pytorch 第一天
一、两大法宝函数1、dir()打开,看见 2、help()查看说明书 二、三个运行方式的区别 三、如何导入数据两种数据形式:Dataset、Dataloader Dataset1、如何获取每一个数据及其label? 2、总共有多少条数据? 12345678910111213141516171819202122232425262728293031from torch.utils.data import Datasetfrom PIL import Imageimport os # 用于获取图片的地址class MyData(Dataset): def __init__(self, root_dir, label_dir): self.root_dir = root_dir self.label_dir = label_dir self.path = str(os.path.join(self.root_dir, self.label_dir)) # 拼接地址 self.img_path =...