今天在实现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函数会自动对其显示灰度值进行缩放,规则如下:
可以发现,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取值表如下所示(来自官方文档)
最终的代码:
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()