本人小白,还是折腾了 2 ,3 天,先找了几个开源项目,自动检测水印区域还是不太靠谱(基于我的图片)。
于是经过多个模型的摸索,先用labelme
标记自己需要去的水印,前期就标记了几十张。
然后用用脚本将标记的数据,生成掩码,然后用milesial/Pytorch-UNet
训练,用于找出水印区域
找出水印区域后,再拿着识别到的 mask 在Sanster/IOPaint
去掉水印
效果的话,勉强能用,过得去,此贴在于抛砖引玉,始终觉得这个方案不是很完美。
像 https://dewatermark.ai/zh-CN 这个网站提供的在线服务,基本可以很快的自动识别水印位置(不需要手动涂抹)并去掉水印。
要是可以达到这个网站的效果就完美了,各位大大有更好的方案吗?
![]() |
1
kaizceo8 10 天前
牛的
|
![]() |
2
kaizceo8 10 天前
看错了,以为网站是博主做的
|
![]() |
3
qsnow6 10 天前
思路应该是放在如何找出水印的区域,水印的规律就是以一定间隔重复,有没有什么方式可以识别出图片中重复的元素?
|
![]() |
4
importmeta 10 天前
花钱买国内云厂商的 API, 一毛多一次, 百度阿里都有.
|
![]() |
5
mightybruce 10 天前
你自己都是小白就不要瞎搞什么模型训练, 用用 openai 的 GPT-Image-1 轻松解决各种图像编辑问题包括这种去水印。
|
![]() |
6
a566 10 天前
要是能去视频的移动的就 nb
|
![]() |
7
WOSHILIDAYE OP @a566 视频的很多都是提帧处理的,要是能实现图片自动识别和移除,那么视频也就可以了
|
![]() |
8
WOSHILIDAYE OP @mightybruce 如果图片比较少那是自然,那如果每天都需要处理几千张图片,那个费用也不少的呢
|
9
wsbqdyhm 9 天前
如果水印都差不多的,可以 ps 的批处理?
|
10
Wesly 9 天前
能去马赛克吗
|
![]() |
11
yuxian 8 天前
恰好本人也在研究这个,想要自己训练标注还是很难的。我认为最好的办法是先识别出来,水印的位置,给到用户确认,再将水印位置的蒙版,提交给擦去物体的模型,应该就很好的实现去水印的效果。比如这个模型,https://github.com/Anonym0u3/AttentiveEraser
|
![]() |
12
a566 8 天前
@WOSHILIDAYE 等大佬的杰作
|
![]() |
13
WOSHILIDAYE OP @yuxian 问题就在这里,擦除修复现在已经有很多现成的方案,关键是自动识别水印区域,目前还没找到比较好用的方案,我发出来的那个网站就很厉害,在很快的时间自动识别到了水印区域并且去除。
|