摘要
为了实现养殖场环境下无接触、高精度的奶牛个体有效识别,针对SSD(single shot multibox detector)算法识别准确率不高的问题,提出一种基于浅层特征模块的改进SSD(shallow feature module SSD,SFM-SSD)算法。将原始SSD算法的主干网络由VGG16替换为MobileNetV2,以降低网络的运算量,改善检测的实时性;针对SSD网络结构的浅层特征图设计浅层特征模块,扩大浅层特征图的感受视野,提高浅层特征图对目标物体的特征提取能力;利用K均值聚类算法重构区域候选框,提高算法的检测精度。实验结果表明:在奶牛个体识别任务中,SFM-SSD算法的平均准确率比原始的SSD算法提升3.13个百分点。同时检测的实时性也得到改善。
To implement the non-contact and high-precision identification of dairy cows in farm environment, an improved SSD(shallow feature module SSD, SFM-SSD)algorithm based on shallow feature module is proposed to solve the problem that SSD(single shot multibox detector)algorithm is not accurate for target recognition. Firstly, the backbone network of the original SSD algorithm is replaced by VGG16 with MobileNetV2. Secondly, the shallow feature module is designed for the shallow feature map of SSD network structure to expand the perception field of shallow feature map and improve the feature extraction ability of shallow feature map for target object. Finally, the K-means clustering algorithm is used to reconstruct the region candidate frame to improve the detection accuracy of the algorithm. Experimental results show that the average accuracy of SFM-SSD algorithm is 3.13 percentage points higher than that of traditional SSD algorithm. Meanwhile, the real-time performance of detection is also improved.
作者
邢永鑫
孙游东
王天一
XING Yongxin;SUN Youdong;WANG Tianyi(College of Big Data and Information Engineering,Guizhou University,Guiyang 550025,China)
出处
《计算机工程与应用》
CSCD
北大核心
2022年第2期208-214,共7页
Computer Engineering and Applications
基金
贵州省科技支撑计划(SY[2017]2881)。
关键词
深度学习
目标检测
反残差网络
深度可分离卷积
特征增强模块
deep learning
target detection
anti residual network
deep separable convolution
feature enhancement module
作者简介
邢永鑫(1993-),男,硕士研究生,主要研究方向为深度学习、目标检测;孙游东(1995-),男,硕士研究生,主要研究方向为人工智能、量子通信;通信作者:王天一(1989-),男,博士,副教授,主要研究方向为量子通信、图像处理、计算机视觉,E-mail:tywang@gzu.edu.cn。
