如何使用枕头库创建的图像作为神经网络的输入

Question

我有一个函数，可以使用PIL库从头开始创建图像对象。我想使用此图像作为神经网络的输入，而不必保存图像并再次加载

下面的代码创建一个带有矩形的图像

from PIL import Image, ImageDraw
img_size = 300
image = Image.new(mode='RGB', size=(img_size, img_size), color=0)
draw = ImageDraw.Draw(image)
draw.rectangle([(50, 50), (250, 250)], fill=255, outline=255, width=1)
image.show()

如何使用这个物体的"图像"作为任何神经网络的输入？tensorflow和pytorch都很好，我只需要能正常工作的

Answer 1

这就是如何将PIL图像传递给torch模型
如果image是您的PIL <Image> class
则

from torchvision import transforms
pil_to_tensor = transforms.ToTensor()(image).unsqueeze_(0)
output = model(pil_to_tensor ) #