批量灰度图像转换成伪彩色图像，数据可视化，灰度图像上色

不念不忘少年蓝@ 2022-05-27 03:20 307阅读 0赞

实验当中总要可视化自己的数据，比方说标签是33类，你要可视化，单纯的灰度值拉伸不明显，如果是11类那会明显很多，但是类别越多，边界线越模糊，所以最好还是上彩色图像。但是上色程序里面的颜色值又不能随机化，否则本次可视化结果和下一次就不一样了，不利于demo。

所以可以写个python程序自动生成颜色数组，打个比方，我要生成matlab的：

![70][]

import random
    for i in range(11,34):
        s = "color{" + repr(i) + "} = ["
        r = random.randint(1,255)
        g = random.randint(1,255)
        b = random.randint(1,255)
        s = s + repr(r) + ", "+ repr(g) + ", " + repr(b) + "];"
        print(s)

这样就生成了颜色的元祖，可能会有颜色相近的。

那么如果应用到matlab，代码如下：

我做了比较多的边缘检测，因为没有无符号整数化就会导致图片显示有问题，会把大部分颜色显示成白色。后来我才知道问题出在uint8上。bgFile是我的图片。

fileFolder = fullfile('F:\STUDY\CAFFE\测试图片');
    dirOutput = dir(fullfile(fileFolder,'*.jpg'));
    fileNames = {dirOutput.name};
    OtpDir = 'F:\STUDY\CAFFE\result_rgb';%输出路径
    OrgDir = 'F:\STUDY\CAFFE\result1';
    dirChange = dir(fullfile(OrgDir,'*.jpg'));
    fileChange = {dirChange.name};
    hit = [];
    hit(:,:)=0;
    color={};
    %color{0}=[50,79,140];
    color{1}=[100,125,17];
    color{2}=[0,55,0];
    color{3}=[255,0,0];
    color{4}=[0,25,45];
    color{5}=[0,0,255];
    color{6}=[0,255,0];
    color{7}=[255,55,0];
    color{8}=[255,0,25];
    color{9}=[0,56,255];
    color{10}=[56,0,125];
    color{11}=[226,70,2];
    color{12}=[172,5,238];
    color{13}=[2,210,246];
    color{14}=[47,52,5];
    color{15}=[177,212,31];
    color{16}=[84,139,254];
    color{17}=[91,6,127];
    color{18}=[63,43,159];
    color{19}=[255,20,147];
    color{20}=[46,38,70];
    color{21}=[167,227,187];
    color{22}=[229,210,166];
    color{23}=[138,8,12];
    color{24}=[115,16,11];
    color{25}=[34,133,149];
    color{26}=[73,234,11];
    color{27}=[1,42,41];
    color{28}=[159,22,226];
    color{29}=[48,197,162];
    color{30}=[99,96,90];
    color{31}=[149,40,82];
    color{32}=[84,18,214];
    color{33}=[171,85,240];
    color{34} = [170, 118, 46];
    color{35} = [183, 13, 194];
    color{36} = [155, 4, 42];
    color{37} = [247, 90, 55];
    for i = 1:1:57
        bgFile = imread(fileChange{1,i});
        I=double(bgFile);
        [m,n]=size(I);
        for p = 1:m
            for q = 1:n
               % hit(I(p,q))=1;
                if I(p,q)>=1 && I(p,q)<=37
                   % I(p,q)
                out(p,q,1)=color{I(p,q)}(1);
                out(p,q,2)=color{I(p,q)}(2);
                out(p,q,3)=color{I(p,q)}(3);
                if out(p,q,1) >= 200 && out(p,q,2)>=200 && out(p,q,3) >= 200
                    I(p,q)
                    color{I(p,q)}
                end
                    
                else
                    I(p,q)
                end
            end 
        end
        %img = bgFile * 7;
        for p =1:m
            for q= 1:n
        if out(p,q,1) >= 200 && out(p,q,2)>=200 && out(p,q,3) >= 200
                    I(p,q)
                    color{I(p,q)}
        end
            end
        end
        imshow(out)
        filename = fileNames{1,i};
        path = fullfile(OtpDir,filename);
        out = uint8(out);
        imwrite(out,path,'jpg');
    end

cmap = colormap(jet(256));
    rgb = ind2rgb(labels*10,cmap);
    rgb=rgb*100;rgb=uint8(rgb);
    imshow(rgb);

下面这个是更简单的，其中label是你那个图片。师兄给的代码，随机的颜色。问题是上一张图片随机的颜色是否和下一张一样？

[70]: /images/20220527/b3809ec25c8e468eb87197af21ef57dd.png