使用 Sobel 边缘检测方法时出错

Posted 2023-03-16

【中文标题】使用 Sobel 边缘检测方法时出错

【英文标题】：Error while using Sobel edge detection method

【发布时间】：2021-10-07 07:03:18

【问题描述】：

img = cv2.imread('skeleton.JPG')

gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)


sobelx = cv2.Sobel(gray, cv2.CV_64F, 1, 0,ksize=5)
sobely = cv2.Sobel(gray, cv2.CV_64F, 0, 1,ksize=5)

plt.imshow(cv2.cvtColor(sobelx, cv2.COLOR_BGR2RGB))

我收到如下错误，

error                                     Traceback (most recent call last)
<ipython-input-51-3e640684ed0a> in <module>
----> 1 plt.imshow(cv2.cvtColor(sobelx, cv2.COLOR_BGR2RGB))

error: OpenCV(4.2.0) c:\projects\opencv-python\opencv\modules\imgproc\src\color.simd_helpers.hpp:94: error: (-2:Unspecified error) in function '__cdecl cv::impl::`anonymous-namespace'::CvtHelper<struct cv::impl::`anonymous namespace'::Set<1,-1,-1>,struct cv::impl::A0xe227985e::Set<3,4,-1>,struct cv::impl::A0xe227985e::Set<0,2,5>,2>::CvtHelper(const class cv::_InputArray &,const class cv::_OutputArray &,int)'
> Unsupported depth of input image:
>     'VDepth::contains(depth)'
> where
>     'depth' is 6 (CV_64F)

【问题讨论】：

你忘了问一个实际问题。有用的答案是什么？告诉你为什么你会得到错误？告诉你如何修复错误？请明确！

也就是说，您的错误信息非常清楚：CV_64F 是不受支持的深度。尝试使用不同的深度值。

【参考方案1】：

在 Sobel 运算中，计算 sobelx 和 sobely 之后，你必须计算 np.sqrt(sobelx ** 2 + sobely ** 2)。之后，您应该选择一个阈值来查找边缘。此外，您将获得黑白图像，因此无法将其转换为 RGB 或 BGR 图像。 正确的代码是：

import cv2
import matplotlib.pyplot as plt
import numpy as np

threshold = 6.

img = cv2.imread('skeleton.JPG')

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) / 255

sobelx = cv2.Sobel(gray, cv2.CV_64F, 1, 0, ksize=5)
sobely = cv2.Sobel(gray, cv2.CV_64F, 0, 1, ksize=5)
sobel_xy = np.sqrt(sobelx ** 2 + sobely ** 2)
edge_image = np.float32(sobel_xy > threshold)

plt.imshow(edge_image, 'gray')
plt.show()

如果你想显示sobelx，你可以应用np.abs并将其标准化然后显示：

sobelx_disp = np.abs(sobelx)
sobelx_disp = sobelx_disp / np.max(sobelx_disp)
plt.imshow(sobelx_disp, 'gray')
plt.show()