-
题名改进YOLOX的药品泡罩铝箔表面缺陷检测方法
被引量：8
- 1
-
-
作者
胡海涛
杜昊晨
王素琴
石敏
朱登明
-
机构
华北电力大学控制与计算机工程学院
中国科学院计算技术研究所
太仓中科信息技术研究院
-
出处
《图学学报》
CSCD
北大核心
2022年第5期803-814,共12页
-
基金
国家自然科学基金项目(61972379)
国家重点研发计划课题(2020YFB1710400)。
-
文摘
药品泡罩包装中铝箔表面包含各种字体和图案信息,而且铝箔表面凹凸不平,拍摄中会出现明暗分布不均的情况,可导致缺陷特征和铝箔表面特征相似度较高。针对YOLOX模型无法更加准确区分缺陷特征和铝箔表面特征的问题,提出一种改进YOLOX模型的表面缺陷检测方法。首先,为了使输入到Prediction网络的信息更具全局性,需要对Neck网络中特征图的全局特征进行分析,于是将Neck网络的CSP模块替换成transformer encoder模块。同时YOLOX模型具有较深的深度,为了有效地提高分类精度,使用Mish激活函数替换Swish激活函数。然后针对缺陷特征和铝箔表面特征相似导致缺陷区域和背景区域分类困难的问题,在损失函数中引入focal loss。实验结果表明,改进的模型对铝箔表面缺陷检测的mAP为90.17%,比原始的YOLOX模型提高了4.95%,并且改进的模型能够降低和铝箔表面特征相似度较高的缺陷误检和漏检的概率。
-
关键词
铝箔表面
缺陷检测
YOLOX
transformer编码器
Mish
focal损失函数
-
Keywords
aluminum foil surface
defect detection
YOLOX
transformer encoder
Mish
focal loss
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-
