kyyaoyao的博客

原创 yolov5如何训练公开数据集中某几个标签

受到启发，其中第二种办法并不适合我们，因为我的数据集共有四类，但是只想使用后三类。他提到手动更改数据集标签这种方式比较呆。于是想到，能不能写一个代码来快速更改标签文件内容。我想使用数据集中后三类，也就是说删除标签中每一行首个数值为0的这一行。在使用公开数据集进行训练的时候，发现我们只使用其中某几个标签就可以了。然后再将后三类数值分别减一就OK了。最后更改nc和类别，完美！然后将后面信息进行补位。