使用 imgaug 旋转后图像的形状不会改变

Question

我正在按以下方式旋转图片：

# Read in image
img = cv2.imread("pic.jpg")
height, width, _ = img.shape
print("height ", height)
print("width ", width)

# Rotate image by 90 degrees
augmentation = iaa.Affine(rotate=90)
img_aug = augmentation(image=img)

height, width, _ = img_aug.shape
print("Height after rotation ", height)
print("Width after rotation ", width

> height  1080
> width  1920
> Height after rotation  1080
> Width after rotation  1920

为什么图片的形状没有变化？

Answer 1

图像增强不会改变图像的实际或物理shape。将原始的shape 想象成一个window，您可以从那里看到图像或外部世界。这里所有的转换，即rotations、stretchings、translation 或通常任何homography 或非线性warps 都应用于您窗外的世界。无论外面的世界如何变化，即图像如何变化，您查看图像的开口（在我的比喻中是窗户）都保持不变。

现在它显然可以将原始视图中不存在的其他区域引入增强视图。大多数情况下，新引入的像素是黑色的，或者可能取决于应用了哪种padding。

简而言之，增强操作不像矩阵transpose 那样实际形状可能会发生变化。

img = cv2.imread("img.png")
augmentation = iaa.Affine(rotate=90)
img_aug = augmentation(image=img)

让我们看看图片：

Original Image

Rotated Image