python - 如何将Gabor滤镜应用于具有六边形采样的图像？

在将正方形采样的图像转换为六角形采样的图像之后，我想使用Gabor滤波器作为插值方法。

我可以在六边形采样的图像中使用普通的Gabor滤波器实现吗？还是应该修改代码？如果是，那么我应该为六角形采样的图像修改Gabor函数的哪一部分？

我已经尝试实现算法，但是我做对了。
这是Gabor过滤taken from GitHub的代码。

import numpy as np
import cv2

# cv2.getGaborKernel(ksize, sigma, theta, lambda, gamma, psi, ktype)
# ksize - size of gabor filter (n, n)
# sigma - standard deviation of the gaussian function
# theta - orientation of the normal to the parallel stripes
# lambda - wavelength of the sunusoidal factor
# gamma - spatial aspect ratio
# psi - phase offset
# ktype - type and range of values that each pixel in the gabor kernel can hold

g_kernel = cv2.getGaborKernel((21, 21), 8.0, np.pi/4, 10.0, 0.5, 0, ktype=cv2.CV_32F)

img = cv2.imread('test.jpg')
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
filtered_img = cv2.filter2D(img, cv2.CV_8UC3, g_kernel)

cv2.imshow('image', img)
cv2.imshow('filtered image', filtered_img)

h, w = g_kernel.shape[:2]
g_kernel = cv2.resize(g_kernel, (3*w, 3*h), interpolation=cv2.INTER_CUBIC)
cv2.imshow('gabor kernel (resized)', g_kernel)
cv2.waitKey(0)
cv2.destroyAllWindows()

最佳答案

假定六边形网格的普通数据结构，您可能可以对六边形像素图像应用2d滤镜，但是您需要创建一个在适当坐标处求值的滤镜。您会看到，二维过滤器只是通过对网格中xy坐标上的函数求值而生成的值的矩阵。因此，一个5x5的Gabor滤波器矩阵只是在如下所示的xy坐标处评估的Gabor函数：

像素是“等间距”的，因此我们可以简单地选择网格中每个点之间的距离，以x为1，以y为1，然后在每个像素的中心处评估Gabor函数。

但是，六边形像素不是这样排列的。六角形像素的中心按以下方式排列：

因此，为了对此应用过滤器，我们需要在这些点上评估适当的功能。由于股票过滤器是在矩形网格上评估的，因此我们无法使用它们（尽管它们会产生看起来合理的东西）。

幸运的是，转换相对容易。如果我们假设两行之间的垂直距离为1，则坐标几乎只是一个np.arange。

import numpy as np
import matplotlib.pyplot as plt

ALTERNATE_ROW_SHIFT = 0+np.sqrt(3)/3 # every other row is "offset" by half a hexagon.  If the sides are len 2/3, the shift is root 3 over 3

def hex_grid(rect_grid):
    rect_grid = np.copy(rect_grid)
    rect_grid[0,:,1::2] += ALTERNATE_ROW_SHIFT
    return rect_grid

如果您有权访问创建过滤器的函数，通常会有一些逻辑可以创建一个矩形网格，然后在该矩形网格上对函数进行评估。之后将hex_grid函数放到该行上，以获取六角形间隔的坐标。

例如，the wikipedia page on Gabor filters具有创建Gabor过滤器的python实现，如下所示：

def gabor_fn(sigma, theta, Lambda, psi, gamma):
    sigma_x = sigma
    sigma_y = float(sigma) / gamma

    # Bounding box
    nstds = 3 # Number of standard deviation sigma
    xmax = max(abs(nstds * sigma_x * np.cos(theta)), abs(nstds * sigma_y * np.sin(theta)))
    xmax = np.ceil(max(1, xmax))
    ymax = max(abs(nstds * sigma_x * np.sin(theta)), abs(nstds * sigma_y * np.cos(theta)))
    ymax = np.ceil(max(1, ymax))
    xmin = -xmax
    ymin = -ymax

    (y,x) = np.meshgrid(np.arange(ymin, ymax + 1), np.arange(xmin, xmax + 1))

    # Rotation
    x_theta = x * np.cos(theta) + y * np.sin(theta)
    y_theta = -x * np.sin(theta) + y * np.cos(theta)

    gb = np.exp(-.5 * (x_theta ** 2 / sigma_x ** 2 + y_theta ** 2 / sigma_y ** 2)) * np.cos(2 * np.pi / Lambda * x_theta + psi)
    return gb

请注意涉及np.meshgrid的行。这将创建一个间距为1的矩形网格，该网格将用于后续的行。我们可以简单地转换这些坐标以创建一个新的hex_gabor函数（请注意，这与gabor_fn代码具有95％的相同性）：

def hex_gabor_fn(sigma, theta, Lambda, psi, gamma):
    sigma_x = sigma
    sigma_y = float(sigma) / gamma

    # Bounding box
    nstds = 3 # Number of standard deviation sigma
    xmax = max(abs(nstds * sigma_x * np.cos(theta)), abs(nstds * sigma_y * np.sin(theta)))
    xmax = np.ceil(max(1, xmax))
    ymax = max(abs(nstds * sigma_x * np.sin(theta)), abs(nstds * sigma_y * np.cos(theta)))
    ymax = np.ceil(max(1, ymax))
    xmin = -xmax
    ymin = -ymax

    yx = np.meshgrid(np.arange(ymin, ymax + 1), np.arange(xmin, xmax + 1))
    (y,x) = hex_grid(yx)

    # Rotation
    x_theta = x * np.cos(theta) + y * np.sin(theta)
    y_theta = -x * np.sin(theta) + y * np.cos(theta)

    gb = np.exp(-.5 * (x_theta ** 2 / sigma_x ** 2 + y_theta ** 2 / sigma_y ** 2)) * np.cos(2 * np.pi / Lambda * x_theta + psi)
    return gb

if __name__ == "__main__":
    g = gabor_fn(4,np.pi/4,4,0,2)
    hg = hex_gabor_fn(4,np.pi/4,4,0,2)
    plt.imshow(g)
    plt.show()
    plt.imshow(hg)
    plt.show()

您应该能够将生成的内核放入此行cv2.filter2D(img, cv2.CV_8UC3, g_kernel)。

关于python - 如何将Gabor滤镜应用于具有六边形采样的图像？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55013954/