減少Matlab代碼的計算時間

Question

為了計算輸入圖像的增強功能，我編寫了以下代碼：

Ig = rgb2gray(imread('test.png'));
N = numel(Ig);
meanTotal = mean2(Ig);
[row,cal] = size(Ig);

IgTransformed = Ig;
n = 3;
a = 1;
b = 1;
c = 1;
k = 1;
for ii=2:row-1
    for jj=2:cal-1
        window = Ig(ii-1:ii+1,jj-1:jj+1);
        IgTransformed(ii,jj) = ((k*meanTotal)/(std2(window) + b))*abs(Ig(ii,jj)-c*mean2(window)) + mean2(window).^a;
    end
end

如何減少計算時間？ 顯然，因素之一是每次都應在循環中創建的小窗口（3x3）。

Answer 1

干得好 -

Igd = double(Ig);
std2v = colfilt(Igd, [3 3], 'sliding', @std);
mean2v = conv2(Igd,ones(3),'same')/9;
Ig_out = uint8((k*meanTotal)./(std2v + b).*abs(Igd-cal*mean2v) + mean2v.^a);

這也會改變邊界元素，如果不需要，可以通過很少的額外步驟將邊界元素設置回原始元素，例如-

Ig_out(:,[1 end]) = Ig(:,[1 end])
Ig_out([1 end],:) = Ig([1 end],:)