我遇到了同样的麻烦,我终于设法找到了一个与我测试的图像配合得很好的解决方案。我使用的是我想在图像上应用剪切、旋转和平移变换的事实。
- 首先,我们计算旋转所需的空间(不需要
计算如果不使用)。
- 其次,我们计算图像的最后一个像素所需的空间,该像素将通过剪切在图像中具有最大平移。
- 第三,我们需要计算必要的平移,以补偿剪切将移动图像中心这一事实。
如果您希望所有图像的尺寸相同,请使用 45 度角,因为这个角度需要更多空间。
import numpy as np
import cv2
#Parameters of the affine transform:
angle = 45; #Angle in degrees.
shear = 1;
translation = 5;
type_border = cv2.BORDER_CONSTANT;
color_border = (255,255,255);
original_image = cv2.imread(name_image_file);
rows,cols,ch = original_image.shape;
#First: Necessary space for the rotation
M = cv2.getRotationMatrix2D((cols/2,rows/2), angle, 1);
cos_part = np.abs(M[0, 0]); sin_part = np.abs(M[0, 1]);
new_cols = int((rows * sin_part) + (cols * cos_part));
new_rows = int((rows * cos_part) + (cols * sin_part));
#Second: Necessary space for the shear
new_cols += (shear*new_cols);
new_rows += (shear*new_rows);
#Calculate the space to add with border
up_down = int((new_rows-rows)/2); left_right = int((new_cols-cols)/2);
final_image = cv2.copyMakeBorder(original_image, up_down, up_down,left_right,left_right,type_border, value = color_border);
rows,cols,ch = final_image.shape;
#Application of the affine transform.
M_rot = cv2.getRotationMatrix2D((cols/2,rows/2),angle,1);
translat_center_x = -(shear*cols)/2;
translat_center_y = -(shear*rows)/2;
M = M_rot + np.float64([[0,shear,translation + translat_center_x], [shear,0,translation + translat_center_y]]);
final_image = cv2.warpAffine(final_image , M, (cols,rows),borderMode = type_border, borderValue = color_border);
示例(我将“cv2.copyMakeBorder”和“cv2.getRotationMatrix2D”的边框设置为白色):
角度 = -45 度。剪切 = -0.5