使用OpenCV处理NumPy数组连通组件时遇断言错误求助

问题描述 #

在Python中使用OpenCV进行连通组件分析时，导入.czi格式图像（通过aicsimageio转为NumPy数组）或使用随机生成的NumPy数组测试时，触发以下错误：

OpenCV(4.6.0) D:\a\opencv-python\opencv-python\opencv\modules\imgproc\src\connectedcomponents.cpp:5632: error: (-215:Assertion failed) iDepth == CV_8U || iDepth == CV_8S in function 'cv::connectedComponents_sub1'

测试代码如下：

import numpy as np
import cv2

test = np.random.rand(512,512)
(T, thresh) = cv2.threshold(test, 0.5, 255, cv2.THRESH_BINARY)
num_labels, labels = cv2.connectedComponents(thresh)

注：使用OpenCV导入.jpg图像时，相同代码可正常运行，且两种方式得到的thresh均为NumPy数组。

问题原因 #

OpenCV的connectedComponents函数要求输入图像必须是8位单通道格式（CV_8U无符号或CV_8S有符号）。np.random.rand()生成的是float64类型数组，经cv2.threshold处理后输出仍为浮点型，不符合函数格式要求；而OpenCV读取的.jpg图像默认是uint8（8位无符号整数）格式，因此可以正常执行。

解决方法 #

针对随机数组测试的修改 #

有两种可行的修改方式：

1. 生成数组时直接指定uint8类型：

import numpy as np
import cv2

# 生成0-255范围内的uint8随机数组
test = np.random.randint(0, 256, size=(512,512), dtype=np.uint8)
(T, thresh) = cv2.threshold(test, 127, 255, cv2.THRESH_BINARY)
num_labels, labels = cv2.connectedComponents(thresh)

2. 对阈值处理后的浮点数组转换格式：

import numpy as np
import cv2

test = np.random.rand(512,512)
(T, thresh) = cv2.threshold(test, 0.5, 255, cv2.THRESH_BINARY)
# 将浮点型结果转为uint8格式
thresh = thresh.astype(np.uint8)
num_labels, labels = cv2.connectedComponents(thresh)

针对.czi图像的处理 #

使用aicsimageio导入图像后，需将数组转换为uint8格式再进行后续操作：

from aicsimageio import AICSImage

# 导入.czi图像，根据图像实际维度调整参数
img = AICSImage("your_image.czi")
img_array = img.get_image_data("YX", S=0, T=0, C=0)
# 将图像缩放到0-255区间后转为uint8（若原图像位深更高）
img_array = (img_array / img_array.max() * 255).astype(np.uint8)
# 阈值处理
(T, thresh) = cv2.threshold(img_array, 127, 255, cv2.THRESH_BINARY)
# 执行连通组件分析
num_labels, labels = cv2.connectedComponents(thresh)

内容的提问来源于stack exchange，提问作者jfried