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

改进YOLOv5的白细胞检测算法
引用本文:王静,孙紫雲,郭苹,张龙妹.改进YOLOv5的白细胞检测算法[J].计算机工程与应用,2022,58(4):134-142.
作者姓名:王静  孙紫雲  郭苹  张龙妹
作者单位:西安科技大学 通信与信息工程学院,西安 710054
基金项目:陕西省教育厅科学研究计划项目(19JK0528);国家自然科学基金(61701393);西安市科技计划(GXYD13.8)。
摘    要:针对白细胞数据样本少、类间差别小及目标尺寸小导致的检测精度低、效果不佳等问题,提出一种基于改进YOLOv5的白细胞检测算法YOLOv5-CHE。在主干特征提取网络的卷积层中添加坐标注意力机制,以提升算法的特征提取能力;使用四尺度特征检测,重新获取锚点框,增加浅层检测尺度,来提高小目标的识别精度;改变边框回归损失函数,以提升检验框检测的准确率。实验结果表明,对比标准的YOLOv5算法,YOLOv5-CHE算法的平均精度均值(mean average precision,mAP)、精准率和召回率分别提升了3.8个百分点、1.8个百分点和1.5个百分点,验证了该算法对白细胞检测具有很好的效果。

关 键 词:白细胞检测  YOLOv5算法  坐标注意力机制  四尺度特征检测  损失函数  

Improved Leukocyte Detection Algorithm of YOLOv5
WANG Jing,SUN Ziyun,GUO Ping,ZHANG Longmei.Improved Leukocyte Detection Algorithm of YOLOv5[J].Computer Engineering and Applications,2022,58(4):134-142.
Authors:WANG Jing  SUN Ziyun  GUO Ping  ZHANG Longmei
Affiliation:College of Communication and Information Engineering, Xi’an University of Science and Technology, Xi’an 710054, China
Abstract:Aiming at the problems of low accuracy and poor effect caused by small white blood cell data samples, small difference between classes and small target size, this paper proposes a white blood cell detection algorithm YOLOv5-CHE based on improved YOLOv5. Firstly, coordinate attention mechanism is added to the convolutional layer of the backbone feature extraction network to improve the feature extraction capability of the algorithm. Secondly, the purpose of using four-scale feature detection and reacquiring anchor point frame is to increase the detection scale of shallow layer and improve the recognition accuracy of small targets. Finally, the purpose of changing the bounding box regression loss function is to improve the accuracy of check box detection. Experimental results show that the mean average precision(mAP), precision and recall of YOLOv5-CHE are improved by 3.8 percentage points, 1.8 percentage points and 1.5 percentage points in comparison with the benchmark YOLOv5 algorithm, respectively, which shows that the proposed algorithm is effective for leukocyte detection.
Keywords:leukocyte detection  YOLOv5 algorithm  coordinate attention mechanism  four-scale feature detection  loss function
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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

京公网安备 11010802026262号