[发明专利]一种基于对抗网络的图像去水印的方法

摘要

本发明特别涉及一种基于对抗网络的图像去水印的方法。该基于对抗网络的图像去水印的方法，将带水印图片输入到对抗网络的生成器中，生成消除水印的图片并将其送入判别器中，区分是生成的图片还是真实图片，计算损失；通过生成器和判别器的对抗过程逐渐优化对抗网络模型，将带水印图片输入到训练好的对抗网络的生成器中，生成消除水印的图片即可。该基于对抗网络的图像去水印的方法，不需要标注水印位置，减少了人工标注数据集位置的成本；模型简单，可以实现端到端的训练与推测；不仅采用三种损失函数大大提升了去水印效果，能够适应不同尺度的水印；而且对抗模型能够学习不同类别，不同样式水印的潜在先验知识，泛化性强。