图像处理Java代码-椒盐噪声、胡椒噪声、盐噪声

谁借莪１个温暖的怀抱￠ 2023-05-31 07:57 7阅读 0赞

### 图像处理-椒盐噪声、胡椒噪声、盐噪声 ###

package com.ydfind.image.biz.util;
    
    import com.ydfind.image.util.ImgUtils;
    import org.springframework.stereotype.Component;
    import org.springframework.stereotype.Service;
    
    import javax.imageio.ImageIO;
    import java.awt.image.BufferedImage;
    import java.io.File;
    import java.io.IOException;
    import java.util.Random;
    
    /**
     * 椒盐噪声 service
     * @author ydfind
     */
    @Component
    @Service
    public class PepperSaltNoiseProcessor extends ImageProcessor {
    
        public static final String PEPPER_SALT_NOISE_NAME = "椒盐噪声";
    
        public static final String PEPPER_NOISE_NAME = "胡椒噪声";
    
        public static final String SALT_NOISE_NAME = "盐噪声";
    
        /**
         * 添加椒盐噪声：根据信噪比，随机把点设置为0或1
         * @param srcImg 原图
         * @param trgImg 目标图像
         * @param param 信噪比
         * @param type 椒盐，椒，盐
         */
        public static void addPepperSaltNoise(BufferedImage srcImg, BufferedImage trgImg, double param, String type){
            int total = srcImg.getWidth() * srcImg.getHeight();
            int count = new Double(total * (1 - param)).intValue();
            Random random = new Random();
            for(int i = 0; i < count; i++) {
                int randomX = random.nextInt(srcImg.getWidth());
                int randomY = random.nextInt(srcImg.getHeight());
                int[] channels = ImgUtils.getChannelColor(srcImg, randomX, randomY);
                int newColor;
                if(PEPPER_SALT_NOISE_NAME.equals(type)){
                    newColor = (random.nextInt(2) + 1) % 2 == 0 ? 0 : 255;
                }else if(PEPPER_NOISE_NAME.equals(type)){
                    newColor = 0;
                }else{
                    newColor = 255;
                }
                for(int k = ImgUtils.COLOR_CHANNEL_COUNT - 1; k > 0; k--){
                    channels[k] = newColor;
                }
                int color = ImgUtils.colorToRgb(channels);
                trgImg.setRGB(randomX, randomY, color);
    
            }
        }
    
        /**
         * 添加椒盐噪声：根据信噪比，随机把点设置为0或1
         * @param srcFilename 原图
         * @param trgFilename 目标图像
         * @param param 信噪比
         * @throws IOException 报错
         */
        public static void addPepperSaltNoise(String srcFilename, String trgFilename, double param) throws IOException {
            BufferedImage srcImg = ImageIO.read(new File(srcFilename));
            BufferedImage trgImg = ImgUtils.copyImage(srcImg);
            addPepperSaltNoise(srcImg, trgImg, param, PEPPER_SALT_NOISE_NAME);
            ImgUtils.saveImage(trgFilename, trgImg);
        }
    
        /**
         * 添加椒盐噪声：根据信噪比，随机把点设置为0或1
         * @param srcFilename 原图
         * @param trgFilename 目标图像
         * @param param 信噪比
         * @param type 0椒盐，1椒，2盐
         * @throws IOException 报错
         */
        public static void addNoise(String srcFilename, String trgFilename, double param, String type) throws IOException {
            BufferedImage srcImg = ImageIO.read(new File(srcFilename));
            BufferedImage trgImg = ImgUtils.copyImage(srcImg);
            addPepperSaltNoise(srcImg, trgImg, param, type);
            ImgUtils.saveImage(trgFilename, trgImg);
        }
    }

/**
         * 获取图像指定位置的grb
         * @param img 图像
         * @param x x位置
         * @param y y位置
         * @return agrb颜色值
         */
        public static int[] getChannelColor(BufferedImage img, int x, int y){
            int color = img.getRGB(x, y);
            int[] channels = new int[4];
            for(int k = ImgUtils.COLOR_CHANNEL_COUNT - 1; k >= 0; k--){
                channels[k] = color & 0xff;
                color >>= 8;
            }
            return channels;
        }

测试类如下：

/**
         * 胡椒噪声
         */
        public void pepperAndReHarmonicFilter(double param) throws IOException {
            String src = Constant.TEST_IMAGE_DIR + "5/5-1-512x512.png";
            String trg = FileUtils.suffixToFilename(src, "-pepper-" + param);
            PepperSaltNoiseProcessor.addNoise(src, trg, param, PepperSaltNoiseProcessor.PEPPER_NOISE_NAME);
        }
    
        /**
         * 盐噪声
         */
        public void salt(double param) throws IOException {
            String src = Constant.TEST_IMAGE_DIR + "5/5-1-512x512.png";
            String trg = FileUtils.suffixToFilename(src, "-salt-" + param);
            PepperSaltNoiseProcessor.addNoise(src, trg, param, PepperSaltNoiseProcessor.SALT_NOISE_NAME);
        }
        /**
         * 文件名称 添加后缀
         * @param filename 原文件名称 old.png
         * @param suffix 后缀 -suffix
         * @return 新文件名称old-suffix.png
         */
        public static String suffixToFilename(String filename, String suffix){
            if(filename.indexOf(FILE_JOIN_MARK) == -1) {
                return "";
            }
            int dotIndex = filename.lastIndexOf(FILE_JOIN_MARK);
            String result = filename.substring(0, dotIndex) + suffix + filename.substring(dotIndex, filename.length());
            return result;
        }