基于背景变换下目标注意力一致性的目标分割模型

摘要

本发明涉及一种基于背景变换下目标注意力一致性的目标分割模型，该目标分割模型基于pix2pix模型实现，其生成器由三个分支编码网络、注意力模块和解码器组成；判别器也包括一个注意力模块，判别器接收生成的图像和真实分割结果，注意力模块区分真实的和生成的图像，引导生成器关注需要改进的区域；损失函数包括对抗损失函数、L1损失函数、生成器和鉴别器的分类损失函数及目标注意力一致性损失。本发明将背景移除的图像作为具有相同目标、不同背景图像集合的一个典型范例，并分别约束集合中每个图像和他们共同的背景移除的图像的在目标区域的注意力一致性，增强了不同背景下所有图像的注意力一致性，提高了对目标区域的注意力和目标分割的准确性。