1.dataset

 

 

 

主要编写这两个程序

 

 首先需要提供图片地址

 

 

 创建数据读取路径,但是有点疑惑,为啥label也是地址。

 

root_dir是根目录,可以是绝对路径,也可以是相对路径。
label_dir是标签
self.path=os.path.join(self.root_dir,self.label_dir)的作用是把根目录和标签拼在一起,所形成的就是一个保存有训练所需数据的文件夹的目录。

 

 

 

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

 

posted @ 2022-05-30 19:31  澄镜  阅读(114)  评论(0)    收藏  举报