一种基于 CGAN 的可见水印去除方案

来源：计算机技术与发展杂志2022年第02期北京时间：

作者：王家亮;刘晓强;李柏岩;冯珍妮

摘要：自然场景下采集的卡证、文档中存在的可见水印,是影响人们阅读效率、机器识别准确度的障碍。 为此,提出了一种结合基于特征点匹配的水印检测和基于条件生成对抗网络 CGAN 的水印去除方案。 水印检测部分,通过 SIFT 特征点检测、FLANN 特征点匹配和 PROSAC 误匹配消隐,估计出几何变换的最佳透视模型实现目标水印定位。 水印去除部分采用了 pix2pix( image-to-image translation with conditional generative adversarial networks)的模型架构,它借鉴了 CGAN 的核心思想,混合了 L1 距离损失和 CGAN 损失函数,减少了输出图像的模糊度且保留了更多的正确特征。 最终能满足自然业务场景下快速且精准的水印去除需求,具有较好的水印检测鲁棒性和去水印效果。 此外,还给出了详细的 CGAN 模型所需的成对训练集扩充方式,构建了大量有效的训练集,提升了去水印模型训练的拟合优度。

关键词：图像转换;水印去除;特征点匹配;条件生成对抗网络( CGAN) ;监督学习