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

你可能感兴趣的:(深度学习,人工智能,深度学习,python,人工智能)