现代计算机2022，Vol.28Issue(19) ：81-86.DOI:10.3969/j.issn.1007-1423.2022.19.014

轻量化卷积神经网络在药片检测分类中的应用

Application of Lightweight Convolution Neural Network in Tablet Detection Classification

黄开坤 徐兴
现代计算机2022，Vol.28Issue(19) ：81-86.DOI:10.3969/j.issn.1007-1423.2022.19.014
✕
  • 维普
  • 万方数据

轻量化卷积神经网络在药片检测分类中的应用

Application of Lightweight Convolution Neural Network in Tablet Detection Classification

黄开坤 1徐兴1
扫码查看

作者信息

  • 1. 南华大学机械工程学院,衡阳 421001
  • 折叠

摘要

在药片残缺检测分类任务中,为衡量分类模型的精度与模型大小,保证模型精度的同时,减少计算量和参数量.提出一种改进的GhostNet轻量化神经网络,保证在加工过程中能够有效地检测出残缺药片.构建数据集并增强处理,压缩网络模型,将注意力机制升级为ECA模块.实验结果显示,改进后的网络模型能够达到98.85％的分类精度,模型参数量仅为0.635×106.与其它轻量化网络进行了对比分析,取得了更高的分类精度,更少的参数量和计算量,更优的分类性能.

关键词

图像处理/GhostNet/注意力机制/轻量化网络

引用本文复制引用

出版年

2022
现代计算机
中大控股

现代计算机

影响因子：0.292
ISSN：1007-1423
参考文献量5
段落导航相关论文