沿 y=x 有效翻转/转置图像

Question

我想像这样沿 y=x 轴翻转图像。

我已经制作了这个函数来做我想做的事，但我想知道是否有更优化的方法来做到这一点。我做的功能在处理大图时有点慢

def flipImage(img):
    # Get image dimensions
    h, w = img.shape[:2]
    # Create a image
    imgYX = np.zeros((w, h, 3), np.uint8)
    for y in range(w):
        for x in range(h):
            imgYX[y,x,:]=img[x,y,:] #Flip pixels along y=x
    return imgYX

Answer 1

只需swap the first two axes即对应高度和宽度-

img.swapaxes(0,1) # or np.swapaxes(img,0,1)

我们也可以用transpose 置换坐标轴-

img.transpose(1,0,2) # or np.transpose(img,(1,0,2))

我们也可以roll axes达到同样的效果-

np.rollaxis(img,0,-1)

We use the same trick when working with images in MATLAB.