Применение оператора Собеля в numpy к простому двоичному изображению - PullRequest
Новая
       54

Применение оператора Собеля в numpy к простому двоичному изображению

0 голосов
/ 26 марта 2019

Я пытаюсь применить оператор Собеля к простому двоичному изображению, но результирующий градиент переворачивается (при сравнении с выводом функции signal.convolve2d Сципи. 

from scipy import signal
import numpy as np
import matplotlib.pyplot as plt


def sobelx_homemade(arr, kx):
    offset = 1
    sx = np.zeros(arr.shape)
    for y in range(offset, arr.shape[0] - offset):
        for x in range(offset, arr.shape[1] - offset):
            rstart, rend = y-offset, y+offset+1
            cstart, cend = x-offset, x+offset+1
            w = arr[rstart:rend, cstart:cend]
            Ix = kx * w
            sx[y, x] = Ix.sum()
    return sx

A = np.array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                   [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                   [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
                   [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                   [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                   [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]])


kx = np.array([
    [-1, 0, 1],
    [-2, 0, 2],
    [-1, 0, 1]
], dtype=np.float)

ky = np.array([
    [1, 2, 1], 
    [0, 0, 0], 
    [-1, -2, -1]
], dtype = np.float)

Gx = signal.convolve2d(A, kx, boundary='symm', mode='same')
Gy = signal.convolve2d(A, ky, boundary='symm', mode='same')

# calculate homemade sobel x gradient
myGx = sobelx_homemade(A, kx)

plt.subplot(131)
plt.title("Original")
plt.imshow(A, cmap="gray")
plt.subplot(132)
plt.title("Gx")
plt.imshow(Gx, cmap="gray")
plt.subplot(133)
plt.title("My Gx")
plt.imshow(myGx, cmap="gray")

output of script, gx and my gx should match

Я ожидаю, что изображения с пометкой "Gx" и "My Gx" будут идентичны.

1 Ответ

0 голосов
/ 26 марта 2019

Итак, получается, что true свертка переворачивает матрицу ядра / фильтра, что объясняет перевернутый результат.

Это видео от Эндрю Нга объясняет различия между сверткой учебника и взаимной корреляцией, обычно используемой в обработке изображений в стиле машинного обучения: https://youtu.be/tQYZaDn_kSg?t=308

...