Matlab旋转目标时保持尺寸不变的异常问题求助

问题背景 #

需在MATLAB中旋转目标时保持原始尺寸不变，当前代码有时会出现目标被裁剪的情况，使用imrotate的"loose"参数也无法解决：

原代码 #

close all;
clear;
clc;
  
x = 256*rand(1,4);
y = 256*rand(1,4);
x(end+1) = x(1);
y(end+1) = y(1);

BW = poly2mask(x,y,ceil(max(y))+50,ceil(max(x))+50);
imshow(BW)

LI = bwlabel(BW);
mea = regionprops(LI, 'Orientation');
RI = imrotate(BW, -mea(1).Orientation,'crop');
imshow(RI);

out = imtile({BW,RI});
imshow(out)

问题现象 #

• 裁剪问题示例：
  
  
• 使用"loose"参数的效果：
  

解决思路 #

问题核心是初始掩码画布的padding不足：原代码仅固定添加50像素padding，旋转时目标可能超出画布范围；crop模式会裁剪到原始尺寸导致目标缺失，loose模式会扩大画布无法保持尺寸。

解决方案需先计算旋转所需的最小画布尺寸，创建足够大的初始掩码，旋转后再裁剪回原始尺寸。

修正代码 #

close all;
clear;
clc;
  
x = 256*rand(1,4);
y = 256*rand(1,4);
x(end+1) = x(1);
y(end+1) = y(1);

% 1. 获取目标边界框
x_min = min(x); x_max = max(x);
y_min = min(y); y_max = max(y);
bbox_width = x_max - x_min;
bbox_height = y_max - y_min;

% 2. 计算旋转后边界框的最小尺寸
temp_mask = poly2mask(x,y,ceil(y_max),ceil(x_max));
theta = regionprops(bwlabel(temp_mask), 'Orientation').Orientation;
theta_rad = deg2rad(-theta);
rot_width = abs(bbox_width*cos(theta_rad)) + abs(bbox_height*sin(theta_rad));
rot_height = abs(bbox_width*sin(theta_rad)) + abs(bbox_height*cos(theta_rad));

% 3. 计算所需padding，创建足够大的初始掩码
pad_size = max(ceil((rot_width - bbox_width)/2), ceil((rot_height - bbox_height)/2)) + 10; % 额外加10像素容错
BW = poly2mask(x + pad_size, y + pad_size, ceil(y_max) + 2*pad_size, ceil(x_max) + 2*pad_size);

% 4. 旋转后裁剪回原始尺寸
LI = bwlabel(BW);
mea = regionprops(LI, 'Orientation');
RI = imrotate(BW, -mea(1).Orientation, 'crop');
original_height = ceil(max(y)) + 50;
original_width = ceil(max(x)) + 50;
% 从旋转后图像中心裁剪至原始尺寸
RI_cropped = imcrop(RI, [(size(RI,2)-original_width)/2, (size(RI,1)-original_height)/2, original_width, original_height]);

% 对比显示
out = imtile({BW(1:original_height,1:original_width), RI_cropped});
imshow(out)

关键说明 #

• 先通过临时掩码获取目标旋转角度，计算旋转后边界框的尺寸，确保初始掩码的padding足够容纳旋转后的目标。
• 旋转后从图像中心裁剪回原始尺寸，既保证尺寸不变，又完整保留目标内容。

内容的提问来源于stack exchange，提问作者alirazi