图片替换颜色（漂白）工具类

最新推荐文章于 2025-05-05 23:54:39 发布
tlovet_1314
最新推荐文章于 2025-05-05 23:54:39 发布
阅读量371
点赞数
文章标签：图像识别
本文链接：https://blog.csdn.net/tlovet_1314/article/details/108024676
版权
package com.jeesit.api.commons.dao;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.jeesit.api.railcar.util.MongoUtil;
import com.jeesit.api.railcar.util.ObtainJsonApiUtil;
import com.jeesit.api.util.AesEncodeUtil;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;

public class text {
    public static void main(String[] args) {
        /**
         * 要处理的图片目录
         */
        File dir = new File("d:/test/11");
        /**
         * 列出目录中的图片，得到数组
         */
        File[] files = dir.listFiles();
        /**
         * 遍历数组
         */
        for(int x=0;x<files.length;x++){
            /**
             * 定义一个RGB的数组，因为图片的RGB模式是由三个 0-255来表示的 比如白色就是(255,255,255)
             */
            int[] rgb = new int[3];
            /**
             * 用来处理图片的缓冲流
             */
            BufferedImage bi = null;
            try {
                /**
                 * 用ImageIO将图片读入到缓冲中
                 */
                bi = ImageIO.read(files[x]);
            } catch (Exception e) {
                e.printStackTrace();
            }
            /**
             * 得到图片的长宽
             */
            int width = bi.getWidth();
            int height = bi.getHeight();
            int minx = bi.getMinX();
            int miny = bi.getMinY();
            System.out.println("正在处理："+files[x].getName());
            /**
             * 这里是遍历图片的像素，因为要处理图片的背色，所以要把指定像素上的颜色换成目标颜色
             * 这里 是一个二层循环，遍历长和宽上的每个像素
             */
            int a=0,b=0;
            double c=0;
            for (int i = minx; i < width; i++) {
                for (int j = miny; j < height; j++) {
                    // System.out.print(bi.getRGB(jw, ih));
                    /**
                     * 得到指定像素（i,j)上的RGB值，
                     */
                    int pixel = bi.getRGB(i, j);
                    /**
                     * 分别进行位操作得到 r g b上的值
                     */
                    Double gray=getGray(pixel);
                    c=c+gray;
                }
            }
            double d= c/(width*height);
            for (int i = minx; i < width; i++) {
                for (int j = miny; j < height; j++) {
                    // System.out.print(bi.getRGB(jw, ih));
                    /**
                     * 得到指定像素（i,j)上的RGB值，
                     */
                    int pixel = bi.getRGB(i, j);
                    /**
                     * 分别进行位操作得到 r g b上的值
                     */
                    Double gray=getGray(pixel);
                    if (gray>d){
                        a++;
//                        bi.setRGB(i, j, 0xffffff);
                    }else {
                        b++;
                    }

                }
            }
            for (int i = minx; i < width; i++) {
                for (int j = miny; j < height; j++) {
                    // System.out.print(bi.getRGB(jw, ih));
                    /**
                     * 得到指定像素（i,j)上的RGB值，
                     */
                    int pixel = bi.getRGB(i, j);
                    /**
                     * 分别进行位操作得到 r g b上的值
                     */
                    Double gray=getGray(pixel);
//                    System.out.println(gray);
//                    System.out.println(d);
                    if (a>b){
                        if (gray>d){
                            bi.setRGB(i, j, 0xffffff);
                        }else {
//                            System.out.println("gray:"+gray);
//                            System.out.println("d:"+d);
//                            System.out.println("---------");
                            bi.setRGB(i, j, 0x000000);
                        }
                    }
                    else {
                        if (gray>d){
                            bi.setRGB(i, j, 0x000000);
                        }else {
                            bi.setRGB(i, j, 0xffffff);
                        }

                    }

                }
            }
//            //去除可能出现的斑点
//            for (int i = minx; i < width; i++) {
//                for (int j = miny; j < height; j++) {
//                    // System.out.print(bi.getRGB(jw, ih));
//                    /**
//                     * 得到指定像素（i,j)上的RGB值，
//                     */
//                    int pixel = bi.getRGB(i, j);
//                    /**
//                     * 分别进行位操作得到 r g b上的值
//                     */
//                    Double gray=getGray(pixel);
                    System.out.println(gray);
                    System.out.println(d);
//                    if (a>b){
//                        if (gray>d){
                            bi.setRGB(i, j, 0xffffff);
//                        }else {
//                            if (i==0){
//                                if (j==0){
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    Double drGray=getGray(bi.getRGB(i+1, j+1));
//                                    if (rGray>d&&dGray>d&&drGray>d){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else if (j==(height-1)){
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    Double ldGray=getGray(bi.getRGB(i-1, j+1));
//                                    if (lGray>d&&dGray>d&&ldGray>d){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else {
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double urGray=getGray(bi.getRGB(i+1, j-1));
//                                    Double drGray=getGray(bi.getRGB(i+1, j+1));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    if ((isGrayValue(uGray,d)+isGrayValue(rGray,d)+isGrayValue(drGray,d)
//                                            +isGrayValue(urGray,d)+isGrayValue(dGray,d))<1){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }
//                            }else if (i==(width-1)){
//                                if (j==0){
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double urGray=getGray(bi.getRGB(i+1, j-1));
//                                    if (rGray>d&&uGray>d&&urGray>d){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else if (j==(height-1)){
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    Double luGray=getGray(bi.getRGB(i-1, j-1));
//                                    if (lGray>d&&uGray>d&&luGray>d){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else {
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    Double luGray=getGray(bi.getRGB(i-1, j-1));
//                                    Double dlGray=getGray(bi.getRGB(i-1, j+1));
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    if ((isGrayValue(lGray,d)+isGrayValue(dGray,d)+isGrayValue(luGray,d)
//                                            +isGrayValue(dlGray,d)+isGrayValue(uGray,d))<1){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }
//                            }else {
//                                if (j==0){
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double dlGray=getGray(bi.getRGB(i-1, j+1));
//                                    Double drGray=getGray(bi.getRGB(i+1, j+1));
//                                    if ((isGrayValue(lGray,d)+isGrayValue(dGray,d)+isGrayValue(rGray,d)
//                                            +isGrayValue(dlGray,d)+isGrayValue(drGray,d))<1){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else if (j==(height-1)){
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    Double luGray=getGray(bi.getRGB(i-1, j-1));
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double urGray=getGray(bi.getRGB(i+1, j-1));
//                                    if ((isGrayValue(lGray,d)+isGrayValue(uGray,d)+isGrayValue(rGray,d)
//                                            +isGrayValue(luGray,d)+isGrayValue(urGray,d))<1){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }else {
//                                    Double lGray=getGray(bi.getRGB(i-1, j));
//                                    Double uGray=getGray(bi.getRGB(i, j-1));
//                                    Double luGray=getGray(bi.getRGB(i-1, j-1));
//                                    Double rGray=getGray(bi.getRGB(i+1, j));
//                                    Double urGray=getGray(bi.getRGB(i+1, j-1));
//                                    Double dGray=getGray(bi.getRGB(i, j+1));
//                                    Double dlGray=getGray(bi.getRGB(i-1, j+1));
//                                    Double drGray=getGray(bi.getRGB(i+1, j+1));
//                                    if ((isGrayValue(lGray,d)+isGrayValue(uGray,d)+isGrayValue(rGray,d)
//                                            +isGrayValue(luGray,d)+isGrayValue(urGray,d)+isGrayValue(dGray,d)
//                                            +isGrayValue(dlGray,d)+isGrayValue(drGray,d))<1){
//                                        bi.setRGB(i, j, 0xffffff);
//                                    }
//                                }
//                            }
//                        }
//                    }
//                    else {
//                        if (gray>d){
                            bi.setRGB(i, j, 0x000000);
//                        }else {
                            bi.setRGB(i, j, 0xffffff);
//                        }
//
//                    }
//
//                }
//            }

            System.out.println("\t处理完毕："+files[x].getName());
            System.out.println();String[] strArray = files[x].getName().split("\\.");
            int suffixIndex = strArray.length -1;
            System.out.println(strArray[suffixIndex]);

            /**
             * 将缓冲对象保存到新文件中
             */
            try {
                FileOutputStream ops = new FileOutputStream(new File("d:/test/11/"+strArray[0]+"11."+strArray[suffixIndex]));
                ImageIO.write(bi,strArray[suffixIndex], ops);
                ops.flush();
                ops.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    private static Double getGray(int pixel){
        int[] rgb = new int[3];
        rgb[0] = (pixel & 0xff0000) >> 16;
        rgb[1] = (pixel & 0xff00) >> 8;
        rgb[2] = (pixel & 0xff);
        Double gray=rgb[0]*0.3+rgb[1]*0.59+rgb[2]*0.11;
        return gray;
    }
    //判断灰度是否大于平均灰度0为否1为是
    private static int isGrayValue(Double a,Double b){
        if (a>b){
            return 1;
        }else {
            return 0;
        }
    }
}