java - 索贝尔滤波器很奇怪

Question

最近我正在准备像这样的Low Poly:

我的第一件事是选取输入图像的一些点。我想尝试使用 Sobel 进行检测，我已阅读 soble article .

现在我遇到了问题，我不知道如何实现索贝尔过滤器。以下是我的尝试:

首先 -> 从输入图像中获取灰度数据

 try {
        mImage = ImageIO.read(new File("D:\\Documents\\Pictures\\engine.png"));
    } catch (IOException e) {
        e.printStackTrace();
    }
    mWidth = mImage.getWidth();
    mHeight = mImage.getHeight();
    mNewImage = new BufferedImage(mWidth, mHeight, mImage.getType());
    mGrayData = new int[mWidth * mHeight];
    // get pixel data from image
    for (int i = 0; i < mHeight; i++) {
        for (int j = 0; j < mWidth; j++) {
            int rgb = mImage.getRGB(j, i);
            int r = (rgb >> 16) & 0xff;
            int g = (rgb >> 8) & 0xff;
            int b = rgb & 0xff;
            int grayLevel = (r + g + b) / 3;
            mGrayData[i * mWidth + j] = grayLevel;
        }
    }

然后 -> 计算每个点的梯度

  private int[] getGradient() {
    int[] gradient = new int[mWidth * mHeight];
    for (int x=1;x<mWidth-1;x++){
        for (int y=1;y<mHeight-1;y++){
            int grayX = getGrayPoint(x+1,y-1)+2*getGrayPoint(x+1,y)+getGrayPoint(x+1,y+1)-
                    (getGrayPoint(x-1,y-1)+2*getGrayPoint(x-1,y)+getGrayPoint(x-1,y+1));
            int grayY = (getGrayPoint(x-1, y+1) + 2*getGrayPoint(x,y+1)+getGrayPoint(x+1,y+1))-
                    (getGrayPoint(x-1,y-1) + 2*getGrayPoint(x,y-1) + getGrayPoint(x+1,y-1));
            gradient[x+y*mWidth] = (int) Math.sqrt(grayX*grayX+grayY*grayY);
        }
    }
    return gradient;
}

最后 -> 使用上述渐变创建新图像

 private void createImage() {
    int[] gradient = getGradient();

    for (int y = 1; y < mHeight - 1; ++y)
        for (int x = 1; x < mWidth - 1; ++x)
            mNewImage.setRGB(x,y,gradient[y * mWidth + x]);

    File file = new File("D:\\Documents\\Pictures\\engine3.png");
    if (!file.exists()) {
        file.getParentFile().mkdir();
        try {
            file.createNewFile();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    try {
        ImageIO.write(mNewImage, "png", file);
    } catch (IOException e) {
        e.printStackTrace();
    }
}

原始图像

它应该是什么样子？

我得到了什么

颜色有点蓝，太奇怪了

编辑1:

现在我得到:

为什么这么黑？

Answer 1

您可以在此处设置颜色:

mNewImage.setRGB(x,y,gradient[y * mWidth + x]);

您没有采取任何措施来确保红色、绿色和蓝色 channel 中的值相同。

该整数的 32 位用于打包 4 个 8 位 channel ，分别代表 alpha、红、绿和蓝。

0x01234567 = overall color

  01       = alpha
    23     = red
      45   = green
        67 = blue

您仅设置一个值；这看起来并不超出 0 到 255 的范围，因此您实际上只是在蓝色 channel 中设置位。 (大概您还使用 RGB 颜色模型，而不是 ARGB，这就是像素不完全透明的原因)。

将三个 channel 设置为相同的值(并将值限制在 0-255 之间，这样您就不会意外地设置其他 channel 中的位):

int gray = Math.max(0, Math.min(gradient[...], 255);
int color = (0xff << 24) | (gray << 16) | (gray << 8) | (gray);
          // Alpha         Red            Green         Blue

请注意，您可能还需要缩放渐变值，以便为具有最大渐变的点分配灰度值 255:

int maxGradient = 0;
for (int g : gradient) { 
  maxGradient = Math.max(maxGradient, g);
}

然后:

int gray = Math.max(0, gradient) * 255 / maxGradient;

(您不再需要将其限制为最多 255)。

此外，您可能希望使渐变数组 float[] 或 double[] (并为 maxGradient 使用相同的元素类型)，这样你就可以得到更少的量化输出。