批量实现 cv2.cvtColor(images, cv2.COLOR_GRAY2RGB)

本文介绍如何使用NumPy对批量灰度图像转换为RGB。通过np.expand_dims增加维度,然后利用np.tile进行通道复制,实现(32,28,28)->(32,28,28,3)的转变,适用于处理图像批次。
摘要由CSDN通过智能技术生成
  • 背景:我想在一批灰度图像上画彩色的框,那么首先需要对灰度图转为 RGB,但是 RGB 信息已经损失了,所以只能对通道复制 3 份,变成彩色的三通道,这里使用 OpenCV 的 cv2.cvtColor(images, cv2.COLOR_GRAY2RGB) 方法只能针对单张图片,所以本文给出能直接处理 batch 图像的 灰度转彩色
# init dummy mat
mat = np.zeros((32,28,28))
mat = np.expand_dims(mat, -1)
np.tile(mat, (1,1,1,3))

上面的代码实现了从 (32, 28, 28) -> (32, 28, 28, 3) 的转变

  • 解析:这里使用的 np.tile() 函数,这个函数很好理解,你有 n 个维度,那么你就需要对指定对相应维度复制的次数即可,上面的程序中,我先对 batch 在最后(-1) expand_dims 了一下,得到了(32,28,28,1),然后(1,1,1,3)是对相应的维度复制的次数
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值