基于Mask R-CNN的胶囊缺陷检测方法

作者：段仲静; 李少波^*; 胡建军; 杨静; 王铮

来源：无线电工程, 2020, 50(10): 857-862.

• 深度学习
• Mask R-CNN
• ResNet
• 胶囊
• 缺陷检测

针对传统的空心胶囊缺陷检测多用于人工抽样检测,无论是时间成本还是人力成本都较高,并且带有一定的主观性的问题,研究分析了胶囊缺陷的特点,设计并搭建了胶囊制造缺陷在线检测实验平台,设计并开源了药用空心胶囊凹陷缺陷数据集,提出了一种基于Mask R-CNN算法的胶囊缺陷检测识别方法,使用ResNet作为特征提取网络,ROIAlign对特征图ROI提取是完成像素级的对齐,能很好地解决胶囊缺陷的检测难题。经过实验验证,基于Mask R-CNN的胶囊缺陷检测识别的正确率较高。

• 单位
  贵州大学