开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >如何从MSCOCO数据集中获取person类和分段？

问如何从MSCOCO数据集中获取person类和分段？
EN

Stack Overflow用户

提问于 2021-12-30 12:03:55

回答 1查看 406关注 0票数 3

我只想从COCO数据集下载person类和二进制分段。我该怎么做呢？

image-segmentation

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-01-06 05:04:59

使用化茧工具。

从pycocotools.coco导入COCO导入库
加载coco注释coco = COCO('/home/office/cocoDataset/annotations/instances_train2017.json')的json文件
获取coco数据集的类别ID category_ids = coco.getCatIds(catNms='person')
获取单个图像注释= coco.getAnnIds(imgIds=image_id、catIds=category_ids、iscrowd=False)的注释
在这里，每个人都有不同的注释，我的注释是annotation[i]，因此合并所有注释并保存它，如果注释:掩码=coco.annToMask(注释)在范围内(len(注释))：掩码|= coco.annToMask(注释)掩码=掩码* 255 im =Image.fromarray(掩码)im.save(‘~/掩码_name.png’)

票数 5

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/70531408

复制

相似问题