pytorch框架--数据方面--数据类转换演示

from PIL import Image
from torchvision import transforms
import cv2
import numpy as np
img_path = "./data/train/Cat/10.jpg"
# 方法一
img = Image.open(img_path)
# 使用array数据类型也可，一般直接用cv2读取,因为cv2就是array类型
# img = np.array(img)

# 方法二
# img = cv2.imread(img_path)

# 实例化
tensor = transforms.ToTensor()
# 对象(图像)
tensor_img = tensor(img)
print(tensor_img)