[发明专利]太赫兹图像中目标物体的检测分割方法及计算机存储介质

摘要

本发明公开了一种太赫兹图像中目标物体的检测分割方法及计算机存储介质，利用条件生成对抗网络将检测和分割两个任务有机地融合在一起，对于包含武器的样本生成其分割掩膜，而对于不包含武器的样本生成一张所有像素的值为零的图片，设置合适的损失函数和高效地网络结构，在训练集较小的并且图像信噪比较低的情况下，训练出的模型能够检测输入的太赫兹图像是否包含目标物体，并且将目标物体从图像中分割出来。本发明能够能够在训练样本数量很少而且图片的信噪比很低的情况下训练出一个精确的检测和分割模型，并且该模型具有很快的处理速度和很高的时效性，可以应用在实时安检系统中，从而降低人工安检的劳动强度，提高效率。

主权项

1.一种太赫兹图像中目标物体的检测分割方法，其特征在于包括以下步骤：(1)将训练集中的原始太赫兹图像x作为训练样本，采用人工方式将x中的目标物体区域标注出来，得到真实分割掩膜y，同时对x和y进行预处理，将图像中的所有像素值转换到‑1到1之间；(2)将所述的原始太赫兹图像x和随机噪声z输入到生成器中，生成器生成假分割掩膜；(3)将真实分割掩膜和对应的原始太赫兹图像组成真实样本对，将假分割掩膜和对应的原始太赫兹图像组成假样本对，共同输入到判别器中进行评判；(4)分别计算生成器和判别器的损失函数，利用梯度下降算法对生成器进行优化，使生成器的损失函数尽可能小，梯度上升算法对判别器进行优化，使判别器的损失函数尽可能大，直到模型收敛，回到步骤(1)，遍历完训练集后进入步骤(5)；(5)回到步骤(1)，直到达到设定的迭代次数；(6)保存上述步骤训练好的模型，将新的待检测图片输入到生成器中，得到与之对应的检测和分割结果。