方法一:

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博客

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐