摘要
本发明提出一种考虑方向入口线和辅助标记点的停车位检测方法,旨在通过图注意力机制将方向入口线和辅助标记点相结合以提高停车位检测的鲁棒性和简易性。首先选择合适的数据集制作训练样本和测试样本;其次,将训练样本输入到卷积网络中提取深度特征,将深度特征输入到方向入口线检测器和辅助标记点检测器,综合考虑两者的结果,获取每个标记点的坐标以及相对坐标特征,经过双线性插值、特征融合获取每个标记点的融合特征;然后通过图注意力机制获取包含所有标记点信息的注意力加权特征;最后将重要标记点的注意力加权特征拼接输入到停车位鉴别网络,判断其能否形成合理的停车位。本发明可以使模型快速、鲁棒地检测出各种类型的停车位。