MASKRCNN修改display_instances

class_names = ["background",""]





demoimg = IMAGE_DIR+"/"+"test1.png"

读取图像skimage.io.imread

image =skimage.io.imread(demoimg)  

result = model.detect([image], verbose=1)

rL = result[0]

visualize.display_instances(image, rL['rois'], rL['masks'], rL['class_ids'],

                                    class_names,scores=None,show_bbox=False,captions=None)



通过以上方法可以直接使输出图片仅有mask .

sitemap