opencv读取图片HWC转CHW

Opencv读取图片HWC转CHW

在读取图片时，一般可采取PIL或cv2实现，但是二者读取得到的数据格式有些不同。

对于给定的RGB图片，当使用PIL加载图片时，可将其直接转换为[C,H,W]的tensor，并且此时格式为RGB。

img = torch.ByteTensor(torch.ByteStorage.from_buffer(img.tobytes()))

当采用cv2读取图片时，不同之处有两点：

• 读取的维度为[H,W,C]；
• 颜色通道为BGR，而不是RGB。

因此，在PIL和CV2读取图片间进行转换时，需要同时注意维度和RGB通道。

具体地，将HxWxC转换为CxHxW：

• 通过transpose()实现，分别img.transpose(0,1)和img.transpose(0,2)即可；
• 也可直接使用permute()，即img.permute(2,0,1)

此外，若设计BGR转RGB，则首先需要对通道进行逆序，通过下面命令实现。

img = img[::-1,:,:]

参考资料：http://www.zzvips.com/article/205586.html