分割结果可视化_将mask的轮廓添加到image

最新推荐文章于 2023-08-17 16:08:17 发布

yxbkl

最新推荐文章于 2023-08-17 16:08:17 发布

阅读量299

点赞数 1

分类专栏：实用工具 AI 文章标签： opencv 计算机视觉 python

本文链接：https://blog.csdn.net/qq_41844449/article/details/129306924

版权

实用工具同时被 2 个专栏收录

27 篇文章 0 订阅

订阅专栏

11 篇文章 0 订阅

订阅专栏

import cv2
#原图
img = cv2.imread('img.png')
#mask
mask = cv2.imread('gt.png')
#将mask转换为灰度图
mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
#设阈值
ret, thresh = cv2.threshold(mask, 127, 255, 0)
#获取二值化后的mask的轮廓
contours, im = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) #第一个参数是轮廓
#在原图上画轮廓信息
cv2.drawContours(image=img, contours=contours, contourIdx=-1, color=(0, 255, 0), thickness=1)
#保存图像
cv2.imwrite('img_contours.png', img)
#如果需要添加两张mask的信息，则在img_contours.png上继续画轮廓即可