yolov5训练数据集中的部分类别

引用：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列表即可