k-means-gpu

使用k均值聚类计算图像中的k个平均颜色，利用GPU进行繁重的工作。

完全受 kmeans-colors 启发。

限制

由于此工具将图像作为纹理加载到您的显卡中，因此也带来了一些基于GPU后端的限制：如果原始图像大于 8192x8192 像素，则无法工作。

示例

创建用其k均值变体替换颜色的图像

cargo run --release -- reduce -i .\gfx\tokyo.png -c 8

创建使用k均值减少颜色的抖动图像

cargo run --release -- reduce -i .\gfx\tokyo.png -c 8 -m dither

输出调色板

cargo run --release -- palette -i .\gfx\tokyo.png -c 8 -s 40

查找颜色并将它们用作替换

cargo run --release -- find -i .\gfx\tokyo.png -p "#050505,#ffffff,#ff0000"

查找颜色并将它们用于抖动图像

cargo run --release -- find -i .\gfx\tokyo.png -p "#050505,#ffffff,#ff0000" -m dither

通过加载调色板来查找颜色以抖动图像

cargo run --release -- find -i .\gfx\tokyo.png -p .\gfx\apollo-1x.png -m dither

源代码

为了理解所有这些，我不得不阅读很多东西。

• 首先，优秀的 kmeans-colors 启发了这个项目。
• Muthukrishnan 几篇文章
  • https://muthu.co/reduce-the-number-of-colors-of-an-image-using-uniform-quantization/
  • https://muthu.co/reduce-the-number-of-colors-of-an-image-using-k-means-clustering/
  • https://muthu.co/mathematics-behind-k-mean-clustering-algorithm/
• 关于前缀和
  • https://en.wikipedia.org/wiki/Prefix_sum
  • 关于wgsl中的前缀和: https://github.com/googlefonts/compute-shader-101/blob/prefix/compute-shader-hello/src/shader.wgsl
  • https://github.com/linebender/piet-gpu/blob/prefix/piet-gpu-hal/examples/shader/prefix.comp
• 关于抖动
  • https://en.wikipedia.org/wiki/Ordered_dithering
  • http://alex-charlton.com/posts/Dithering_on_the_GPU/
• 资源
  • 由 Kerrie Lake 的64色调色板复活。
  • 由 Adam C Younis 的Apollo调色板。

许可证

MIT