代码之家 › 专栏 › 技术社区 › Farshid Rayhan

如何从COCO数据集创建遮罩图像?

image-segmentation computer-vision tensorflow image-processing python

Farshid Rayhan · 技术社区 · 6 年前

所以我一直在使用这个代码。我正在尝试从COCO数据集中生成图像的原始蒙版。

datadir='g:'
datatype='train2014'
annfile='/annotations/instances.json'.格式(datadir,datatype)


coco=coco(annfile)
annfile='/annotations/person.json'.格式(datadir,datatype)
coco_kps=coco(annfile)


catids=coco.getcatids(catnms=['人'])
imgids=coco.getimgids(catids=catids);
imgids=coco.getimgids(imgids=imgids[0])
img=coco.loadimgs(imgids[np.random.randint(0,len(imgids))])[0]
i=io.imread('g:/train2014/'+img['文件名'])

plt.imshow(i);plt.axis('off')
annids=coco.getannids(imgids=img['id'],catids=catids,iscrowd=none)
anns=coco.loadanns(annid)
Coco.Showanns(美国国家标准)
但我得到的是这样的东西



但我想要这样的东西



如何获取每个图像的原始遮罩?

但我得到的是这样的东西



但我想要这样的东西



如何获取每个图像的原始遮罩?

2 回复 | 直到 6 年前

Farshid Rayhan 6 年前

mask = coco.annToMask(anns[0])
for i in range(len(anns)):
    mask += coco.annToMask(anns[i])

plt.imshow(mask)

filippo 6 年前

annToMask

annIds = coco.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None)

mask = np.zeros_like(img)
for i, ann in enumerate(annIds):
    mask += coco.annToMask(ann) * i

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

Anshuman Sinha · 将tensforflow中的所有nan值转换为零

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

rob me of my freedom · 在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

2 年前

Chris · 如何捕获导入脚本功能的打印

3 年前

Jay Bhie Santos · 剩余GRU:ValueError:输入具有不兼容的形状。收到的形状(5)和(24)

3 年前

Danijel · C中TensorFlow模型文件的路径?

3 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

3 年前

Arya Anish · tensorflow安装中面临的问题

3 年前

so beautiful memory · 如何将我在本地训练的tensorflow模型正确部署到谷歌云人工智能平台?我部署了它,但没有图像返回

3 年前