基于自适应注意力机制的YOLOv4无人超市商品检测

YOLOV4 FOR UNMANNDE SUPERMARKET COMMODITY DETECTION BASED ON ADAPTIVE ATTENTION MECHANISM

  • 摘要: 针对无人超市智能结算任务中商品检测实时性不高、堆叠商品检测效果差、相似商品难分类的问题, 提出一种基于YOLOv4改进的商品识别算法。使用轻量级网络MobileNetv2进行特征提取加快检测速度; 在MobileNetv2的倒残差结构中引入通道注意力和空间注意力放大局部特征权重, 从而加强对堆叠商品的检测能力; 在损失函数中使用焦点损失(Focal loss)解决类间差异小的难分类问题。实验结果表明, 该方法在自建商品数据集Goods上准确率达到了80.3%, 检测速度达到73帧/s, 优于YOLOv4算法。

     

    Abstract: Aimed at the problems of low real-time commodity detection, poor detection of stacked commodities, and difficulties in classifying similar commodities in the intelligent settlement task of unmanned supermarkets, a commodity recognition algorithm based on improved YOLOv4 is proposed. The algorithm used the lightweight network MobileNetv2 for feature extraction to speed up the detection speed. The channel attention and spatial attention were introduced into the inverted residual structure of MobileNetv2 to amplify the local feature weight, and thus enhancing the detection ability of stacked commodities. Focal loss was used in the loss function to solve the difficult classification problem with small inter-class difference. The experimental results show that this method achieves 80.3% accuracy and a detection speed of 73 FPS on self-built product data sets, which is better than the YOLOv4 algorithm.

     

/

返回文章
返回