基于改进EfficientDet的药丸检测算法
DOI:
CSTR:
作者:
作者单位:

南京信息工程大学 电子与信息工程学院 南京 210044

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

国家自然科学基金(41775165和41775039),南京信息工程大学人才启动经费(2021r034)


Pill detection algorithm based on improved EfficientDet
Author:
Affiliation:

Electronics and Information Engineering College, Nanjing University of Information Science and Technology, Nanjing 210044, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对药剂师在药丸分拣过程中因疲劳而出错的问题,本文提出了一种基于改进EfficientDet的药丸检测算法。首先,引入Mosaic数据增强技术来提高采样数据的复杂度;然后,对主干网络EfficientNet进行改进优化,嵌入了CBMA模块的特征融合层,通过增强学习特征提高对药丸关键特征的提取能力;最后,对BiFPN特征融合部分增加了下层到上层的跨级数据流,通过充分利用多级数据,提高了不同层次的多尺度特征融合效率。实验表明,改进后的EfficientDet算法在测试中mAP值达到99.84%,相比较原始EfficientDet算法提高了0.65%,同时也比YOLOv3,YOLOv4和YOLOv4-Tiny等性能较好的目标检测网络具有更高的准确率和更好的实际应用性。

    Abstract:

    Aiming at the problem that pharmacists make mistakes due to fatigue in the process of pill sorting, a pill detection algorithm based on improved EfficientDet is proposed in this paper. Firstly, mosaic data enhancement technology is introduced to improve the complexity of sampling data; Then, the backbone network EfficientNet is improved and optimized, and the feature fusion layer of CBMA module is embedded to improve the extraction ability of key features of pills by enhancing learning features; Finally, a cross level data stream from the lower layer to the upper layer is added to the feature fusion part of BiFPN. By making full use of multi-level data, the efficiency of multi-scale feature fusion at different levels is improved. Experiments show that the improved EfficientDet algorithm has a map value of 99.84% in the test, which is 0.65% higher than the original EfficientDet algorithm. At the same time, it also has higher accuracy and better practical application than the target detection networks with better performance such as YOLOv3, YOLOv4 and YOLOv4-Tiny.

    参考文献
    相似文献
    引证文献
引用本文

王敏,王康,李晟,孙硕,吴佳.基于改进EfficientDet的药丸检测算法[J].电子测量技术,2022,45(19):136-142

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-03-29
  • 出版日期:
文章二维码