使用MATLAB计算连续图像之间的偏移

时间:2011-02-25 23:52:13

标签: matlab loops image-processing for-loop performance

我正在使用隧道显微镜拍摄图像。但是,范围在连续图像之间漂移。我正在尝试使用MatLab来计算图像之间的偏移。下面的代码以秒为单位计算小图像(例如64x64像素),但需要> 2小时来处理我正在处理的512x512像素图像。你对加速这段代码有什么建议吗?或者您是否知道在MatLab中跟踪图像的更好方法?谢谢你的帮助!

%Test templates
template = .5*ones(32);
template(25:32,:) = 0;
template(:,25:64) = 0;
data_A = template;
close all
imshow(data_A);
template(9:32,41:64) = .5;
template(:,1:24) = 0;
data_B = template;
figure, imshow(data_B);

tic

[m n] = size(data_B);
z = [];

% Loop over all possible displacements
for x = -n:n

for y = -m:m

    paddata_B = data_B;
    ax = abs(x);
    zerocols = zeros(m,ax);

    if x > 0
        paddata_B(:,1:ax) = [];
        paddata_B = [paddata_B zerocols];

    else
        paddata_B(:,(n-ax+1):n) = [];
        paddata_B = [zerocols paddata_B];

    end

    ay = abs(y);
    zerorows = zeros(ay,n);


    if y < 0
        paddata_B(1:ay,:) = [];
        paddata_B = vertcat(paddata_B, zerorows);

    else
        paddata_B((m-ay+1):m,:) = [];
        paddata_B = vertcat(zerorows, paddata_B);

    end

% Full matrix sum after array multiplication
C = paddata_B.*data_A;        
matsum = sum(sum(C));

% Populate array of matrix sums for each displacement    
z(x+n+1, y+m+1) = matsum;

end
end

toc

% Plot matrix sums
figure, surf(z), shading flat

% Find maximum value of z matrix
[max_z, imax] = max(abs(z(:)));
[xpeak, ypeak] = ind2sub(size(z),imax(1))

% Calculate displacement in pixels
corr_offset = [(xpeak-n-1) (ypeak-m-1)];
xoffset = corr_offset(1)
yoffset = corr_offset(2)  

4 个答案:

答案 0 :(得分:4)

您计算的内容被称为两张图片的cross-correlation。您可以一次using Discrete Fourier Transforms(DFT或FFT)计算所有偏移的互相关。所以尝试像

这样的东西
z = ifft2( fft2(dataA) .* fft2(dataB).' );

如果在傅里叶域中使用零填充,您甚至可以使用这种数学运算来获得像素分数的偏移量,并将像素分数的偏移量应用于图像。

答案 1 :(得分:3)

解决此类问题的一种典型方法是使用它可以快速处理小图像的事实。当您有大图像时,将它们抽取以制作小图像。快速注册小图像并使用计算的偏移量作为下一次迭代的初始值。在下一次迭代中,您不会对图像进行大量抽取,但是您需要对偏移量进行良好的初始估计,这样您就可以将搜索范围限制在初始估算值附近的小邻域中。 / p>

虽然没有考虑隧道显微镜,但可能有一些帮助的评论文章是:&#34;基于信息的相互医学图像注册:一项调查&#34;由Pluim,Maintz和Viergever发表于IEEE Transactions on Medical Imaging,Vol。 22,No.8,p。 986。

答案 2 :(得分:1)

下面的链接将帮助您找到2张图像之间的转换并纠正/恢复失真(在您的情况下,带偏移的图像)

http://in.mathworks.com/help/vision/ref/estimategeometrictransform.html

index_pairs = matchFeatures(featuresOriginal,featuresDistorted, 'unique', true);
matchedPtsOriginal  = validPtsOriginal(index_pairs(:,1));
matchedPtsDistorted = validPtsDistorted(index_pairs(:,2));
[tform,inlierPtsDistorted,inlierPtsOriginal] = estimateGeometricTransform(matchedPtsDistorted,matchedPtsOriginal,'similarity');
figure; showMatchedFeatures(original,distorted,inlierPtsOriginal,inlierPtsDistorted);

inlierPtsDistored,inlierPtsOriginal具有称为位置的属性。 这些只是一个图像在另一个图像上的匹配位置。我认为从那时起计算偏移量非常容易。

答案 3 :(得分:0)

以下功能是我尝试手动计算两个图像的互相关。但事情并不完全正确。如果我有时间的话,本周末会再看一遍。您可以使用以下内容调用该函数:

>> oldImage = rand(64);
>> newImage = circshift(oldImage, floor(64/2)*[1 1]);
>> offset = detectOffset(oldImage, newImage, 10)

offset =

    32    -1
function offset = detectOffset(oldImage, newImage, margin)

    if size(oldImage) ~= size(newImage)
        offset = [];
        error('Test images must be the same size.');
    end

    [imageHeight, imageWidth] = size(oldImage);

    corr = zeros(2 * imageHeight - 1, 2 * imageWidth - 1);

    for yIndex = [1:2*imageHeight-1; ...
                  imageHeight:-1:1 ones(1, imageHeight-1); ...
                  imageHeight*ones(1, imageHeight) imageHeight-1:-1:1];
        oldImage = circshift(oldImage, [1 0]);
        for xIndex = [1:2*imageWidth-1; ...
                      imageWidth:-1:1 ones(1, imageWidth-1); ...
                      imageWidth*ones(1, imageWidth) imageWidth-1:-1:1];
            oldImage = circshift(oldImage, [0 1]);
            numPoint = abs(yIndex(3) - yIndex(2) + 1) * abs(xIndex(3) - xIndex(2) + 1);
            corr(yIndex(1),xIndex(1)) = sum(sum(oldImage(yIndex(2):yIndex(3),xIndex(2):xIndex(3)) .* newImage(yIndex(2):yIndex(3),xIndex(2):xIndex(3)))) * imageHeight * imageWidth / numPoint;
        end
    end

    [value, yOffset] = max(corr(margin+1:end-margin,margin+1:end-margin));
    [dummy, xOffset] = max(value);
    offset = [yOffset(xOffset)+margin-imageHeight xOffset+margin-imageWidth];