姓名:謝意遠
學號:19021110366T
嵌牛導讀:圖像中的目標物體是連接在一起的,則分割起來很困難,分水嶺分割算法經(jīng)常用于處理這類問題,通常會取得比較好的效果。
嵌牛鼻子:圖像分割、分水嶺算法
嵌牛提問:分水嶺算法具體有哪些步驟?
嵌牛正文:
一、綜述
分水嶺分割算法把圖像看成一幅“地形圖”,其中亮度比較強的區(qū)域像素值較大,而比較暗的區(qū)域像素值較小,通過尋找“匯水盆地”和“分水嶺界限”,對圖像進行分割。而直接應用分水嶺分割算法的效果往往并不好,如果在圖像中對前景對象和背景對象進行標注區(qū)別,再應用分水嶺算法會取得較好的分割效果。基于標記控制的分水嶺分割方法有以下基本步驟:
1?綜述
分水嶺分割算法把圖像看成一幅“地形圖”,其中亮度比較強的區(qū)域像素值較大,而比較暗的區(qū)域像素值較小,通過尋找“匯水盆地”和“分水嶺界限”,對圖像進行分割。直接應用分水嶺分割算法的效果往往并不好,如果在圖像中對前景對象和背景對象進行標注區(qū)別,再應用分水嶺算法會取得較好的分割效果。基于標記控制的分水嶺分割方法有以下基本步驟:
1.計算分割函數(shù)。圖像中較暗的區(qū)域是要分割的對象
2.計算前景標志。這些是每個對象內(nèi)部連接的斑點像素。
3.計算背景標志。這些是不屬于任何對象的要素。
4.修改分割函數(shù),使其僅在前景和后景標記位置有極小值。
5.對修改后的分割函數(shù)做分水嶺變換計算。
使用MATLAB圖像處理工具箱
注:期間用到了很多圖像處理工具箱的函數(shù),例如fspecial、imfilter、watershed、label2rgb、imopen、imclose、imreconstruct、imcomplement、imregionalmax、bwareaopen、graythresh和imimposemin函數(shù)等。
2?步驟
?第一步:讀入彩色圖像,將其轉(zhuǎn)化成灰度圖像
clc; clear all; close all;
rgb = imread('pears.png');
if ndims(rgb) == 3
?I = rgb2gray(rgb);
else
?I = rgb;
end
figure('units', 'normalized', 'position', [0 0 1 1]);
第2步:將梯度幅值作為分割函數(shù)
使用Sobel邊緣算子對圖像進行水平和垂直方向的濾波,然后求取模值,sobel算子濾波后的圖像在邊界處會顯示比較大的值,在沒有邊界處的值會很小。
hy = fspecial('sobel');
hx = hy';
Iy = imfilter(double(I), hy, 'replicate');
Ix = imfilter(double(I), hx, 'replicate');
gradmag = sqrt(Ix.^2 + Iy.^2);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(I,[]), title('灰度圖像')
subplot(1, 2, 2); imshow(gradmag,[]), title('梯度幅值圖像')
可否直接對梯度幅值圖像使用分水嶺算法?
L = watershed(gradmag);
Lrgb = label2rgb(L);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(gradmag,[]), title('梯度幅值圖像')
subplot(1, 2, 2); imshow(Lrgb); title('梯度幅值做分水嶺變換')
直接使用梯度模值圖像進行分水嶺算法得到的結(jié)果往往會存在過度分割的現(xiàn)象。因此通常需要分別對前景對象和背景對象進行標記,以獲得更好的分割效果。
第3步:標記前景對象
有多種方法可以應用在這里來獲得前景標記,這些標記必須是前景對象內(nèi)部的連接斑點像素。這個例子中,將使用形態(tài)學技術“基于開的重建”和“基于閉的重建”來清理圖像。這些操作將會在每個對象內(nèi)部創(chuàng)建單位極大值,使得可以使用imregionalmax來定位。
開運算和閉運算:先腐蝕后膨脹稱為開;先膨脹后腐蝕稱為閉。開和閉這兩種運算可以除去比結(jié)構(gòu)元素小的特定圖像細節(jié),同時保證不產(chǎn)生全局幾何失真。開運算可以把比結(jié)構(gòu)元素小的突刺濾掉,切斷細長搭接而起到分離作用;閉運算可以把比結(jié)構(gòu)元素小的缺口或孔填充上,搭接短的間隔而起到連接作用。
開操作是腐蝕后膨脹,基于開的重建(基于重建的開操作)是腐蝕后進行形態(tài)學重建。下面比較這兩種方式。首先,用imopen做開操作。
se = strel('disk', 20);
Io = imopen(I, se);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(I, []); title('灰度圖像');
subplot(1, 2, 2); imshow(Io), title('圖像開操作')
接下來,通過腐蝕后重建來做基于開的重建計算。
Ie = imerode(I,se)
Iobr = imreconstruct(Ie,I);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(I, []); title('灰度圖像');
subplot(1, 2, 2); imshow(Iobr, []), title('基于開的重建圖像')
開操作后,接著進行閉操作,可以移除較暗的斑點和枝干標記。對比常規(guī)的形態(tài)學閉操作和基于閉的重建操作。首先,使用imclose:
Ioc = imclose(Io, se);
Ic = inclose(I,se);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(I, []); title('灰度圖像');
subplot(2, 2, 2); imshow(Io, []); title('開操作圖像');
subplot(2, 2, 3); imshow(Ic, []); title('閉操作圖像');
subplot(2, 2, 4); imshow(Ioc, []), title('開閉操作');
現(xiàn)在使用imdilate,然后使用imreconstruct。注意必須對輸入圖像求補,對imreconstruct輸出圖像求補。IM2 = imcomplement(IM)計算圖像IM的補集。IM可以是二值圖像,或者RGB圖像。IM2與IM有著相同的數(shù)據(jù)類型和大小。
Iobrd = imdilate(Iobr, se);
Iobrcbr = imreconstruct(imcomplement(Iobrd), imcomplement(Iobr));
Iobrcbr = imcomplement(Iobrcbr);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(I, []); title('灰度圖像');
subplot(2, 2, 2); imshow(Ioc, []); title('開閉操作');
subplot(2, 2, 3); imshow(Iobr, []); title('基于開的重建圖像');
subplot(2, 2, 4); imshow(Iobrcbr, []), title('基于閉的重建圖像');
通過比較Iobrcbr和loc可以看到,在移除小污點同時不影響對象全局形狀的應用下,基于重建的開閉操作要比標準的開閉重建更加有效。計算Iobrcbr的局部極大來得到更好的前景標記。
fgm = imregionalmax(Iobrcbr);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 3, 1); imshow(I, []); title('灰度圖像');
subplot(1, 3, 2); imshow(Iobrcbr, []); title('基于重建的開閉操作');
subplot(1, 3, 3); imshow(fgm, []); title('局部極大圖像');
為了幫助理解這個結(jié)果,疊加前景標記到原圖上。
It1 = rgb(:, :, 1);
It2 = rgb(:, :, 2);
It3 = rgb(:, :, 3);
It1(fgm) = 255; It2(fgm) = 0; It3(fgm) = 0;
I2 = cat(3, It1, It2, It3);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(rgb, []); title('原圖像');
subplot(2, 2, 2); imshow(Iobrcbr, []); title('基于重建的開閉操作');
subplot(2, 2, 3); imshow(fgm, []); title('局部極大圖像');
subplot(2, 2, 4); imshow(I2); title('局部極大疊加到原圖像');
注意到大多閉塞處和陰影對象沒有被標記,這就意味著這些對象在結(jié)果中將不會得到合理的分割。而且,一些對象的前景標記會一直到對象的邊緣。這就意味著應該清理標記斑點的邊緣,然后收縮它們。可以通過閉操作和腐蝕操作來完成。
se2 = strel(ones(5,5));
fgm2 = imclose(fgm, se2);
fgm3 = imerode(fgm2, se2);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(Iobrcbr, []); title('基于重建的開閉操作');
subplot(2, 2, 2); imshow(fgm, []); title('局部極大圖像');
subplot(2, 2, 3); imshow(fgm2, []); title('閉操作');
subplot(2, 2, 4); imshow(fgm3, []); title('腐蝕操作');
這個過程將會留下一些偏離的孤立像素,應該移除它們。可以使用bwareaopen,用來移除少于特定像素個數(shù)的斑點。BW2 = bwareaopen(BW,P)從二值圖像中移除所以少于P像素值的連通塊,得到另外的二值圖像BW2。
fgm4 = bwareaopen(fgm3, 20);
It1 = rgb(:, :, 1);
It2 = rgb(:, :, 2);
It3 = rgb(:, :, 3);
It1(fgm4) = 255; It2(fgm4) = 0; It3(fgm4) = 0;
I3 = cat(3, It1, It2, It3);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(I2, []); title('局部極大疊加到原圖像');
subplot(2, 2, 2); imshow(fgm3, []); title('閉腐蝕操作');
subplot(2, 2, 3); imshow(fgm4, []); title('去除小斑點操作');
subplot(2, 2, 4); imshow(I3, []); title('修改局部極大疊加到原圖像');
第4步:計算背景標記
現(xiàn)在,需要標記背景。在清理后的圖像Iobrcbr中,暗像素屬于背景,所以可以從閾值操作開始。
bw =im2bw(Iobrcbr, graythresh(Iobrcbr));
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(Iobrcbr, []); title('基于重建的開閉操作');
subplot(1, 2, 2); imshow(bw, []); title('閾值分割');
背景像素在黑色區(qū)域,但是理想情形下,不必要求背景標記太接近于要分割的對象邊緣。通過計算“骨架影響范圍”來“細化”背景,或者SKIZ,bw的前景。這個可以通過計算bw的距離變換的分水嶺變換來實現(xiàn),然后尋找結(jié)果的分水嶺脊線(DL==0)。D = bwdist(BW)計算二值圖像BW的歐幾里得矩陣。對BW的每一個像素,距離變換指定像素和最近的BW非零像素的距離。bwdist默認使用歐幾里得距離公式。BW可以由任意維數(shù),D與BW有同樣的大小。
D = bwdist(bw);
DL = watershed(D);
bgm = DL == 0;
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2, 2, 1); imshow(Iobrcbr, []); title('基于重建的開閉操作');
subplot(2, 2, 2); imshow(bw, []); title('閾值分割');
subplot(2, 2, 3); imshow(label2rgb(DL), []); title('分水嶺變換示意圖');
subplot(2, 2, 4); imshow(bgm, []); title('分水嶺變換脊線圖');
第5步:計算分割函數(shù)的分水嶺變換
函數(shù)imimposemin可以用來修改圖像,使其只是在特定的要求位置有局部極小。這里可以使用imimposemin來修改梯度幅值圖像,使其只在前景和后景標記像素有局部極小。
gradmag2 = imimposemin(gradmag, bgm | fgm4);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(2,2,1)imshow(bgm,[]);title('分水嶺變換脊線圖');
subplot(2, 2, 2); imshow(fgm4, []); title('前景標記');
subplot(2, 2, 3); imshow(gradmag, []); title('梯度幅值圖像');
subplot(2, 2, 4); imshow(gradmag2, []); title('修改梯度幅值圖像');
最后,可以做基于分水嶺的圖像分割計算。
第6步:查看結(jié)果
一個可視化技術是疊加前景標記、背景標記、分割對象邊界到初始圖像。可以使用膨脹來實現(xiàn)某些要求,比如對象邊界,更加清晰可見。對象邊界定位于L==0的位置。
It1 = rgb(:, :, 1);
It2 = rgb(:, :, 2);
It3 = rgb(:, :, 3);
fgm5 = imdilate(L == 0, ones(3, 3)) | bgm | fgm4;
It1(fgm5) = 255; It2(fgm5) = 0; It3(fgm5) = 0;
I4 = cat(3, It1, It2, It3);
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(rgb, []); title('原圖像');
subplot(1, 2, 2); imshow(I4, []); title('標記和對象邊緣疊加到原圖像');
可視化說明了前景和后景標記如何影響結(jié)果。在幾個位置,部分的較暗對象與它們相鄰的較亮的鄰接對象相融合,這是因為受遮擋的對象沒有前景標記。
另外一個有用的可視化技術是將標記矩陣作為彩色圖像進行顯示。標記矩陣,比如通過watershed和bwlabel得到的,可以使用label2rgb轉(zhuǎn)換到真彩圖像來顯示。
Lrgb = label2rgb(L,'jet', 'w', 'shuffle');
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(rgb, []); title('原圖像');
subplot(1, 2, 2); imshow(Lrgb); title('彩色分水嶺標記矩陣');
可以使用透明度來疊加這個偽彩色標記矩陣在原亮度圖像上進行顯示。
figure('units', 'normalized', 'position', [0 0 1 1]);
subplot(1, 2, 1); imshow(rgb, []); title('原圖像');
subplot(1, 2, 2); imshow(rgb, []); hold on;
himage = imshow(Lrgb);
set(himage, 'AlphaData', 0.3);
title('標記矩陣疊加到原圖像');