Деформирование изображения с использованием контрольных точек

Я хочу преобразовать изображение с помощью контрольных точек в соответствии с этой схемой, извлеченной из здесь:

введите описание изображения здесь

A и B содержит координаты источника целевых вершин.

Я вычисляю матрицу преобразования как:

A = [51 228;  51 127; 191 127; 191 228];
B = [152 57; 219 191;  62 240;  92 109];
X = imread('rectangle.png');
info = imfinfo('rectangle.png');
T = cp2tform(A,B,'projective');

До сих пор он работает правильно, потому что (используя нормированные координаты) исходная вершина создает свою целевую вершину:

H = T.tdata.T;
> [51 228 1]*H
ans =
  -248.2186   -93.0820    -1.6330
> [51 228 1]*H/ -1.6330
ans =
   152.0016    57.0006     1.0000

Проблема заключается в том, что imtransform создает неожиданный результат:

Z = imtransform(X,T,'XData',[1 info.Width], 'YData',[1 info.Height]);
imwrite(Z,'projective.png');

Неожиданный результат

Как я могу использовать imtransform для получения этого ожидаемого результата?:

введите описание изображения здесь

Есть ли альтернативный способ его достижения?

Ответы

Ответ 1

Вам нужно "адаптировать" контрольные точки к размеру изображения, с которым вы работаете. То, как я это делал, заключается в вычислении аффинного преобразования между углами контрольных точек в A и углах исходного изображения (желательно, чтобы вы делали точки в одном и том же порядке по часовой стрелке).

Одна вещь, которую я должен указать, состоит в том, что порядок точек в вашей матрице A не соответствует изображению, которое вы показали, поэтому я исправил это в коде ниже...

Вот код для оценки гомографии (проверенный в MATLAB):

% initial control points
A = [51 228;  51 127; 191 127; 191 228];
B = [152 57; 219 191;  62 240;  92 109];
A = circshift(A, [-1 0]);  % fix the order of points to match the picture

% input image
%I = imread('peppers.png');
I = im2uint8(checkerboard(32,5,7));
[h,w,~] = size(I);

% adapt control points to image size
% (basically we estimate an affine transform from 3 corner points)
aff = cp2tform(A(1:3,:), [1 1; w 1; w h], 'affine');
A = tformfwd(aff, A);
B = tformfwd(aff, B);

% estimate homography between A and B
T = cp2tform(B, A, 'projective');
T = fliptform(T);
H = T.tdata.Tinv

Я получаю:

>> H
H =
   -0.3268    0.6419   -0.0015
   -0.4871    0.4667    0.0009
  324.0851 -221.0565    1.0000

Теперь давайте визуализировать точки:

% check by transforming A points into B
%{
BB = [A ones(size(A,1),1)] * H;        % convert to homogeneous coords
BB = bsxfun(@rdivide, BB, BB(:,end));  % convert from homogeneous coords
%}
BB = tformfwd(T, A(:,1), A(:,2));
fprintf('error = %g\n', norm(B-BB));

% visually check by plotting control points and transformed A
figure(1)
subplot(121)
plot(A([1:end 1],1), A([1:end 1],2), '.-', 'MarkerSize',20, 'LineWidth',2)
line(BB([1:end 1],1), BB([1:end 1],2), 'Color','r', 'Marker','o')
text(A(:,1), A(:,2), num2str((1:4)','a%d'), ...
    'VerticalAlign','top', 'HorizontalAlign','left')
title('A'); legend({'A', 'A*H'}); axis equal ij
subplot(122)
plot(B([1:end 1],1), B([1:end 1],2), '.-', 'MarkerSize',20, 'LineWidth',2)
text(B(:,1), B(:,2), num2str((1:4)','b%d'), ...
    'VerticalAlign','top', 'HorizontalAlign','left')
title('B'); legend('B'); axis equal ij

control_points

Наконец, мы можем применить преобразование исходного изображения:

% transform input image and show result
J = imtransform(I, T);
figure(2)
subplot(121), imshow(I), title('image')
subplot(122), imshow(J), title('warped')

warped_image

Ответ 2

Ваша проблема в том, что вы случайно обрезали выходное изображение, когда указали XData и YData в imtransform. Один из вариантов - использовать tformfwd для преобразования A для вычисления допустимых диапазонов XData и YData.

[U,V] = tformfwd(T, A(:,1), A(:,2));

Z = imtransform(X,T,'XData',[min(U) max(U)], 'YData', [min(V) max(V)]);