python opencv 中将图像由BGR转换为CHW用于后期的深度训练

最新推荐文章于 2021-10-12 15:09:14 发布

alex_陈

最新推荐文章于 2021-10-12 15:09:14 发布

阅读量5.6k

点赞数 1

分类专栏： opencv

本文链接：https://blog.csdn.net/qq_21904665/article/details/79346990

版权

opencv 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

BGR HWC -> CHW 12 -> HCW 01 -> CHW

import cv2 as cv
import  numpy as np

img = cv.imread("lenna.png")
#BGR HWC -> CHW  12 -> HCW 01 -> CHW
transform_img = img.swapaxes(1,2).swapaxes(0,1)
print(img.shape)
print(transform_img.shape)

cv.imshow("image0 ",transform_img[0])
cv.imshow("image1",transform_img[1])
cv.imshow("image2",transform_img[2])
cv.waitKey(0)
cv.destroyAllWindows()