首页 | 本学科首页   官方微博 | 高级检索  
     

融合L-α的YOLO-v4小目标检测算法
作者姓名:张宁  于鸣  任洪娥  陶锐  赵龙
作者单位:1. 东北林业大学信息与计算机工程学院;2. 黑龙江林业智能装备工程研究中心;3. 呼伦贝尔学院;4. 黑龙江东方学院
基金项目:黑龙江省自然科学基金(LH2020F040);;中央高校基本科研业务费专项基金(2572017PZ10);
摘    要:当前基于深度学习的目标检测技术得到了迅速发展,但小目标检测仍然是一个有待改善的难题。相比于大目标,小目标检测任务存在分辨率低、特征易丢失等特点,很多通用的目标检测算法不能直接迁移到小目标检测。特征金字塔融合能有效结合深层和浅层的特征,增强对小目标的检测性能,然而现有模型大都忽略了相邻层间融合时的信息不平衡问题。针对此问题,提出将有效融合因子的思想融入YOLO-v4的PANet结构,添加融合因子L-α控制深层向浅层传递的信息量,从而有效提高信息融合效率,增强YOLO-v4对小目标的检测能力。实验表明,加入了L-α的YOLO-v4模型,在Tiny Person数据集上平均精度AP50tiny和AP50small分别提高了2.14%和1.85%,在MS COCO数据集上平均精度AP和APS分别提高了1.4%和2.7%,且检测结果优于其他小目标检测算法,证明此改进方法对小目标检测有效。

关 键 词:计算机视觉  小目标检测  特征金字塔  融合因子  特征融合
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号