bubbliiiing / mask-rcnn-keras

这是一个mask-rcnn的库,可以用于训练自己的实例分割模型。
MIT License
317 stars 71 forks source link

LVIS数据集使用coco.py问题 #16

Open xdnjust opened 3 years ago

xdnjust commented 3 years ago

大佬,我现在想用LVIS数据集来跑这个代码 LVIS和COCO的标注类似,但是其annotation中没有'iscrowd'这个字段 而在coco.py的第233行有关于这个字段的处理,请问如何修正? if annotation['iscrowd']:

Use negative class ID for crowds

                class_id *= -1
                # For crowd masks, annToMask() sometimes returns a mask
                # smaller than the given dimensions. If so, resize it.
                if m.shape[0] != image_info["height"] or m.shape[1] != image_info["width"]:
                    m = np.ones([image_info["height"], image_info["width"]], dtype=bool)
bubbliiiing commented 3 years ago

你直接把那一大段注释了呗