将PIL图像转换为tensor张量

两种方法都是使用transforms

第一个方法

from torchvision import transforms
import os
from PIL import Image

root_dir = "G:\MachineVision\PythpnPrj\proj003"
label_dir = "img/train/ants_image/0013035.jpg"
img_path = os.path.join(root_dir,label_dir)

img = Image.open(img_path)
img.show()              #显示一下图片
print(img)              #查看图片信息
to_tensor = transforms.ToTensor()  #必须进行实例化
img_tensor = to_tensor(img)
print(img_tensor)           #输出图片转为tensor后的张量值

os.path.join()函数起到python中的将路径结合的作用

第二个方法

from torchvision import transforms
import os
from PIL import Image

root_dir = "G:\MachineVision\PythpnPrj\proj003"
label_dir = "img/train/ants_image/0013035.jpg"
img_path = os.path.join(root_dir,label_dir)

img = Image.open(img_path)
img.show()              #显示一下图片
print(img)              #查看图片信息
pilimag = transforms.PILToTensor()        #这里不一样
totemso = pilimag(img)
print(totemso)
print(totemso.shape)

两种方法相似，只是用的transforms中的类不一样