numpy.fliplr()用法

狼啸风云

修改于 2022-09-02 19:53:36

1K0

文章被收录于专栏：计算机视觉理论及其实现计算机视觉理论及其实现

np.fliplr()

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

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

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

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

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

用于图像的水平翻转

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

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，就是将图像上下翻转了。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/02/08 ，如有侵权请联系 cloudcommunity@tencent.com 删除

tensor