python - 在图像上应用 sobel 过滤器

Question

我正在尝试对墙壁图像使用 sobel 过滤器，但它不起作用。

我的代码是:

im=scipy.misc.imread('IMG_1479bis.JPG')
im = im.astype('int32')
dx=ndimage.sobel(im,1)
dy=ndimage.sobel(im,0)
mag=np.hypot(dx,dy)
mag*=255.0/np.max(mag)
cv2.imshow('sobel.jpg', mag)

我真的不明白我的错误在哪里。任何帮助，将不胜感激 !提前致谢!

Answer 1

由于石头和混凝土的纹理，这是一张很难应用简单边缘检测的图像。纹理使它几乎就像你有一个非常嘈杂的图像，你正在应用一阶导数。您最终会得到许多不需要的小边。

这是您的代码工作(不会导致错误):

import scipy.ndimage as nd
import numpy as np
import matplotlib.pyplot as plt

im = scipy.ndimage.imread('ygqCd.jpg', True)
im = im.astype('int32')
dx = nd.sobel(im,1)
dy = nd.sobel(im,0)
mag = np.hypot(dx,dy)
mag *= 255.0/np.max(mag) 

fig, ax = plt.subplots()
ax.imshow(mag, cmap = 'gray')
plt.xticks([]), plt.yticks([])
plt.show()

图像读取行 scipy.ndimage.imread('ygqCd.jpg', True) 正在转换为灰度。参见 ndimage.imread

和输出图像(如预期的那样非常嘈杂):

我对cv2比较熟悉。在应用 sobel 之前，我使用 cv2.sobel 和中值滤波对图像进行了一些处理。以下是结果，虽然不是很好，但却是一个好的开始:

以及生成图像的代码:

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

def interval_mapping(image, from_min, from_max, to_min, to_max):
    from_range = from_max - from_min
    to_range = to_max - to_min
    scaled = np.array((image - from_min) / float(from_range), dtype=float)
    return to_min + (scaled * to_range)

img = cv2.imread('ygqCd.jpg', 0)
blurred_img = cv2.medianBlur(img, 11)

s_mask = 17

sobelx = np.abs(cv2.Sobel(img, cv2.CV_64F, 1, 0, ksize=s_mask))
b_sobelx = np.abs(cv2.Sobel(blurred_img, cv2.CV_64F, 1, 0, ksize=s_mask))
sobelx = interval_mapping(sobelx, np.min(sobelx), np.max(sobelx), 0, 255)
b_sobelx = interval_mapping(b_sobelx, np.min(sobelx), np.max(sobelx), 0, 255)

sobely = np.abs(cv2.Sobel(img,cv2.CV_64F,0,1,ksize=s_mask))
sobely = interval_mapping(sobely, np.min(sobely), np.max(sobely), 0, 255)
b_sobely = np.abs(cv2.Sobel(blurred_img, cv2.CV_64F, 0, 1, ksize=s_mask))
b_sobely = interval_mapping(b_sobely, np.min(sobely), np.max(sobely), 0, 255)

sobel_xy = 0.5 * sobelx + 0.5 * sobely
b_sobel_xy = 0.5 * b_sobelx + 0.5 * b_sobely

fig = plt.figure(figsize=(10, 14))
plt.subplot(3,2,1),plt.imshow(sobelx,cmap = 'gray')
plt.title('Sobel X'), plt.xticks([]), plt.yticks([])
plt.subplot(3,2,2),plt.imshow(b_sobelx,cmap = 'gray')
plt.title('Blurred Sobel X'), plt.xticks([]), plt.yticks([])
plt.subplot(3,2,3),plt.imshow(sobely,cmap = 'gray')
plt.title('Sobel Y'), plt.xticks([]), plt.yticks([])
plt.subplot(3,2,4),plt.imshow(b_sobely,cmap = 'gray')
plt.title('Blurred Sobel Y'), plt.xticks([]), plt.yticks([])
plt.subplot(3,2,5),plt.imshow(sobel_xy,cmap = 'gray')
plt.title('Sobel XY'), plt.xticks([]), plt.yticks([])
plt.subplot(3,2,6),plt.imshow(b_sobel_xy,cmap = 'gray')
plt.title('Blurred Sobel XY'), plt.xticks([]), plt.yticks([])
plt.tight_layout()

plt.show()