自动检测图片水印区域并去除

[WOSHILIDAYE]

本人小白，还是折腾了 2 ，3 天，先找了几个开源项目，自动检测水印区域还是不太靠谱（基于我的图片）。

于是经过多个模型的摸索，先用labelme标记自己需要去的水印，前期就标记了几十张。

然后用用脚本将标记的数据，生成掩码，然后用milesial/Pytorch-UNet训练，用于找出水印区域

找出水印区域后，再拿着识别到的 mask 在Sanster/IOPaint去掉水印

效果的话，勉强能用，过得去，此贴在于抛砖引玉，始终觉得这个方案不是很完美。

像 https://dewatermark.ai/zh-CN 这个网站提供的在线服务，基本可以很快的自动识别水印位置（不需要手动涂抹）并去掉水印。

要是可以达到这个网站的效果就完美了，各位大大有更好的方案吗?

[kaizceo8]

牛的

[kaizceo8]

看错了，以为网站是博主做的

[qsnow6]

思路应该是放在如何找出水印的区域，水印的规律就是以一定间隔重复，有没有什么方式可以识别出图片中重复的元素？

[importmeta]

花钱买国内云厂商的 API, 一毛多一次, 百度阿里都有.

[mightybruce]

你自己都是小白就不要瞎搞什么模型训练， 用用 openai 的 GPT-Image-1 轻松解决各种图像编辑问题包括这种去水印。

[a566]

要是能去视频的移动的就 nb

[WOSHILIDAYE]

@a566 视频的很多都是提帧处理的，要是能实现图片自动识别和移除，那么视频也就可以了

[WOSHILIDAYE]

@mightybruce 如果图片比较少那是自然，那如果每天都需要处理几千张图片，那个费用也不少的呢

[wsbqdyhm]

如果水印都差不多的，可以 ps 的批处理？

[Wesly]

能去马赛克吗

[yuxian]

恰好本人也在研究这个，想要自己训练标注还是很难的。我认为最好的办法是先识别出来，水印的位置，给到用户确认，再将水印位置的蒙版，提交给擦去物体的模型，应该就很好的实现去水印的效果。比如这个模型，https://github.com/Anonym0u3/AttentiveEraser

[a566]

@WOSHILIDAYE 等大佬的杰作

[WOSHILIDAYE]

@yuxian 问题就在这里，擦除修复现在已经有很多现成的方案，关键是自动识别水印区域，目前还没找到比较好用的方案，我发出来的那个网站就很厉害，在很快的时间自动识别到了水印区域并且去除。