如何显示标注的纯黑mask图

文章目录

前言

通常情况下,使用标注软件标注的标签图看起来都是纯黑的,因为mask图为单通道的灰度图,而灰度图一般要像素值大于128后,才会逐渐显白,255为白色。而标注的时候,不同类别的像素值是从1,2,3...这样的顺序,所以看起来是纯黑的。

一、二分类mask显示

若只是二分类,原本像素值就只有0和1的区分,那只需要将像素1转换为255即可显示白色,可使用inRange函数

c 复制代码
	cv::Mat img = cv::imread("C:/Users/WA.png",0);
	cv::Mat img_i;
	cv::inRange(img, 1, 1, img_i); // 上界和下界都为1,即只匹配值为1的像素;将值为1的像素设置为白色(255),其余像素为黑色(保持不变)
	cv::imshow("img", img);
	cv::imshow("img_i", img_i);

二、多分类mask显示

若存在多个类别,就需要给每个类别单独定义一种颜色。(彩色图是三通道的,mask图是单通道,所以这种操作也称为伪彩色映射)

代码如下(示例):

python 复制代码
import cv2
import numpy as np
from PIL import Image
import glob

color_map = np.zeros((256 * 3)).astype('uint8')
color_map[:3 * 13] = np.array([[0, 0, 0],   # 0像素还是得为0
                                  [41, 43, 204],  # 原像素值为1的
                                  [6, 128, 245],  # 原像素值为2的
                                  [36, 159, 67],
                                  [41, 43, 204],
                                  [190, 104, 145],
                                  [75, 86, 135],
                                  [195,120,219],
                                  [127, 127, 127],
                                  [18, 189, 187],
                                  [207, 190, 72],
                                  [233, 199, 178],
                                  [118, 187, 248]
                                  ],dtype='uint8').flatten()

def labeltocolor(mask):
    im=Image.fromarray(mask)
    im.putpalette(color_map)
    im=np.array(im.convert('RGB'))
    # cv2.imshow("1",im)
    # cv2.waitKey()
    # cv2.destroyAllWindows()
    return im


# 遍历文件夹中的所有图片
image_paths = glob.glob('C:/Users/Desktop/train01/01/mask/*.png')
idx = 0  # 当前图片索引
num_images = len(image_paths)  # 图片数量

for image_path in image_paths:
    # 加载单通道遮罩图(假设为灰度图)
    mask = cv2.imread(image_paths[idx], cv2.IMREAD_GRAYSCALE)

    # 将遮罩图像应用伪彩色映射
    mask_color = labeltocolor(mask)


    cv2.imshow('Result', mask_color)
    key = cv2.waitKey(0) 

    if key == ord('q'):  
        break
    elif key == ord('n'):  
        idx = (idx + 1) % num_images

cv2.destroyAllWindows()
相关推荐
张较瘦_23 分钟前
[论文阅读] 人工智能 | 机器学习工作流的“救星”:数据虚拟化服务如何解决数据管理难题?
论文阅读·人工智能·机器学习
蓝卓工业操作系统1 小时前
天铭科技×蓝卓 | “1+2+N”打造AI驱动的汽车零部件行业智能工厂
人工智能·科技·汽车
zzywxc7871 小时前
编程算法在金融、医疗、教育、制造业等领域的落地案例
人工智能·算法·金融·自动化·copilot·ai编程
zzywxc7871 小时前
编程算法在金融、医疗、教育、制造业的落地应用。
人工智能·深度学习·算法·机器学习·金融·架构·开源
修一呀1 小时前
【数据标注】详解使用 Labelimg 进行数据标注的 Conda 环境搭建与操作流程
人工智能·conda
白熊1885 小时前
【大模型LLM】梯度累积(Gradient Accumulation)原理详解
人工智能·大模型·llm
愚戏师5 小时前
机器学习(重学版)基础篇(算法与模型一)
人工智能·算法·机器学习
F_D_Z6 小时前
【PyTorch】图像多分类项目部署
人工智能·pytorch·python·深度学习·分类
音视频牛哥8 小时前
打通视频到AI的第一公里:轻量RTSP服务如何重塑边缘感知入口?
人工智能·计算机视觉·音视频·大牛直播sdk·机器视觉·轻量级rtsp服务·ai人工智能
Wendy14419 小时前
【灰度实验】——图像预处理(OpenCV)
人工智能·opencv·计算机视觉