在Python中利用Detectron或Mask R-CNN进行图像提取
发布时间
阅读量：
阅读量
依据示例文件infer_simple.py进行调整，将第165行原本调用的vis_utils.vis_one_image函数替换为vis_utils.vis_one_image_opencv。同时，在detectron.utils.vis模块中新增一个名为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_
全部评论 (0)
还没有任何评论哟~