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

实时目标检测算法YOLO的批再规范化处理
引用本文:温捷文,战荫伟,郭灿樟,凌伟林.实时目标检测算法YOLO的批再规范化处理[J].计算机应用研究,2018,35(10).
作者姓名:温捷文  战荫伟  郭灿樟  凌伟林
作者单位:广东工业大学计算机学院,广东工业大学计算机学院,广东工业大学计算机学院,广东工业大学计算机学院
摘    要:针对实时目标检测YOLO(You Look Only Once)算法中存在的检测精度低和网络模型训练速度慢等问题,结合批再规范化算法处理小批样本以及非独立同分布数据的优势,提出了在YOLO网络结构加入批再规范化处理的改进算法。该YOLO改进算法把卷积层经过卷积运算产生的特征图看作一个个神经元,然后对这些神经元进行规范化处理。同时,在网络结构中移除了Dropout,并增大了网络训练的学习率。实验结果表明,该改进算法相对于原YOLO算法具有更高的检测精度、更快的实时检测速度以及通过适当设置批样本大小可使网络模型在训练时间和硬件设备方面成本有一定的降低。

关 键 词:目标检测  深度学习  卷积神经网络  批再规范化  YOLO
收稿时间:2017/6/23 0:00:00
修稿时间:2018/9/2 0:00:00

Batch renormalize the realtime object detection algorithm YOLO
Wen Jiewen,Zhan Yinwei,Guo Canzhang and Ling Weilin.Batch renormalize the realtime object detection algorithm YOLO[J].Application Research of Computers,2018,35(10).
Authors:Wen Jiewen  Zhan Yinwei  Guo Canzhang and Ling Weilin
Affiliation:School of Computer Science, Guangdong University of Technology,,,
Abstract:Abstract: In order to overcome the shortcomings that low detection accuracy and slow network training of the realtime object detection algorithm YOLO (You Look Only Once), combining the characteristic of Batch Reormalization, which has the strength to deal with small or non-i.i.d. minibatches, this paper proposed an algorithm that add the Batch Renormalization to the YOLO network structure. The improved algorithm view the feature maps, which generated from the convolutional layers, as activations, and then batch renormalize the activations, meanwhile it remove the Dropout from the original network structure and increase learning rate. The experimental results shows that the proposal algorithm own better detection accuracy and faster than before, and furthermore, it can decrease the model training time and the requirement of hardware equipment.
Keywords:Object detection  Deep learning  Convolutional Neural Network  Batch Renormalization  YOLO
点击此处可从《计算机应用研究》浏览原始摘要信息
点击此处可从《计算机应用研究》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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