首页 | 官方网站   微博 | 高级检索  
     

基于改进损失函数的YOLOv3网络
引用本文:吕铄,蔡烜,冯瑞.基于改进损失函数的YOLOv3网络[J].计算机系统应用,2019,28(2):1-7.
作者姓名:吕铄  蔡烜  冯瑞
作者单位:复旦大学计算机科学技术学院,上海 201203;上海视频技术与系统工程研究中心,上海 201203;复旦大学智能信息处理实验室,上海 201203;公安部第三研究所物联网技术研发中心,上海,201204
基金项目:国家重点研发计划(2017YFC0803700);上海市科委项目(17511101702);复旦大学工程与应用技术研究院先导项目(gyy2917-003)
摘    要:为了提高卷积神经网络在目标检测的精度,本文提出了一种基于改进损失函数的YOLOv3网络.该网络模型应用一种新的损失函数Tan-Squared Error (TSE),将原有的平方和损失(Sum Squared Error,SSE)函数进行转化,能更好地计算连续变量的损失;TSE能有效减低Sigmoid函数梯度消失的影响,使模型收敛更加快速.在VOC数据集上的实验结果表明,与原网络模型的表现相比,利用TSE有效提高了检测精度,且收敛更加快速.

关 键 词:深度学习  损失函数  目标检测  卷积神经网络
收稿时间:2018/8/12 0:00:00
修稿时间:2018/9/5 0:00:00

YOLOv3 Network Based on Improved Loss Function
LYU Shuo,CAI Xuan and FENG Rui.YOLOv3 Network Based on Improved Loss Function[J].Computer Systems& Applications,2019,28(2):1-7.
Authors:LYU Shuo  CAI Xuan and FENG Rui
Affiliation:School of Computer Science, Fudan University, Shanghai 201203, China;Shanghai Engineering Research Center for Video Technology and System, Shanghai 201203, China;Laboratory of Intelligent Information Processing, Fudan University, Shanghai 201203, China,R & D Center of Internet of Things, The Third Research Institute of Ministry of Public Security, Shanghai 201204, China and School of Computer Science, Fudan University, Shanghai 201203, China;Shanghai Engineering Research Center for Video Technology and System, Shanghai 201203, China;Laboratory of Intelligent Information Processing, Fudan University, Shanghai 201203, China
Abstract:To improve the object detect precision of Convolutional Neural Network (CNN), we present a YOLOv3 network which based on improved loss function. This network model uses a new loss function Tan-Squared Error (TSE) which transferred from primary Sum Squared Error(SSE), and works better on continuous variable error computing. Meanwhile, the properties of TSE could decrease the impact of vanishing gradient problem in sigmoid function, and speed up model converging. The experiment results in Pascal VOC dataset show that TSE improves the detect precision effectively compared with the performance of primary network model, and the convergence is accelerated.
Keywords:deep learning  loss function  object detection  Convolutional Neural Network (CNN)
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机系统应用》浏览原始摘要信息
点击此处可从《计算机系统应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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

京公网安备 11010802026262号