k-means-gpu

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

完全受 kmeans-colors 的启发。

限制

由于此工具将图像作为纹理加载到图形卡上，因此它还受到 GPU 后端的限制：如果原始图像大于 8192x8192 像素，则不会工作。

示例

创建用 k-means 变体替换颜色的图像

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

创建使用 k-means 减少颜色的抖动图像

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