‘Results do not correspond to current coco set‘

最新推荐文章于 2025-04-09 21:50:46 发布

河海CV小菜鸡

最新推荐文章于 2025-04-09 21:50:46 发布

阅读量7k

点赞数 12

文章标签：大数据 python

本文链接：https://blog.youkuaiyun.com/m0_59135992/article/details/125802608

版权

在训练YOLOV6时遇到'Results do not correspond to current coco set'的问题，原因是dataset.py中generate_coco_format_labels方法的img_id处理不当，导致json文件中的image_id与实际图片id不匹配。解决方案是修改img_id的获取方式，以'j'分割而不是'.'，确保图片id正确。修复后代码能成功运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在训练YOLOV6的时候碰上了

'Results do not correspond to current coco set'

这个问题。

assert set(annsImgIds) == (set(annsImgIds) & set(self.getImgIds())) 根据这个断言检查了下。

调用coco API检查了下，打印出来len((person_imgs_id)的值为297.

from pycocotools.coco import COCO
val_info = r'VOC/annotations/instances_val.json'

coco = COCO(val_info)  # 导入验证集
person_imgs_id = coco.getImgIds()
print(len(person_imgs_id))

而在检查set(annsImgIds)时，在evaler.py文件的eval_model这个方法中

annsImgIds = [ann['image_id'] for ann in pred_results]
print(len(set(annsImgIds)))

打印出来的长度为497（也就是我验证集的数量）

然后检查了下instances_val.json文件，发现

image_id是6，而我的数据图片id为6.8CAXX.jpg，那么就找到了问题的所在。

在dataset.py中的generate_coco_format_labels方法中

img_id = osp.basename(img_path).split('.')[0]

这样就导致以一个点分割了这个图片id，6.8CAXX.jpg被分割乘了6、8CAXX、jpg 三个字符串，所以导致了json文件中出现了image_id为6的情况。

解决办法：

在dataset.py中的generate_coco_format_labels方法中

img_id = osp.basename(img_path).split('j')[0].strip('.')

将img_id改成上述代码，以j分割，再去掉剩下来的点，代码就能成功跑通了