MATLAB SPMD и nlfilter - изменить размер окна - PullRequest
Новая
       5

MATLAB SPMD и nlfilter - изменить размер окна

0 голосов
/ 18 октября 2011

Я использую модифицированную версию CONTRAST2_SPMD Джона Буркардта для выполнения вычисления дисперсии 3x3 на большом изображении в градациях серого. Преимущество этого состоит в том, что я могу использовать PCT для использования 8 ядер на моей локальной машине, однако я хотел бы изменить размер окна (на данный момент это 3x3), но я хотел бы попробовать несколько размеров окна. Как бы я изменил код ниже, чтобы учесть переменные размеры окна? (Я буду использовать нечетные квадратные окна, например, 3x3,5x5,7x7,9x9 и т. Д.) 

function y = parwinvar ( x )

%**************************************************************************
%
%
%% PARWINVAR uses MATLAB's SPMD command for parallel windowed variance.
%
%  Discussion:
%    Calculates windowed standard deviation (squared to get variance).
%    Based on CONTRAST2_SPMD by John Burkardt.
%
%
%  Parameters:
%
%    Input, image X, the initial black and white image.
%    Output, image Y, the contrast-enhanced black and white image.
%
%
%
%  Open the matlabpool
%
    matlabpool open local 8
%
% Check image is grayscale, if not covert it
% 
    if ndims(x)>2
         x=rgb2gray(x);
    end
%
%  Since the image is black and white, it is a 2D array.
%  Hence, it will be distributed by columns.
%
  xd = distributed ( x );
%
%  Have each worker enhance the contrast in its portion of the picture.
%
%  You will see lots of meaningless error messages, because NLFILTER
%  wants to put out a "waitbar" telling you to wait.  But the workers
%  don't have an associated display.
%
  spmd

    xl = getLocalPart ( xd );
%
%  In order to use LABSENDRECEIVE, we need to reference the previous
%  and next labindex.
%
    if ( labindex ~= 1 )
      previous = labindex - 1;
    else
      previous = numlabs;
    end

    if ( labindex ~= numlabs )
      next = labindex + 1;
    else
  next = 1;
end
%
%  Each worker sends its first column to the previous worker,
%  and receives corresponding data from the next worker.
%
column = labSendReceive ( previous, next, xl(:,1) );

if ( labindex < numlabs )
  xl = [ xl, column ];
end
%
%  Each worker sends its last column to the next worker,
%  and receives corresponding data from the previous worker.
%
column = labSendReceive ( next, previous, xl(:,end) );

if ( 1 < labindex )
  xl = [ column, xl ];
end
%
%  Now do the enhancement.
%  We can only do a 3x3 neighborhood, because we're only sending
%  one column to the left and right.
%   
xl = nlfilter ( xl, [3,3], @std2 );
%
%  Now strip off the extra columns.
%
    if ( 1 < labindex )
      xl = xl(:,2:end);
    end

    if ( labindex < numlabs )
      xl = xl(:,1:end-1);
    end

   xl = uint8 ( xl );

  end
%
%  We are working with a black and white image, so we can simply
%  concatenate the submatrices to get the whole object.
%
  y = [ xl{:} ];
  y = (y .* y); % square to get variance
% Close matlabpool
matlabpool close
  return
end

Заранее спасибо за любую помощь.

1 Ответ

1 голос
/ 18 октября 2011

Вам нужно изменить строки labSendReceive, чтобы отправить больше данных.В частности, что-то вроде этого: 

%
%  Each worker sends its first column to the previous worker,
%  and receives corresponding data from the next worker.
%
column = labSendReceive ( previous, next, xl(:,1) );

Необходимо стать: 

%
%  Each worker sends its first N columns to the previous worker,
%  and receives corresponding data from the next worker.
%
columns = labSendReceive ( previous, next, xl(:,1:N) );

, где N равно (blocksize-1)/2.Соответствующее изменение должно быть сделано при отправке «призрачных ячеек» с правой стороны и при удалении «лишних столбцов».

...