含mask的单通道灰度图内容可视化python

输入:单通道的灰度图,灰度图内含不同像素值的掩膜mask

输出:灰度图内的掩膜mask在RGB图像中输出

方法很简单,就是读取灰度图,根据掌握的像素值信息,分别赋予不同的颜色值像素进行保存输出。

如下输入的单通道灰度图:

使用如下代码处理:

python 复制代码
import os
import sys
import cv2
import numpy as np 



def main():
    mask_dir = './mask'
    show_dir = './mask_out'
    for mskf in os.listdir(mask_dir):
        msfile = mask_dir + '/' + mskf
        # msimg = cv2.imread(msfile, cv2.IMREAD_GRAYSCALE) 读出是单通道
       

        msimg = cv2.imread(msfile)
                
        # poly = mask2poly(msimg)
        """提供的mask的像素值是0,1,2,0是背景,现在在三通道图像中针对像素值赋予颜色像素"""
        # msimg[msimg==1]=255
        rms = msimg[:,:,0]
        gms = msimg[:,:,1]
        bms = msimg[:,:,2]

        #1-0,255,0
        rms[rms==1] = 0
        bms[bms==1] = 255
        gms[gms==1] = 0

        #2-0,0,255
        rms[rms==2] = 0
        bms[bms==2] = 0
        gms[gms==2] = 255

        showimg = show_dir + '/' + mskf

        cv2.imwrite(showimg,msimg)
        

if __name__ == '__main__':
    main()

输出结果如下:

相关推荐
偶尔微微一笑1 小时前
AI网络渗透kali应用(gptshell)
linux·人工智能·python·自然语言处理·编辑器
jndingxin1 小时前
OpenCV 图形API(60)颜色空间转换-----将图像从 YUV 色彩空间转换为 RGB 色彩空间函数YUV2RGB()
人工智能·opencv·计算机视觉
知舟不叙2 小时前
OpenCV中的SIFT特征提取
人工智能·opencv·计算机视觉
船长@Quant2 小时前
文档构建:Sphinx全面使用指南 — 基础篇
python·markdown·sphinx·文档构建
喵手2 小时前
从 Java 到 Kotlin:在现有项目中迁移的最佳实践!
java·python·kotlin
liuweidong08022 小时前
【Pandas】pandas DataFrame rsub
开发语言·python·pandas
CH3_CH2_CHO3 小时前
不吃【Numpy】版
开发语言·python·numpy
-曾牛4 小时前
企业级AI开发利器:Spring AI框架深度解析与实战
java·人工智能·python·spring·ai·rag·大模型应用
Light604 小时前
智启未来:深度解析Python Transformers库及其应用场景
开发语言·python·深度学习·自然语言处理·预训练模型·transformers库 |·|应用场景
坤岭4 小时前
Python基础
python