ACDet:强化自我注意力机制的药品包装轮廓检测方法

ACDet: Enhanced self-attention mechanism for pharmaceutical packaging contour detection

  • 摘要: 提出了一种基于卷积神经网络的物品矢量检测识别方法:ACDet(self-attention and concatenation based detector),旨在解决照度变化下密集无序药品包装轮廓的高效检测问题。该方法采用组合图像增强技术提升模型学习物品外观特征的能力,对计算模块C2F-A(C2F with attention)采用多条梯度流输出来进行多维度的强化自我注意力增强,包括特征维度和空间维度。设计的WConcat(weighted concatenation)模块可以对不同层次的特征图进行加权拼接并捕捉更关键的特征图,从而使网络具备更好的认知能力。在医药案例(cancer pathological and pharmaceutical dataset, CPPD)数据集实验中实现了81.0%的(mean average precision, mAP),79.5%的 \textSmooth\textmAP ,平均领先其他YOLO(you only look once)架构的模型5.5%~16.6%,在公开数据集分别达到52.2%和51.0%。同时,零样本测试中复核成功率达到99.9%。研究结果显示,ACDet能克服复杂检测场景难题,实现网络鲁棒性提升及轻量化,为工业智能化生产提供了技术支持。

     

    Abstract: This article introduces ACDet(self-attention and concatenation based detector), an object vector detection and recognition method based on convolutional neural networks. This method aims to efficiently detect dense and unordered pharmaceutical packaging contours under varying lighting conditions. By employing combined image enhancement techniques, the method enhances the model's ability to learn the appearance features of objects. The C2F-A (C2F with attention) computational module utilizes multiple gradient flows for multidimensional self-attention enhancement, encompassing both feature and spatial dimensions. The WConcat (weighted concatenation) module facilitates weighted concatenation of various levels of feature maps, capturing more critical features, thereby enhancing the network's cognitive ability. In experiments on the CPPD (cancer pathological and pharmaceutical dataset) for pharmaceutical cases, ACDet achieved 81.0% \textmAP (mean average precision) and 79.5% \textSmooth\textmAP , outperforming other YOLO(you only look once) architecture models by an average of 5.5% to 16.6%, and leading by 0.7% to 6.9% on public datasets. Additionally, zero-shot testing achieved a review success rate of 99.9%. The research results suggest that ACDet can overcome complex detection scenarios, enhance network robustness, and support intelligent industrial production.

     

/

返回文章
返回