本人小白,还是折腾了 2 ,3 天,先找了几个开源项目,自动检测水印区域还是不太靠谱(基于我的图片)。
于是经过多个模型的摸索,先用labelme标记自己需要去的水印,前期就标记了几十张。
然后用用脚本将标记的数据,生成掩码,然后用milesial/Pytorch-UNet训练,用于找出水印区域
找出水印区域后,再拿着识别到的 mask 在Sanster/IOPaint去掉水印
效果的话,勉强能用,过得去,此贴在于抛砖引玉,始终觉得这个方案不是很完美。
像 https://dewatermark.ai/zh-CN 这个网站提供的在线服务,基本可以很快的自动识别水印位置(不需要手动涂抹)并去掉水印。
要是可以达到这个网站的效果就完美了,各位大大有更好的方案吗?