clear;clc;
IDlist=[0 51 102 153 204 255];
RGBlist=[255 0 0;0 255 0;0 0 255;255 255 0;0 255 255; 255 0 255];
info=dir('*.png');
names={info.name};
for i=1:numel(names)
I=imread(names{i});
sz=size(I);
I2=I(:,:,1);
IsaveR=zeros(sz(1:2));
IsaveG=zeros(sz(1:2));
IsaveB=zeros(sz(1:2));
for j=1:numel(IDlist)
idx=find(I2==IDlist(j));
IsaveR(idx)=RGBlist(j,1);
IsaveG(idx)=RGBlist(j,2);
IsaveB(idx)=RGBlist(j,3);
end
Isave=uint8(cat(3,IsaveR,IsaveG,IsaveB));
imwrite(Isave,['rgbLabel_',names{i}])
end
figure;imshowpair(I2,Isave,'montage')
こんにちは、例えば上のように書くと、ID⇒RGBが変換できます。もう少し短縮もできるかとは思いますが、
読みやすさも考えて以上のようにしました。IDのPNGファイルがあるディレクトリで実行すれば大丈夫と思います。
このように変換すると、公式ドキュメントにそのままあてはめることが可能と思います。左の練習用のID画像は添付のコードで作成しています。
