java - 二维图像中的 Gabor 小波

Question

我使用 lire, java对于图像检索项目，我想为输入图像的每个像素计算 gabor 小波的振幅 ref here . lire 库中的默认函数返回一个下采样的特征向量。我想要的是 gabor 小波的计算幅度。我正在尝试创建一个函数 getMagnitude:

public double[][] getMagnitude(BufferedImage image) {
    image = ImageUtils.scaleImage(image, MAX_IMG_HEIGHT);

    Raster imageRaster = image.getRaster();
    System.out.println(imageRaster);
    int[][] grayLevel = new int[imageRaster.getWidth()][imageRaster.getHeight()];

    int[] tmp = new int[3];
    for (int i = 0; i < imageRaster.getWidth(); i++) {
        for (int j = 0; j < imageRaster.getHeight(); j++) {
            grayLevel[i][j] = imageRaster.getPixel(i, j, tmp)[0];
        }
    }

    double[][] magnitudes = computeMagnitudes(grayLevel);

    return magnitudes;
}

gabor 特征是类 gabor 私有(private)变量:

private static final double U_H = .4;
private static final double U_L = .05;
private static final int S = 1, T = 1; // filter mask size
private static final int M = 4, N = 5; // scale & orientation

private static final int MAX_IMG_HEIGHT = 64;

此外，最终幅度数组的大小似乎受 M、N 尺度和方向的大小影响。我必须做什么才能得到我想要的案例？

计算幅度函数:

 private double[][] computeMagnitudes(int[][] image) {
    double[][] magnitudes = new double[M][N];
    for (int i = 0; i < magnitudes.length; i++) {
        for (int j = 0; j < magnitudes[0].length; j++) {
            magnitudes[i][j] = 0.;
        }
    }

    if (this.gaborWavelet == null) {
        precomputeGaborWavelet(image);
    }

    for (int m = 0; m < M; m++) {
        for (int n = 0; n < N; n++) {
            for (int x = S; x < image.length; x++) {
                for (int y = T; y < image[0].length; y++) {
                    magnitudes[m][n] += Math.sqrt(Math.pow(this.gaborWavelet[x - S][y - T][m][n][0], 2) + Math.pow(this.gaborWavelet[x - S][y - T][m][n][1], 2));

                }
            }
        }
    }
    return magnitudes;
}

Answer 1

我认为你必须使用 5d private double[][][][][] gaborWavelet，它实际上是计算图像和母小波之间的卷积。

 private double[] gaborWavelet(int[][] img, int x, int y, int m, int n) {
    double re = 0;
    double im = 0;
    for (int s = 0; s < S; s++) {
        for (int t = 0; t < T; t++) {
            re += img[x][y] * selfSimilarGaborWavelets[s][t][m][n][0];
            im += img[x][y] * -selfSimilarGaborWavelets[s][t][m][n][1];
        }
    }

    return new double[]{re, im};
}

您必须选择 M、N、S、T，返回的小波包含 re 和 im 两个矩阵。