公开(公告)号：CN115937525A

公开(公告)日：2023-04-07

申请号：CN202310010353.9

申请日：2023-01-04

Applicant: 北京工业大学

Inventor： 张婷 ,  姜冠伦 ,  刘兆英

IPC: G06V10/26 ,  G06V10/774 ,  G06V10/82 ,  G06N3/084 ,  G06N3/047 ,  G06V10/764

Abstract: 本发明公开了一种双分支半监督红外舰船目标分割方法，首先，按照一定的比例划分训练集和测试集，并且把训练集中的数据按照不同的比例划分为有标签图像和无标签图像；其次，构建双分支半监督红外舰船目标分割模型；接着，训练模型：第一步，以改进的分割交叉熵损失作为损失函数，使用反向传播和梯度下降算法Adam训练双学习网络分支中的学习网络A、B，然后，利用训练好的学习网络A、B为无标签图像生成伪标签，接着，利用有标签数据和无标签数据训练分割网络分支中的分割网络F，以分割交叉熵损失作为损失函数，使用反向传播和梯度下降算法Adam训练分割网络F。本发明的双分支的半监督红外舰船目标分割方法有效地提高了红外舰船目标的分割性能。