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上继续画轮廓即可
分割结果可视化_将mask的轮廓添加到image
最新推荐文章于 2023-08-17 16:08:17 发布