pytorch 把图片数据转化成tensor的操作

在图像识别当中，一般步骤是先读取图片，然后把图片数据转化成tensor格式，再输送到网络中去。本文将介绍如何把图片转换成tensor。解释如下：PIL.Image.open()得到HWC格式，直接使用numpy 去转换得到格式，而用to_tensor得到格式且值已经除了255。byte()相当于to，tensor.numpy()是把tensor 转化为numpy.array格式。在这里需要注意的是PIL和OPENCV的图像读取得到的格式都是HWC格式，一般模型训练使用的是CHW格式, H为Y轴是竖直方向，W为X轴水平方向。