关于yolov5读取训练数据时强行跳过存在标签的图像

场景：

我有一个数据集，8个类别，voc格式，现在我只想训练其中的4个类别，正常我们的做法就是把类别名和数量改下就行了，但是在yolov5的官方代码中却存在一个bug，该bug在utils/datasets.py的cache_labels()方法中，主要问题是该方法会读取标注文件的objects，获取类别名，并根据类别名找到在4个类别中的索引位置，因为这个标注文件含有8个类别，另外4个类别是没有索引位置的，所以会报错，报错后后面的object就不处理了，这就导致读取的标注框大大减少。