python 基于detectron或mask_rcnn的mask遮罩区域进行图片截取

基于示例infer_simple.py

修改165行vis_utils.vis_one_image为vis_utils.vis_one_image_opencv

在detectron.utils.vis中新增def cut_mask(img, mask)

def cut_mask(img, mask):
    mask = mask*255
    # 灰度图片‘cv2.cvtColor(mask,cv2.COLOR_BAYER_GB2GRAY)’转成三通道
    mask = cv2.cvtColor(cv2.cvtColor(mask,cv2.COLOR_BAYER_GB2GRAY),cv2.COLOR_GRAY2BGR)
    cv2.imshow('mask',mask)   
    ROI = cv2.bitwise_and(mask, img)
    cv2.imshow('roi', ROI)
    return ROI

在vis_one_image_opencv方法中

注释

im = vis_mask(im, masks[..., i], color_mask)

改为调用

im = cut_mask(im, masks[..., i])

此时遮罩区域截图已经完成

查看上方的vis_bbox方法可以获取识别框的位置与大小，这样就可以截取指定位置的遮罩区域。

补充：基于mask_rcnn注释152行masked_image = apply_mask(masked_image, mask, color)

改为上述调用。