PIL.Image转tensor和name ‘transforms‘ is not defined的解决

仅作为记录，大佬请跳过。

出现name 'transforms' is not defined的解决：

import torchvision
transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor()]) 

即可。

---

PIL.Image转tensor

from PIL import Image
import torchvision
img = Image.open('/Data4T/samples/her2_3.png').convert('RGB')
print(type(img))
img=transform(img)
print(type(img))

---

或者直接使用

from torchvision import datasets, models, transforms

data_transforms={transforms.Compose([
        transforms.Resize([1024,512]),
        transforms.ToTensor(),
        transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
    ])}