将cv2+plt+Image读取的图片转换为tensor格式

将cv2+plt+Image读取的图片转换为tensor格式

import torchvision.transforms as trans
import matplotlib.pyplot as plt
from PIL import Image
import cv2
import numpy as np

trans1 = trans.ToTensor()

# 样例
img_sample = np.zeros([100, 200, 3])  # h, w, c
img_np_tensor = trans1(img_sample)
print(img_np_tensor.size())  # c, h, w   torch.Size([3, 100, 200])

print('\n', '------------------------------------------------------------')

img1 = Image.open('tufen_0137.jpg')
print('转换前的img1的shape,为w,h', '\n', img1.size)  # w, h
img_PIL_tensor = trans1(img1)
print('转换后的img1的shape,为c,h,w', '\n', img_PIL_tensor.size(), '\n', )  # c, h, w

img2 = cv2.imread('tufen_0137.jpg')
print('转换前的img2的shape,为h,w,c', '\n', img2.shape)  # h, w, c
img_cv2_tensor = trans1(img2)
print('转换后的img2的shape,为c,h,w', '\n', img_cv2_tensor.size(), '\n', )  # c, h, w

img3 = plt.imread('tufen_0137.jpg')
print('转换前的img3的shape,为h,w,c', '\n', img3.shape)  # h, w, c
img_plt_tensor = trans1(img3)
print('转换后的img2的shape,为c,h,w', '\n', img_plt_tensor.size(), '\n', )  # c, h, w

print('-----------------------------------------------------------------')