引用:yolov5只训练数据集中的某几个类别_Starkiron的博客-CSDN博客
需求:数据集中有两个类别,数据集较大,想只训练其中的类别label1
引用链接里提供了一个不错的思路,由于要丢掉的类别不是label值最大的,所以直接过滤label存在报错“Possible class labels***”。
解决方案:
dataset.py中vertify_image_label函数修改:
withopen(lb_file) as f:
l = [x.split() for x in f.read().strip().splitlines() if len(x) and x[0]!=='1']#这里按需修改
for ll in l:
ll[0] = '0'#这里不改的话,配置文件里nc设为1会报错:***Possible class labels are 0-0
同时修改data,yaml中的nc和bnames列表即可