【问题标题】:Reduce the an image to fixed numbers of given colors in JAVA在 JAVA 中将图像减少到固定数量的给定颜色
【发布时间】:2020-03-17 11:07:05
【问题描述】:

我想以特殊方式编辑图像:

我已经给出了 40 种颜色的列表。现在我有一张所有可能颜色的图像。我想将图像减少到只有 12 种颜色(40 种可能的颜色)并获得最佳效果。有没有一个好的库来存档(JAVA)。

【问题讨论】:

  • 您可以过滤每个像素并计算它最接近哪一种预定义颜色。我不知道是否有一个可以定义自己颜色的库。或许你能找到灵感here或here
  • 问题是没有找到最接近的颜色。我已经给出了 40 种颜色,但我只想使用其中的 12 种。这 12 个应该是最佳的
  • 检查像素并确定图像中有多少种颜色。使用 40 种颜色中的每一种,检查像素并确定每个像素需要更改多少红色、绿色和蓝色。将 40 种颜色中的每一种的变化相加。选择总和最低的 12 个。查找颜色校正以获得在求和过程中使用的公式。
  • @MullisS 您希望代码实际修改图像,还是只返回/打印 12 种解决方案颜色?我已经发布了答案,但如果您还希望使用解决方案颜色修改实际图像,我很乐意对其进行修改。
  • 你能解释一下为什么你需要做这样的事情吗?它是什么样的形象?它有几种颜色?

标签: java image editing


【解决方案1】:

在这个答案中,我做出以下假设:

  • 您的列表包含40 不同颜色。
  • 每种颜色都包含24 位信息(8 位表示红色,8 位表示绿色,8 位表示蓝色)。
  • 12-color 解决方案应该是人类感知颜色的最佳组合。

因为您的目标是确定哪些12 颜色(在提供的40 中)最接近图像中的所有颜色,您可以使用以下算法:

  1. 遍历输入颜色的所有40。
  2. 对于每种颜色,迭代输入图像中的所有像素。
  3. 对于当前像素,计算其颜色与我们当前颜色之间的差异。
    • 这可能是算法中最困难的部分,因为我们需要一个函数来返回两种颜色之间的差异。
    • 我认为实现这一点最有效的方法是先将RGB颜色转换为XYZ颜色空间,然后将XYZ颜色转换为CIELAB颜色空间。
    • 然后我们可以使用this formula 来计算两种 CIELAB 颜色之间的差异。

  1. 将颜色映射到相应颜色的差异总和。
  2. 按值对Map<Integer, Double>进行排序(升序);解决方案是第一个12 键。

这是实现此目的的 Java 代码:

import javax.imageio.ImageIO;
import java.awt.Color;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.nio.file.Paths;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadLocalRandom;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class ColorDifference {

    public static void main(String[] args) {
        ConcurrentMap<Color, Double> colorDifferenceMap = new ConcurrentHashMap<>();
        ExecutorService executorService = Executors.newFixedThreadPool(
            Runtime.getRuntime().availableProcessors());

        BufferedImage inputImage;

        try {
            // Read in the input image
            inputImage = ImageIO.read(Paths.get("input.png").toFile());
        } catch (IOException e) {
            throw new UncheckedIOException("Unable to read input image!", e);
        }

        generateInfiniteColors().distinct().limit(40).forEach(color -> {
            executorService.execute(() -> {
                CIELab cieLabColor = CIELab.from(color);

                double sum = 0d;

                for (int y = 0; y < inputImage.getHeight(); y++) {
                    for (int x = 0; x < inputImage.getWidth(); x++) {
                        Color pixelColor = new Color(inputImage.getRGB(x, y));
                        CIELab pixelCIELabColor = CIELab.from(pixelColor);
                        sum += cieLabColor.difference(pixelCIELabColor);
                    }
                }

                colorDifferenceMap.put(color, sum);
            });
        });

        executorService.shutdown();

        try {
            executorService.awaitTermination(Long.MAX_VALUE, TimeUnit.DAYS);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        // The 12 solution colors are held in this list
        List<Color> colorSolutions = colorDifferenceMap.entrySet()
            .stream()
            .sorted(Map.Entry.comparingByValue())
            .limit(12)
            .map(Map.Entry::getKey)
            .collect(Collectors.toList());

        colorSolutions.forEach(System.out::println);

        for (int y = 0; y < inputImage.getHeight(); y++) {
            for (int x = 0; x < inputImage.getWidth(); x++) {
                CIELab cieLabColor = CIELab.from(new Color(inputImage.getRGB(x, y)));

                int finalX = x;
                int finalY = y;

                colorSolutions.stream()
                    .min(Comparator.comparingDouble(color -> 
                        cieLabColor.difference(CIELab.from(color))))
                    .ifPresent(closestColor -> 
                        inputImage.setRGB(finalX, finalY, closestColor.getRGB()));
            }
        }

        try {
            ImageIO.write(inputImage, "png", new File("output.png"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // Inspiration taken from https://stackoverflow.com/a/20032024/7294647
    private static Stream<Color> generateInfiniteColors() {
        return Stream.generate(() -> 
            new Color(ThreadLocalRandom.current().nextInt(0x1000000)));
    }

    static class CIELab {

        private final double L, a, b;

        public CIELab(double L, double a, double b) {
            this.L = L;
            this.a = a;
            this.b = b;
        }

        public double difference(CIELab cieLab) {
            return Math.sqrt(Math.pow(cieLab.L - L, 2) + Math.pow(cieLab.a - a, 2) +
                Math.pow(cieLab.b - b, 2));
        }

        public static CIELab from(Color color) {
            int sR = color.getRed();
            int sG = color.getGreen();
            int sB = color.getBlue();

            // Convert Standard-RGB to XYZ (http://www.easyrgb.com/en/math.php)
            double var_R = ( sR / 255d );
            double var_G = ( sG / 255d );
            double var_B = ( sB / 255d );

            if ( var_R > 0.04045 ) var_R = Math.pow( ( var_R + 0.055 ) / 1.055, 2.4 );
            else                   var_R = var_R / 12.92;
            if ( var_G > 0.04045 ) var_G = Math.pow( ( var_G + 0.055 ) / 1.055, 2.4 );
            else                   var_G = var_G / 12.92;
            if ( var_B > 0.04045 ) var_B = Math.pow( ( var_B + 0.055 ) / 1.055, 2.4 );
            else                   var_B = var_B / 12.92;

            var_R = var_R * 100;
            var_G = var_G * 100;
            var_B = var_B * 100;

            double X = var_R * 0.4124 + var_G * 0.3576 + var_B * 0.1805;
            double Y = var_R * 0.2126 + var_G * 0.7152 + var_B * 0.0722;
            double Z = var_R * 0.0193 + var_G * 0.1192 + var_B * 0.9505;

            // Convert XYZ to CIELAB (http://www.easyrgb.com/en/math.php
            double var_X = X / 96.422;
            double var_Y = Y / 100.000;
            double var_Z = Z / 82.521;

            if ( var_X > 0.008856 ) var_X = Math.pow( var_X, 1D / 3D );
            else                    var_X = ( 7.787 * var_X ) + ( 16D / 116 );
            if ( var_Y > 0.008856 ) var_Y = Math.pow( var_Y, 1D / 3D );
            else                    var_Y = ( 7.787 * var_Y ) + ( 16D / 116 );
            if ( var_Z > 0.008856 ) var_Z = Math.pow( var_Z, 1D / 3D );
            else                    var_Z = ( 7.787 * var_Z ) + ( 16D / 116 );

            double L = ( 116 * var_Y ) - 16;
            double a = 500 * ( var_X - var_Y );
            double b = 200 * ( var_Y - var_Z );

            return new CIELab(L, a, b);
        }
    }
}

为了测试代码,我从您在评论中发送的链接中抓取了以下473x356 图片:

你可以在代码中看到所有40的颜色都是随机生成的,这是程序的输出:

java.awt.Color[r=182,g=176,b=32]
java.awt.Color[r=201,g=201,b=55]
java.awt.Color[r=142,g=149,b=6]
java.awt.Color[r=223,g=158,b=36]
java.awt.Color[r=226,g=143,b=63]
java.awt.Color[r=144,g=83,b=39]
java.awt.Color[r=119,g=153,b=117]
java.awt.Color[r=50,g=136,b=72]
java.awt.Color[r=244,g=118,b=59]
java.awt.Color[r=69,g=79,b=52]
java.awt.Color[r=149,g=78,b=76]
java.awt.Color[r=62,g=190,b=28]

一个可能的输出图像如下:


上面的代码可以在 Java 8+ 上运行,对于上面发布的图像,它在一秒钟内完成。

【讨论】:

  • 这看起来很不错,但我认为结果也应该包含一些绿色和蓝色,因为您想要最好的结果而不是 12 种颜色?如何遍历所有 40 种颜色和像素并找到最常用的颜色。将此颜色放入结果集中。现在再次遍历所有 40 种颜色和所有像素并找到最常用的颜色,但如果该像素的壁橱颜色在结果集中,则忽略该像素。但结果集中的下一个颜色。
  • @MullisS 该解决方案不包含主要为绿色或蓝色的颜色,因为我特意包含了主要为红色的12 颜色,而图像本身主要包含红色像素。如果您想在解决方案中看到更多的绿色/蓝色,那么您有两个选择:均匀分布40 输入颜色以包含更少的红色,或者选择更绿色/蓝色的图像。我将编辑我的答案以使用您发布的链接中包含的向日葵图像。
  • @MullisS 我刚刚用一个更好的例子编辑了我的答案。
  • 谢谢。这周我会尝试用你的代码检查我的结果并给出反馈
  • 工作完美 =)
【解决方案2】:

解决方案并不像给定的那样简单(请参阅 Jacob 的回答)。为了理解原因,假设我们在图像中有 10 个像素:9 个红色和 1 个绿色。假设我们在调色板中有 39 个红色和 1 个绿色。最佳答案是 11 个红色和 1 个绿色(完美匹配)。提出的算法会选择12个偏红的。

要找到真正的最佳结果,您需要遍历所有可能的 12 色调色板,以找到一个使总误差(颜色距离之和,或平方距离之和)最小化的调色板,因为通常的做法是最小化平方误差)。

40 色调色板中 12 色调色板的数量为:

C(40,12) = 40! / (12! * 28!) = 5,586,853,480

数量很多,因此需要一些处理时间。当然,这取决于图像大小。

【讨论】:

  • 生成所有 55 亿个组合实际上是我想出的第一个答案,但后来我做了数学并意识到无论如何计算解决方案都是相同的(所以组合不需要生成)。我承认对于尺寸小于或等于 12 像素的输入图像,我的答案确实分崩离析,但输入图像永远不会那么小(OP 表示他们想使用代码生成像素艺术),所以我仍然认为我的解决方案是最佳的。
  • @JacobG。我在示例中采用了 10 个像素,但无论大小如何,问题都相同。如果您拍摄一百万像素的图像,其中大部分是红色和少量绿色,您的解决方案仍然会忽略绿色。
猜你喜欢
  • 2013-10-23
  • 1970-01-01
  • 2013-07-06
  • 2016-04-06
  • 2011-08-19
  • 2013-01-15
  • 1970-01-01
  • 2012-10-08
相关资源
最近更新 更多