[发明专利]一种基于注意力胶囊网络的口红缺陷检测装置及其检测方法

说明书

本发明涉及产品智能质检技术领域，具体为一种基于注意力胶囊网络的口红缺陷检测装置，包括图像采集系统，包括摄像机和网络客户端，所述摄像机用于拍摄待检口红4个方向的口红图像，所述网络客户端用于将拍摄的口红图像进行本地存储；口红缺陷检测系统，包括网络服务端、口红检测模块和缺陷可视化模块，其中，所述网络服务端用于接收网络客户端发送的口红图像传输至口红检测模块内进行缺陷检测，并将检测结果通过缺陷可视化模块进行展示。本发明的检测方法，用于对所述的口红图像进行缺陷检测，相较于传统工作流程，口红的质量缺陷更容易被发现，可减轻质检人员的劳动强度，从而提升工作效率和提高质检的稳定性。

技术领域

本发明涉及产品智能质检技术领域，具体为一种基于注意力胶囊网络的口红缺陷检测装置及其检测方法。

背景技术

目前口红生产企业生产线还未达到全自动化的程度，许多环节需要人工全程在线的干预，其中质检部分是产能提升的瓶颈之一，人工质检的效率存在上限；人力无法长时间持续在线，不仅对质检工人身体有所伤害，质检准确度也会随时间推移而下降；人工质检即使在工人有完备精力的状况下，仍然存在漏检、误检的情况。

经检索，专利申请号为CN201810831640.5的发明专利公开了“基于胶囊网络的绝缘子缺陷识别定位装置及方法”，其主要为智能电网中的输电线路维修技术领域，用于输电线上绝缘子的检测；

专利申请号为CN201810831640.5的发明专利公开了“基于胶囊网络的绝缘子缺陷识别定位装置及方法”，主要问题是胶囊网络的应用只用于目标检测，而实际的缺陷识别采用传统图片处理方法实现，并且没有对缺陷进行分类，功能相对单一，此外，缺陷的标记过程并非由胶囊网络实现，而且由像素比对的方法实现，鲁棒性较差。鉴于此，我们提出一种基于注意力胶囊网络的口红缺陷检测装置及其检测方法。

发明内容

本发明的目的在于提供一种基于注意力胶囊网络的口红缺陷检测装置及其检测方法，以解决上述背景技术中提出的问题。

为实现上述目的，本发明提供如下技术方案：

一种基于注意力胶囊网络的口红缺陷检测装置，包括：

图像采集系统，包括摄像机和网络客户端，所述摄像机用于拍摄待检口红4个方向的口红图像，所述网络客户端用于将拍摄的口红图像进行本地存储；

口红缺陷检测系统，包括网络服务端、口红检测模块和缺陷可视化模块，其中，所述网络服务端用于接收网络客户端发送的口红图像传输至口红检测模块内进行缺陷检测，并将检测结果通过缺陷可视化模块进行展示。

一种检测方法，用于对所述的口红图像进行缺陷检测，包括以下步骤：

步骤1：所述摄像机拍摄待检口红4个方向的口红图像，将口红图像传输至所述网络客户端内进行本地存储，并通过缺陷可视化模块创建口红图像的可视化界面；

步骤2：预先将载缺陷检测与定位模块加载入口红检测模块的主程序中，供主程序传入图像，并且预先将注意力胶囊网络模型载到口红检测模块的GPU显存中；

步骤3：口红检测模块的主程序接收到网络服务端的口红图像，并加载注意力胶囊网络模型对口红图像进行缺陷检测；

步骤4：将检测结果进行保存，并通过所述缺陷可视化模块将检测结果进行可视化展示。

优选的，步骤3中，注意力胶囊网络模型对口红图像进行缺陷检测的方法包括以下步骤：

步骤31：特征提取，输入的口红图像先经过卷积神经网络进行特征提取，获得口红图像的缺陷特征；

步骤32：引入注意力机制通过加权增加对重要缺陷特征的关注；

步骤33：将加权后的缺陷特征和原始卷积特征进行融合，并传至胶囊网络中进行动态路由；