opencv 把图像BGR转RGB,且hwc转chw(python)
·
方法一:
import cv2
# 读取图像
image = cv2.imread("test.png")
# 将颜色格式转为RGB
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
# 将通道格式hwc转为chw
image = image.transpose((2, 0, 1))
print(image.shape)
在opencv里,图像通道格式默认为hwc(0,1,2),颜色格式为BGR,故transpose((2,0,1))改变通道位置,cvtColor为颜色空间转换函数。(注意安装:pip install opencv-python 或 conda install opencv-python)
方法二:
import cv2
# 读取图像
image = cv2.imread("test.png")
# 将数据颜色格式转为RGB,通道格式转为chw
image = image[:, :, ::-1].transpose((2, 0, 1))
print(image.shape)
(1)image[:, :, ::-1]对应hwc(height, width, channel), ::-1代表c顺序相反操作。
(2)image[:, :, 0]代表B蓝色通道分量;image[:, :, 1]代表G绿色通道分量;image[:, :, 0]代表R红色通道分量。
注:
方法二引文:https://blog.csdn.net/qq_39938666/article/details/86701344
::-1操作理解:python[::-1]和[-1]用法-CSDN博客
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)