1、转变形状,将H,W,C转变成C,H,W,这一点可以用在pytorch的模型训练过程中。
2、转变数据格式,该API返回的数据是tensor格式的。
3、改变数据的范围,可以将数据的范围变成[0,1.0]。
4、会改变数据的维度
# 将图以灰度图的形式读取进来
img=cv.imread(r"D:\test\data\imgs\00.png",cv.IMREAD_GRAYSCALE)
img.shape
输出的结果是:(512, 512)
通过ToTensor()操作之后
image=totensor(img)
image.shape
输出的结果是:torch.Size([1, 512, 512])
从以上结果能够看出输入和输出前后维度发生了变化,从一个二维的数组变成了三维的数组。
但是以上主要是针对二维的数组来进行操作,下面针对三维的数组来做同样的操作。
img=cv.imread(r"D:\test\data\imgs\00.png")
img.shape
输出的结果:(512, 512, 3)
通过ToTensor()操作之后
image=totensor(img)
image.shape
输出的结果是:torch.Size([3, 512, 512])
所以最后得出来的结论就是无论你的图是二维的还是三维的最后的出来的结果都是三维的数组。