如何解决使用 Sobel 边缘检测方法时出错
img = cv2.imread('skeleton.JPG')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
sobelx = cv2.sobel(gray,cv2.CV_64F,1,ksize=5)
sobely = cv2.sobel(gray,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,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)
解决方法
在 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,ksize=5)
sobely = cv2.Sobel(gray,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()
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。