用Java写一个图像去噪的代码

小鱼儿 2024-03-27 18:01 2阅读 0赞

下面是一个简单的 Java 代码示例，它可以使用中值滤波器去除图像中的噪点：

import java.awt.image.BufferedImage;
    import java.awt.image.DataBufferByte;
    
    public class MedianFilter {
        public static BufferedImage apply(BufferedImage image) {
            int width = image.getWidth();
            int height = image.getHeight();
            BufferedImage filteredImage = new BufferedImage(width, height, BufferedImage.TYPE_BYTE_GRAY);
    
            // 获取图像数据
            byte[] pixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
            byte[] filteredPixels = ((DataBufferByte) filteredImage.getRaster().getDataBuffer()).getData();
    
            // 应用中值滤波器
            for (int y = 1; y < height - 1; y++) {
                for (int x = 1; x < width - 1; x++) {
                    int[] window = new int[9];
                    int index = 0;
                    for (int j = -1; j <= 1; j++) {
                        for (int i = -1; i <= 1; i++) {
                            window[index++] = pixels[(y + j) * width + x + i] & 0xFF;
                        }
                    }
                    Arrays.sort(window);
                    filteredPixels[y * width + x] = (byte) window[4];
                }
            }
    
            return filteredImage;
        }
    }

上面的代码使用了中值滤波器，它通过计算图像中每个像素周围 3x3 窗口内的中位数来去除噪点。它首先将窗口内的像素值排序，然后使用中位数替换图像中当前像素的值。

注意，上面的代码仅适用于灰度图像。如果要处理彩色图像，则需要对每个颜色分量(R、G、B)单独应用中值滤波器。