基于AE-Tiny YOLOV3的小目标检测模型

• 作者: 林莉 姜麟 张志坚
• 来源期刊: 软件导刊
• 学科: 工学
• 关键词: 小目标检测 Tiny YOLOV3 注意力机制 多尺度检测 绝缘子状态检测
• 年，卷（期）: 2022,（3）
• 所属期刊栏目: 人工智能
• 页码范围: 55-61
• 页数: 7页
• 分类号: TP301
• 语种: 中文
• DOI: 10.11907/rjdk.212252

摘要

小目标检测是现阶段目标检测领域的热点和难点问题.针对小目标检测漏检及对硬件性能要求较高的问题,对Tiny YOLOV3进行改进,提出一种适合在低性能平台上使用的小目标检测算法AE-Tiny YOLOV3.首先,使用EfficientNet-B0骨干网络替换原算法的特征提取网络;其次,在检测网络中增加一个检测分支,形成3尺度预测;最后,引入注意力机制对3个检测分支进行改进.实验结果表明,在VOC07+12数据集上,AE-Tiny YOLOV3算法满足实时检测的要求,并且鲁棒性较高,最高能将mAP值提高16.89％.将AE-Tiny YOLOV3算法应用在架空输电线路中绝缘子状态检测实例上,mAP达到了86.53％,相较于Tiny YOLOV3算法提升了15.27％,能满足对小目标绝缘子状态的实时检测.