不知道如何贴标签给网上下载的图片数据集,我将这些照片分为了几类,怎么根据文件夹贴标签
题主的意思是指图片分类存放在不同的文件夹了吗?如果是的话,直接将每个文件夹的图片读取出来,然后根据文件夹名进行图片命名不就好了,可以命名为文件夹名_x,x是数字编号从0开始,每个图片都一个单独的数字编号不就行了。
遍历你的外层目录,然后多个判断分别进入你的名字文件夹,各自加标签
就0,1,2.。。这样往后对就行了,一个数是一类
先定义文件夹属于什么标签
取文件时,根据所定义的标签分类。
做好的做法就是加载数据集的时候,读取图片的时候按照文件夹来读取,同一个文件夹下面的label设置为文件夹名字就可以了,然后将label和img一一对应返回生成数据集,接下来看你是否需要划分数据集或者你的数据集已经另外划分完毕