乐于分享
好东西不私藏

两天用AI+rust撸了一款本地批量去水印软件,30MB,效果能打

两天用AI+rust撸了一款本地批量去水印软件,30MB,效果能打

去水印是个常见需求,市面上的解决方案不少——以小程序为主,电脑端的画图软件也能处理。至于网页上的 AI 工具,它们会帮你把水印去掉,但下载时又会给你加个新的水印(除非你付钱)。总之,一旦图片量大,操作就会变得非常繁琐。

所以我干脆自己写了一款批量去水印软件。它不是网页应用,不需要上传图片,所有处理都在你本地运行。

技术方面,我选择了 Rust,主要为了减小打包体积。图像数据处理也直接用了 Rust 的图像库,没有上 Python 的 NDArray。

一开始我打算用一个相对简单的算法:采集水印周围的环境像素,然后逐个进行泛化填充,直到把水印区域填满。这个思路可以实现去水印,但对背景变化较大的图片效果就不太理想了。

后来我转向 AI 模型。网上很多人推荐 Lama,但最终我选了 migan。一是实测下来两者的去水印效果差不多,二是 migan 的体积要小得多。

当然,整个过程没写一行代码,claude code + deepseekV4 全搞定

最终打包出来的软件只有 30 MB,实际表现也挺令人满意。 工具连接在这,大家有空可以试试:

https://www.yysoftqa.com/yy_pic_agent.html

效果图如下: 去水印前

去水印后