如何使用OpenCV交换图像中的蓝色和红色通道

9

我在交换图像的通道(特别是红色和蓝色)时遇到了一些问题。我正在使用Opencv 3.0.0和Python 2.7.12。以下是我用于交换通道的代码:

import cv2

img = cv2.imread("input/car1.jpg")

#The obvious approach
Cimg = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

#Manual Approach
red = img[:,:,2]
blue = img[:,:,0]

img[:,:,0] = red
img[:,:,2] = blue

cv2.imshow("frame",Cimg)
cv2.imshow("frame2", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

我无法弄清楚为什么同一张图片经过相同(可能)的操作会产生两个不同的输出。有人能解释一下出了什么问题吗?
原始图片 原始图片 手动操作 手动操作 COLOR_BGR2RGB cv2.COLOR_BGR2RGB 操作
1个回答

14

红色蓝色只是您图像的视图。当您执行img[:,:,0] = red时,这会更改img,但也会更改blue,因为它只是一个视图(基本上只是对子数组img[:,:,0]的引用),而不是副本,因此您会丢失原始的蓝色通道值。基本上您认为是一个临时副本实际上并不是。添加.copy()即可解决问题。

img = np.arange(27).reshape((3,3,3))

red = img[:,:,2].copy()
blue = img[:,:,0].copy()

img[:,:,0] = red
img[:,:,2] = blue

print("with copy:\n", img)

img = np.arange(27).reshape((3,3,3))

red = img[:,:,2]
blue = img[:,:,0]

img[:,:,0] = red
img[:,:,2] = blue

print("without copy:\n",img)

结果:

带有复制文本:

 [[[ 2  1  0]
  [ 5  4  3]
  [ 8  7  6]]

 [[11 10  9]
  [14 13 12]
  [17 16 15]]

 [[20 19 18]
  [23 22 21]
  [26 25 24]]]

没有复制:

 [[[ 2  1  2]
  [ 5  4  5]
  [ 8  7  8]]

 [[11 10 11]
  [14 13 14]
  [17 16 17]]

 [[20 19 20]
  [23 22 23]
  [26 25 26]]]

注意:实际上你只需要一个临时副本来存储一个通道。或者你可以简单地执行img[:,:,::-1],这将创建一个视图,但通道会被交换,img将保持不变,除非你重新赋值它:

img = np.arange(27).reshape((3,3,3))

print(img[:,:,::-1])
print(img)
img = img[:,:,::-1]
print(img)

结果:

[[[ 2  1  0]
  [ 5  4  3]
  [ 8  7  6]]

 [[11 10  9]
  [14 13 12]
  [17 16 15]]

 [[20 19 18]
  [23 22 21]
  [26 25 24]]]


[[[ 0  1  2]
  [ 3  4  5]
  [ 6  7  8]]

 [[ 9 10 11]
  [12 13 14]
  [15 16 17]]

 [[18 19 20]
  [21 22 23]
  [24 25 26]]]


[[[ 2  1  0]
  [ 5  4  3]
  [ 8  7  6]]

 [[11 10  9]
  [14 13 12]
  [17 16 15]]

 [[20 19 18]
  [23 22 21]
  [26 25 24]]]

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接