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

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