基于可变权重损失函数和难例挖掘模块的Faster R-CNN改进算法

• 作者: 施非 李金耿 邱臻 钱惠敏 韩勤 项文波
• 来源期刊: 计算机与现代化
• 学科: 工学
• 关键词: 深度学习 目标检测 焦点损失 难例挖掘
• 年，卷（期）: 2020,（8）
• 所属期刊栏目: 算法设计与分析
• 页码范围: 56-62
• 页数: 7页
• 分类号: TP391.4
• 字数: 4824字
• 语种: 中文
• DOI: 10.3969/j.issn.1006-2475.2020.08.009

摘要

基于深度卷积神经网络的目标检测算法已成为目标检测领域中的研究热点,它包括基于区域提议的两阶段目标检测算法和基于位置回归的一阶段目标检测算法.Faster R-CNN是两阶段目标检测的典型算法之一,但是,训练数据集中简单样本-难分样本数量不平衡,以及样本数据的类间不平衡,都是影响Faster R-CNN检测精度的重要原因.本文提出一种基于可变权重损失函数Focal Loss和难例挖掘模块的改进Faster R-CNN算法.具体地,在网络的分类部分引入Fo-cal Loss函数,通过权重调节样本数据的类间不平衡,改善简单样本-难分样本的数量不平衡;同时,修改网络结构,引入难例挖掘模块,进一步平衡简单样本-难分样本的数量,提高网络的检测性能.本文采用不同数据集,不同基础网络来测试提出的算法性能.实验结果表明,在VGG-16基础网络下,本文算法在Pascal VOC 2007数据集上平均检测精度较原算法提高了0.9个百分点,在Pascal VOC 07+12数据集上提高了1.7个百分点;在Res-101基础网络上,在Pascal VOC 2007数据集上平均检测精度较原算法提高了1.3个百分点,在Pascal VOC 07+12数据集上提高了1.5个百分点.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	项文波	南京理工大学自动化学院	3	8	1.0	2.0
2	钱惠敏	河海大学能源与电气学院	6	10	3.0	3.0
3	施非	河海大学能源与电气学院	1	0	0.0	0.0
4	邱臻		1	0	0.0	0.0
5	韩勤		1	0	0.0	0.0
6	李金耿		1	0	0.0	0.0