numpy.fliplr()用法

np.fliplr()

矩阵左右翻转，在二维的情况下很容易理解。维度更高的话一开始没有搞懂，实际也挺简单。

原来矩阵的写成下面这样:

代码语言：javascript复制

[   [[0,1], [2, 3]]
	[[4,5], [6, 7]] ]

将 [0, 1], [2, 3], [4, 5], [6, 7] 看成二维情况下的一个数。那么左右交换就是 [2, 3] 与 [0, 1] 交换… 最后变成:

代码语言：javascript复制

[[[2, 3], [0, 1]]
[[6, 7], [4, 5]] ]

用于图像的水平翻转

对图像进行翻转，只需要将图像每个通道的矩阵左右翻转即可。

代码语言：javascript复制

def fliplr(x):
    if x.ndim == 3:
        x = np.transpose(np.fliplr(np.transpose(x, (0, 2, 1))), (0, 2, 1))
    elif x.ndim == 4:
        for i in range(x.shape[0]):
            x[i] = np.transpose(
                np.fliplr(np.transpose(x[i], (0, 2, 1))), (0, 2, 1))
    return x.astype(float)
img = torch.from_numpy(fliplr(img.numpy())).float()

原 img 是一个tensor，C x H x W. 将img转换为 ndarray 作为fliplr函数的参数. 上面如果没有np.transpose，就是将图像上下翻转了。

tensor transpose

0 人点赞