一种基于深度学习的图像去水印方法及装置

本申请公开了一种基于深度学习的图像去水印方法及装置。所述基于深度学习的图像去水印方法包括：获取待去水印图像；获取经过训练的YOLOv5模型；将所述待去水印图像输入至所述YOLOv5模型，从而获取水印区域框；对水印区域框进行阈值分割从而获取分割水印区域；对分割水印区域内的水印进行去除，从而获取去除水印的图像。本申请的基于深度学习的图像去水印方法采用YOLO模型对水印区域进行识别，相对于现有技术而言，水印区域框识别更加准确，且所用模型轻量化设计，不需要较大的空间存储。