Pytorch使用PIL和Numpy将单张图片转为Pytorch张量

这里会讲解一下如何通过PIL读取单张图片将其转换为Pytorch张量
#!/usr/bin/python
# -*- coding: UTF-8 -*-

from PIL import Image
import numpy as np
import torch

pil_img = Image.open('/Users/songlu/Desktop/code.jpg')
img = np.array(pil_img)
print torch.from_numpy(img)

0 个评论

要回复文章请先登录注册