沿 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.