分割结果可视化：怎么把标签mask轮廓显示在原图上

原创

已于 2022-11-15 17:02:45 修改 · 1w 阅读

标签

#计算机视觉 #python #opencv

收录于

于 2022-11-15 16:05:49 首次发布

本文探讨了如何使用Numpy处理视盘视杯分割模型的npy文件，包括图像可视化、颜色通道调整、文件保存和轮廓标注的技巧，以及常见问题与解决方案。

训练完模型得到视盘视杯的分割结构之后，查看论文中的示例图，分割结果的显示为：
在这里插入图片描述
下面是可视化过程中遇到的一些问题

问题1： 怎么可视化npy文件

cv2.imshow("ima", ima) # 需要两个参数 记住图片名称不可少
cv2.waitKey() # 等待时间 不写就是手动 0就是1s后会自动关闭

注意！！！
失败的原因 cv2显示的时候图像必须是(h,w,c) 通道数在最后否则出现报错
转换维度ima = image.transpose(1, 2, 0)
这里显示出来的图像颜色也很奇怪
查阅资料发现：
如果希望使用np.load读取图片，且使用cv2.imshow来输出正常照片的话，可以在输出时把颜色通道的顺序从BGR调整为RGB就行了。
opencv的接口使用BGR,而matplotlib.pyplot 则是RGB模式,所以将通道数进行对换一下即可

import cv2
 
image=np.load(image1)
ima = image.transpose(1, 2, 0) # 转化为（h,w,c)
ima_re = ima[:,:,::-1] # 转换成正常的颜色通道
cv2.imshow("ima", ima_re) # 需要两个参数 记住图片名称不可少
cv2.waitKey() # 等待时间 不写就是手动 0就是1s后会自动关闭

![[Pasted image 20221115145419.png|200]]![[Pasted image 20221115150023.png|200]]

问题2：