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)
答案 0 :(得分:0)
在 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()