python - 不支持的输入图像深度 : 'VDepth::contains(depth)' where 'depth' is 4 (CV_32S)

Question

我正在尝试在 OpenCV 中处理一些图像。具体来说，使用以下函数交换颜色 Pane 。

def green_ble_swap(image)
    im_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    im_copy = np.copy(im_rgb)
    blue = im_copy[:,:,2].copy()
    green = im_copy[:,:,1].copy()
    im_copy[:,:,2] = green
    im_copy[:,:,1] = blue
    return im_copy

但是我收到以下错误。

> Unsupported depth of input image:
>     'VDepth::contains(depth)'
> where
>     'depth' is 4 (CV_32S)

不确定这里的错误是什么。

Answer 1

您遇到错误是因为您试图对 4 channel 图像执行 3 channel 操作。具体来说，错误来自尝试转换 BGR图片发送至 RGB当输入图像具有透明 channel 时。正确的方法是做 cv2.COLOR_BGRA2RGB而不是 cv2.COLOR_BGR2RGB .您可以使用 cv2.split() 就地交换蓝色和绿色 channel 获取 BGR channel (用于 3 channel 图像)和 BGRA channel (4 channel 图像)然后使用 Numpy 索引交换 channel 。您还需要使用 cv2.IMREAD_UNCHANGED加载图像时标记或 alpha channel 将被删除。例子:

输入 ->输出

import cv2
import numpy as np

def green_blue_swap(image):
    # 3-channel image (no transparency)
    if image.shape[2] == 3:
        b,g,r = cv2.split(image)
        image[:,:,0] = g
        image[:,:,1] = b
    # 4-channel image (with transparency)
    elif image.shape[2] == 4:
        b,g,r,a = cv2.split(image)
        image[:,:,0] = g
        image[:,:,1] = b
    return image 

# Load image
image = cv2.imread('1.png', cv2.IMREAD_UNCHANGED)
cv2.imshow('image', image)

# Swap channels
swapped = green_blue_swap(image)
cv2.imshow('swapped', swapped)
cv2.waitKey()