图像 -----＞ array与tensor的不同

清风拂杨柳#

已于 2022-05-13 17:55:46 修改

阅读量1.2k

点赞数

文章标签： pytorch

于 2022-05-08 17:29:24 首次发布

本文链接：https://blog.csdn.net/weixin_56405212/article/details/124651238

版权

这篇博客探讨了图像从numpy数组到tensor的转换过程。在数组中，图像以(H,W,C)的顺序存储，数值范围0-255，而在tensor中则变为(C,H,W)，数值范围0-1。通过自定义函数trans，图像完成转换。文章通过实例展示了切片操作在数组和tensor中的相同性和差异性，强调了理解不同数据结构在处理图像时的重要性。

摘要由CSDN通过智能技术生成

#image的size为（w,h）即（w = 1000，h = 500）

img_arr  = np.array(image)或者 img_arr = np.asarray(image) #将image转换成array

img_tesnor = trans(image)   #image转换成tensor，其中函数trans自定义的。

测试可知，image在array中表示为（H,W,C）

在tensor中表示为（C,H,W）

且可以看出，在array中表示为0-255，在tensor中表示为0-1

path="img地址"
img = Image.open(path)

arr = np.array(img)
print(arr.shape)
print(arr[0].shape)

tensor = trans(img)    #trans为自定义函数
print(tensor.shape)
print(tensor[0].shape)

输出

(375, 500, 3)
(500, 3)
torch.Size([3, 375, 500])
torch.Size([375, 500])

arr[0]表示取数组第0维度第0位置上的切片

tensor[0]表示取tensor第0维度上的第0位置的切片。

再如

print(arr.shape[1])
print(tensor[1].shape)

#输出
(375, 500, 3)
(500, 3)
torch.Size([3, 375, 500])
torch.Size([375, 500])

再如

print(arr[1,10].shape)
print(tensor[1,10].shape)

#输出
(375, 500, 3)
(3,)
torch.Size([3, 375, 500])
torch.Size([500])

总结：关于切片，数组和tensor是一样的。

print(arr.shape)
print(arr[1].shape)
print(arr[1].shape[0])

print(tensor.shape)
print(tensor[1].shape)
print(tensor[1].shape[0])

#输出##############################################
(375, 500, 3)
(500, 3)
500
torch.Size([3, 375, 500])
torch.Size([375, 500])
375

对于后续在shape后面再加一个[i]是说（此时的shape内容其实是一个数组）对于当前的一个内容为（375，500，3）的数组进行按索引取值。你应该能看懂。