基于改进YOLOv8的药品名称检测算法  

Drug Name Detection Algorithm Based on Improved YOLOv8

作　　者：郑志豪 赵建光 白瑞瑞 Zheng Zhihao;Zhao Jianguang;Bai Ruirui(School of Information Engineering,Hebei University of Architecture,Zhangjiakou,China)

机构地区：[1]河北建筑工程学院信息工程学院,河北张家口

出　　处：《科学技术创新》2024年第12期100-103,共4页Scientific and Technological Innovation

基　　金：张家口市2023年市级科技计划财政资助项目(项目编号:2311010A);河北省教育厅科学研究项目资助(项目编号:QN2024148);张家口市2022年基础研究和人才培养计划项目(项目编号:2221008A)。

摘　　要：针对当前药品名称检测存在检测精度低、算法参数量大和现有公开数据集在小目标检测上存在限制等问题,并基于YOLOv8算法,提出了一种结合Slim-Neck替换特征融合层、GAM注意力机制的检测精度高、硬件要求低的新模型YOLOv8-SNG。使用Drug Name Detection Dataset数据集验证算法的有效性。实验结果表明,与YOLOv8相比,在Drug Name Detection Dataset数据集上FPS少量上涨的情况下,mAP@0.5提高了11.9%,mAP@0.5:0.95提高了6.43%;改进后的算法具有更强的泛化性能,能够更好的适用于药品名称检测任务。Aiming at the problems of low detection accuracy,large number of algorithm parameters and limitations of existing public data sets in small target detection,a new model YOLOv8-SNG was proposed based on the YOLOv8 algorithm,which combined Slim-Neck replacement feature fusion layer,GAM attention mechanism with high detection accuracy and low hardware requirements.The Drug Name Detection Dataset was used to verify the validity of the algorithm.The experimental results show that,compared with YOLOv8,when FPS in the Drug Name Detection Dataset slightly increased,mAP@0.5 increased by 11.9%and mAP@0.5:0.95 increased by 6.43%.The improved algorithm has stronger generalization performance and can be better applied to the drug name detection task.

关 键 词：药品名称 目标检测 注意力机制 特征融合层 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]