读取的 图片的格式是HxWx3,像素值在0-1之间
img = cv2.imread('b.png')
print(img.shape)
img = img.permute(2,0,1) #如果进行HWC的互换操作的话,pytorch用permute, python使用transpose
print(img.shape)
RGB数据通道的转换
img = cv2.imread('b.png')
print(img.shape)
img1 = img.copy() #改变img1的时候不改变img
img1[:,:,0] = img[:,:,2]
img1[:,:,1] = img[:,:,1]
img1[:,:,2] = img[:,:,0] #cv2读取的是bgr,转换成rgb就要做一下变通
plt.imshow(img1)
plt.show()
如果直接用PIL进行图片的阅读的话, 读出来的像素值在0-255之间,然后PIL进行RGB通道的转换十分的简单,一句话
img_read = PIL.Image.open(img_path).convert('RGB')
关于阅读出来的其他的用法,可以参考
PIL