笔迹消除方法、模型训练方法及相关装置

本申请提供的笔迹消除方法、模型训练方法及相关装置中，电子设备获取待处理图像以及待处理图像的颜色掩膜；其中，颜色掩膜中标记出了待处理图像中的笔迹区域，待处理图像的笔迹消除模型包括编码器以及解码器，编码器包括有特征提取层；特征提取层利用颜色掩膜中的注意力信息对待处理图像中的多种图像特征进行处理，得到特征提取层的输出特征；通过解码器对编码器的输出特征进行处理，得到消除笔迹后的干净图像。如此，直接利用颜色掩膜中的注意力信息来引导特征提取过程，使得模型能够在局部范围内集中计算资源，减少了不必要的全局计算负担，从而提升了笔迹消除过程中整体计算效率。