成功解决训练模型时出现错误 return [self.imgs[id] for id in ids] KeyError: '2'

最新推荐文章于 2025-04-04 16:14:45 发布

梦坠凡尘(AICV与前沿)

最新推荐文章于 2025-04-04 16:14:45 发布

阅读量3.3k

点赞数 3

分类专栏： bug 文章标签： bug python 深度学习

本文链接：https://blog.csdn.net/c2250645962/article/details/105521751

版权

bug 专栏收录该内容

15 篇文章

订阅专栏

本文解决使用自定义数据集训练模型时遇到的COCO格式JSON文件image_id类型错误问题，详细介绍了如何将image_id从字符串类型转换为整数类型，确保模型训练正常进行。

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

使用自己制作的数据集和转换好的coco格式json文件训练时报错：
image_info = self.coco.loadImgs(self.image_ids[image_index])[0]
File “/home/cendelian/anaconda3/envs/tf/lib/python3.6/site-packages/pycocotools/coco.py”, line 229, in loadImgs
return [self.imgs[id] for id in ids]
File “/home/cendelian/anaconda3/envs/tf/lib/python3.6/site-packages/pycocotools/coco.py”, line 229, in
return [self.imgs[id] for id in ids]
KeyError: '2’

后面查找是因为我转换的json文件中的image_id是‘str’类型而不是‘int’类型，将其修改为‘int’类型即可解决。

json文件中annotation字段必须是如下格式：

annotation{
    "id": int,    
    "image_id": int,
    "category_id": int,
    "segmentation": RLE or [polygon],
    "area": float,
    "bbox": [x,y,width,height],
    "iscrowd": 0 or 1,
}