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
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
8
0- 0
已为社区贡献1条内容
所有评论(0)