当前位置:   article > 正文

图像 -----> array与tensor的不同_tensor和array区别

tensor和array区别
  1. #image的size为(w,h)即(w = 1000,h = 500
  2. img_arr = np.array(image)或者 img_arr = np.asarray(image) #将image转换成array
  3. img_tesnor = trans(image) #image转换成tensor,其中函数trans自定义的。

测试可知,image在array中表示为(H,W,C)

tensor中表示为(C,H,W)

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



  1. path="img地址"
  2. img = Image.open(path)
  3. arr = np.array(img)
  4. print(arr.shape)
  5. print(arr[0].shape)
  6. tensor = trans(img) #trans为自定义函数
  7. print(tensor.shape)
  8. print(tensor[0].shape)

输出

  1. (375, 500, 3)
  2. (500, 3)
  3. torch.Size([3, 375, 500])
  4. torch.Size([375, 500])

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

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

再如

  1. print(arr.shape[1])
  2. print(tensor[1].shape)
  3. #输出
  4. (375, 500, 3)
  5. (500, 3)
  6. torch.Size([3, 375, 500])
  7. torch.Size([375, 500])

再如

  1. print(arr[1,10].shape)
  2. print(tensor[1,10].shape)
  3. #输出
  4. (375, 500, 3)
  5. (3,)
  6. torch.Size([3, 375, 500])
  7. torch.Size([500])

总结:关于切片,数组和tensor是一样的。

  1. print(arr.shape)
  2. print(arr[1].shape)
  3. print(arr[1].shape[0])
  4. print(tensor.shape)
  5. print(tensor[1].shape)
  6. print(tensor[1].shape[0])
  7. #输出##############################################
  8. (375, 500, 3)
  9. (500, 3)
  10. 500
  11. torch.Size([3, 375, 500])
  12. torch.Size([375, 500])
  13. 375

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/木道寻08/article/detail/1005387
推荐阅读
相关标签
  

闽ICP备14008679号