1.dataset

主要编写这两个程序

首先需要提供图片地址

创建数据读取路径,但是
有点疑惑,为啥label也是地址。
root_dir是根目录,可以是绝对路径,也可以是相对路径。label_dir是标签self.path=os.path.join(self.root_dir,self.label_dir)的作用是把根目录和标签拼在一起,所形成的就是一个保存有训练所需数据的文件夹的目录。

__getitem__返回的是img和label,img是图像本身,label是标签。

浙公网安备 33010602011771号