科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对正负例锚点框不均衡将降低基于锚点框的单阶段目标检测算法的检测精度的问题,提出一种包含锚点框提升模块和特征对齐模块来均衡正负例锚点框的算法.首先在锚点框提升模块中预测各个锚点框为正例的可能性,并粗略调整初始锚点框的位置和尺寸;然后在特征对齐模块中为调整后的锚点框提取预测所需的对齐特征;最后检测网络借助锚点框提升模块输出信息,从调整后的锚点框中识别出简单负例锚点框,并在训练阶段忽略其梯度.将文中算法应用于以VGG-16和Res Net-101为特征提取网络的编解码架构中,在目标检测数据集MSCOCO和PASCALVOC上进行实验,结果表明,该算法能够显著改善不均衡问题,提高单阶段目标检测算法的检测精度(MSCOCO和PASCALVOC上的精度分别为42.8%和82.7%),并维持28.6帧/s的实时运行速度.
- 单位
