一、问题描述

今天在实现sobel算子时,用了cv的filter2D函数以后,报了如下错误

cv2.error: OpenCV(4.5.5) :-1: error: (-5:Bad argument) in function 'imshow'
> Overload resolution failed:
>  - mat data type = 23 is not supported
>  - Expected Ptr<cv::cuda::GpuMat> for argument 'mat'
>  - Expected Ptr<cv::UMat> for argument 'mat'

报错代码段:

v = cv.filter2D(img_i, -1, G_x)
h = cv.filter2D(img_i, -1, G_y)
img = np.sqrt(v**2 + h**2)
# 此处省略阈值处理
cv.namedWindow("Image_Sobel", cv.WINDOW_NORMAL)
cv.imshow("Image_Sobel", img)
cv.waitKey(0)
cv.destroyAllWindows()

二、解决

看到报错里面有 data type = 23 ,猜测是图像的数据类型有问题,输出一看发现img中元素的数据类型是float16,现在看看imshow的要求。
opencv的官方注释指出:根据图像的深度,imshow函数会自动对其显示灰度值进行缩放,规则如下:

  1. 如果图像数据类型是8U(8位无符号),则直接显示。
  2. 如果图像数据类型是16U(16位无符号)或32S(32位有符号整数),则imshow函数内部会自动将每个像素值除以256并显示,即将原图像素值的范围由[0~ 255*256]映射到[0~255]
  3. 如果图像数据类型是32F(32位浮点数)或64F(64位浮点数),则imshow函数内部会自动将每个像素值乘以255并显示,即将原图像素值的范围由[0~ 1]映射到[0~255](注意:原图像素值必须要归一化)

可以发现,16F是非法的,因此 需要将其转化为32F或者64F,如以下代码中的第4行所示(转成了64F)

v = cv.filter2D(img_i, -1, G_x)
h = cv.filter2D(img_i, -1, G_y)
img = np.sqrt(v**2 + h**2)
img = img.astype(float)
# 此处省略阈值处理
cv.namedWindow("Image_Sobel", cv.WINDOW_NORMAL)
cv.imshow("Image_Sobel", img)
cv.waitKey(0)
cv.destroyAllWindows()

至此,已经不会报错了,但是图像经过阈值化处理,输出是全黑的,开方应该是没有问题的,猜测是cv.filter2D的问题。直接说结论,cv.filter2D函数的第二个参数是ddepth,英文是 desired depth,即期望深度,值为-1说明输出的数据类型和输入相同,问题就出在这,输入的图像是uint8的,所以原本卷积之后是负值的全被抹掉了,后面第三行平方相加也会溢出,,,所以应该把ddepth设为CV_32F,ddepth取值表如下所示(来自官方文档)
ddepth取值表
最终的代码:

v = cv.filter2D(img_i, cv.CV_32F, G_x)
h = cv.filter2D(img_i, cv.CV_32F, G_y)
img = np.sqrt(v**2 + h**2)
# 此处省略阈值处理
cv.namedWindow("Image_Sobel", cv.WINDOW_NORMAL)
cv.imshow("Image_Sobel", img)
cv.waitKey(0)
cv.destroyAllWindows()

最后的输出的图像:
sobel算子输出

GitHub 加速计划 / opencv31 / opencv
77.39 K
55.71 K
下载
OpenCV: 开源计算机视觉库
最近提交(Master分支:3 个月前 )
7be5181b Fixed KLEIDICV_SOURCE_PATH handling for external KleidiCV 1 天前
c3ca3f4f - 2 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐