基于重加权Anchor的车辆检测方法

作者：周智恒; 黄宇; 黄昌正; 陈曦

来源：2019-03-28, 中国, ZL201910243460.X.

本发明公开了一种基于重加权Anchor的车辆检测方法,包括如下步骤：构造单次检测器SSD模型,基础网络使用预训练的VGG16分类网络；根据标签与先验信息计算车辆可见部分的中心点,并定义可见部分的高和宽；在损失函数中给每一正样本重新赋予权重；输入图像数据集,对构建的SSD模型进行训练；采用已训练的SSD模型进行车辆检测。本发明通过给定标签计算可见部分的中心并给每一个Anchor加权,给匹配的特征赋予高权重而不匹配的特征赋予低权重,从而提高了准确率。并且仅仅给损失函数添加权重,保持了SSD原有的速度,使得检测器能够以实时的速度运行并保持高速度。

• 单位
  淮北幻境智能科技有限公司; 广州幻境科技有限公司; 华南理工大学