一种基于Transformer的图像去水印方法及系统

本发明公开了一种基于Transformer的图像去水印方法及系统，该方法包括：获取带水印图像并构建数据集；构建Transformer多尺度注意力网络模型并利用所述数据集进行训练，得到训练完成的去水印模型；所述去水印模型包括编码器、隐藏层和解码器，其中，所述解码器设有水印定位检测分支和背景图像修复分支，所述编码器和所述解码器中均设有Transformer块；将待处理图像输入至所述训练完成的去水印模型进行处理，输出去水印图像。该系统包括：数据集构建模块、模型训练模块和去水印模块。通过使用本发明，能够提高去水印模型的泛化性，进而满足不同类型的水印去除任务。本发明可广泛应用于图像处理领域。