基于循环对抗生成网络的行人图像遮挡检测方法

摘要

本发明公开了一种基于循环对抗生成网络的行人图像遮挡检测方法，步骤：(1)在无遮挡的行人图像中随机加入遮挡图像块，生成带遮挡的行人图像及对应的带遮挡标注的行人图像；(2)将上述带遮挡的行人图像及对应的带遮挡标注的行人图像作为两个数据域，训练循环对抗生成网络模型，构建遮挡图像与遮挡标注之间的映射；(3)对于当前输入的待标注图像，利用训练后的循环对抗生成网络模型去产生该图像对应的遮挡标注，同时对待标注图像进行分割，得到若干个超像素块，根据每个超像素块内遮挡标注的信息，采用区域生长方法得到最终的遮挡区域。本发明方法可以在缺少遮挡标注的情况下，针对不同类型的遮挡进行建模，得到像素级别的遮挡区域检测结果。