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

权利要求书

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

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

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

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

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

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

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

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

3.根据权利要求2所述的检测方法，其特征在于：步骤3中，注意力胶囊网络模型对口红图像进行缺陷检测的方法包括以下步骤：

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

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

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

步骤34：针对融合后不同的卷积特征图构建胶囊节点，经过全连接层后输出口红的缺陷类型。

4.根据权利要求2所述的检测方法，其特征在于：步骤4中，检测结果包括口红图像的缺陷类型及其置信度值。

5.根据权利要求4所述的检测方法，其特征在于：所述缺陷类型包括且不限于刮膏、粘皮、子弹体高度、孔洞。