torch.flip旋转测试

import cv2
import torch
import numpy as np

path = './test.jpg'
img = cv2.imread(path)
print(img.shape)
image = img.transpose(2, 0, 1)
image = np.expand_dims(image, axis=0)
image = torch.Tensor(image)
print(image.shape)

flip1 = torch.flip(image, [0, 3])
flip1 = flip1.squeeze(0)
flip1 = np.array(flip1)
flip1 = flip1.transpose(1,2,0)
print(flip1.shape)
cv2.imwrite('./test_flip1.jpg', flip1)

flip2 = torch.flip(image, [1, 2])
flip2 = flip2.squeeze(0)
flip2 = np.array(flip2)
flip2 = flip2.transpose(1,2,0)
print(flip2.shape)
cv2.imwrite('./test_flip2.jpg', flip2)

flip3 = torch.flip(image, [0, 2])
flip3 = flip3.squeeze(0)
flip3 = np.array(flip3)
flip3 = flip3.transpose(1,2,0)
print(flip3.shape)
cv2.imwrite('./test_flip3.jpg', flip3)

flip4 = torch.flip(image, [1, 3])
flip4 = flip4.squeeze(0)
flip4 = np.array(flip4)
flip4 = flip4.transpose(1,2,0)
print(flip4.shape)
cv2.imwrite('./test_flip4.jpg', flip4)

          原始图                  旋转0和3                旋转1和2                 旋转0和2             旋转1和3

 cv2读入是（618，480，3），原始图像调整维度顺序并且维度增加了一维，shape是（1，3，618，480）