二维卷积 - 如何在 Java 中实现？ [关闭]

Question

我已经用头撞墙了好一阵子了。 我习惯于使用 Matlab 的 conv2 函数，它自己进行卷积。

但是，我只是未能在 Java 中正确实现它。我会提供一个代码 sn-p，但它都搞砸了。我会尽量弄清楚并发布（实际上是凌晨 4:30，我仍在为此苦苦挣扎..）。

任何人都可以发布一个工作代码方法，在 Java 中对两个相同大小的二维数组进行二维卷积吗？

先谢谢了！

Answer 1

一点点谷歌搜索永远不会受到伤害。 This code 可能会对您有所帮助。特别是，

public static double[][] convolution2D(double[][] input,
            int width, int height,
            double[][] kernel,
            int kernelWidth,
            int kernelHeight) {
        int smallWidth = width - kernelWidth + 1;
        int smallHeight = height - kernelHeight + 1;
        double[][] output = new double[smallWidth][smallHeight];
        for (int i = 0; i < smallWidth; ++i) {
            for (int j = 0; j < smallHeight; ++j) {
                output[i][j] = 0;
            }
        }
        for (int i = 0; i < smallWidth; ++i) {
            for (int j = 0; j < smallHeight; ++j) {
                output[i][j] = singlePixelConvolution(input, i, j, kernel,
                        kernelWidth, kernelHeight);
            }
        }
        return output;
    }