基于遮挡数据模拟与特征增强的遮挡行人检索方法及系统

本发明涉及一种基于遮挡数据模拟与特征增强的遮挡行人检索方法及系统，属于计算机视觉技术领域。其包括以下步骤：获取原始行人数据集；通过遮挡模拟对原始行人数据集进行扩充；构建基于遮挡数据模拟与特征增强的遮挡行人检索模型，所述模型以图文模型CLIP‑ReID作为基准模型，包括第一阶段训练模块、第二阶段训练模块以及损失计算模块；通过损失计算模块中的损失函数对模型进行训练优化，得到训练好的模型；将待检测遮挡行人图像输入到训练好的第二阶段训练的基于ViT的图像编码器进行图像特征提取，将提取的图像特征与待检索库中的行人图像进行相似度比对，最终得到遮挡行人检索结果。本发明能够提高遮挡行人检索的准确性。