python - 使用 pycococreator 从自定义数据集转换为 COCO 格式的极简示例

Question

目的是将包含灰度+groundtruth图像对的numpy数组（2164、190、189、2）转换为COCO格式：

我尝试生成 coco 格式的极简注释，如下所示：

from pycococreatortools import pycococreatortools

N = 1
grey = data[N,:,:,0]
labels = data[N,:,:,1]
mask1 = labels == 1
mask2 = labels == 2
mask3 = labels == 3

segmentation_id = "chromosome1_1" 
image_id = "0001"
category_info = "chromosome1"
binary_mask = mask1
image = grey
print(image.size, image.shape)

annotation_info_mask1 = pycococreatortools.create_annotation_info(segmentation_id, image_id, 
                                                              category_info,
                                                              binary_mask,
                                                              image.size, tolerance=2)

我收到以下错误：

---

TypeError Traceback（最近一次调用最后一次）

in () 2 category_info, 3 binary_mask, ----> 4 image.size, tolerance=2)

2 帧

/usr/local/lib/python3.6/dist-packages/PIL/Image.py in resize(self, size, resample, box) 1866 ) 1867 -> 1868 size = tuple(size) 1869 1870 如果 box 为 None：

TypeError：“int”对象不可迭代

这里 image.size 等于35910并且 image.shape 是 (190, 189)。

似乎create_annotation_info()正在等待一个元组或一个列表。这是对的吗？什么是正确的方法？

这里有一个草稿 colab 笔记本来获取整个数据集。