将灰度图像转换为3通道图像

9 浏览
0 Comments

将灰度图像转换为3通道图像

我想将一个尺寸为(height,width)的灰度图像转换为一个尺寸为(height,width,nchannels)的三通道图像。目前的方法是使用一个for-loop，但肯定有更简洁的方式。以下是程序中的一段代码，请给个提示。请指导。

 30         if img.shape == (height,width): # 如果img是灰度图像，则进行扩展
 31             print "将1通道图像转换为", nchannels, "通道图像。"
 32             new_img = np.zeros((height,width,nchannels))
 33             for ch in range(nchannels):
 34                 for xx in range(height):
 35                     for yy in range(width):
 36                         new_img[xx,yy,ch] = img[xx,yy]
 37             img = new_img

0