本文介绍了如何在Pytorch中将图像转换为Tensor，主要涉及torchvision.transforms的to_tensor函数。该函数首先检查输入类型，如果是numpy数组，则进行维度调整并转化为Tensor；如果是PIL Image，会根据模式转换并进行CHW格式转置，最后除以255。源码中的一条注释指出，transpose操作占据了加载时间的80%，暗示可能的性能优化方向。
要学会看源码啊...
最近google 上不去，就用bing来举例好了，一般来讲，看了各种example代码以后，发现transformer 里面有一个 totensor，所以直接搜索torchvision.transforms，看里面的文档，在文档里面搜索to_tensor，里面有对这个函数的解释，然后查看源码 ，源码如下：
def to_tensor(pic):
"""Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor.
See ``ToTensor`` for more details.
Args:
pic (PIL Image or numpy.ndarray): Image to be converted to tensor.
Returns:
Tensor: Converted image.
"""
if not(_is_pil_image(pic) or _is_numpy_image(pic)):
raise TypeError('pic should be PIL Image or ndarray. Got {}'.format(type(pic)))
if isinstance(pic, np.ndarray):
# handle numpy array
if pic.ndim == 2:
pic = pic[:, :, None]
img = torch.from_numpy(pic.transp
最低0.47元/天 解锁文章
708
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
