该字段为通过coco数据集”annotations“中的“segmentation”来计算mask,检查一下coco数据集中的“segmentation”是否为空。可以将“bbox”传入“segmentation”
with open ('valid.json',encoding='utf-8') as f:
json_info = json.load(f)
for i in json_info["annotations"]:
[x,y,w,h] = i["bbox"]
segmentation = [[x,y,(x+w),y,(x+w),(y+h),x,(y+h)]]
i["segmentation"] = segmentation
with open ('valid_1.json','w',encoding='utf-8') as f:
json.dump(json_info, f, indent=1)
参考链接:KeyError: 'mask' --- gt_mask = instance['mask'] · Issue #10961 · open-mmlab/mmdetection (github.com)