Cigarette Pack Recognition and Classification Based on Deep Learning

DAN Wei-bo, ZHU Yong-jian, HUANG Yi

Packaging Engineering ›› 2023 ›› Issue (1) : 133-140.

PDF(2148 KB)
PDF(2148 KB)
Packaging Engineering ›› 2023 ›› Issue (1) : 133-140. DOI: 10.19554/j.cnki.1001-3563.2023.01.015

Cigarette Pack Recognition and Classification Based on Deep Learning

  • DAN Wei-bo1, ZHU Yong-jian2, HUANG Yi3
Author information +
History +

Abstract

The work aims to extract the cigarette pack image data to train the deep learning target detection model, and improve the efficiency and accuracy of cigarette pack assembly lines. A cigarette pack recognition and classification model was established based on deep learning to improve the original YOLOv3 model. The designed multi-space pyramid pooling structure (M-SPP) was added to the original network. The downsampling of the 64×64 feature map was spliced with that of the 32×32 feature map. The prediction feature layer of 16×16 was removed to improve the detection accuracy and speed of the model, and the K-means++ algorithm was used to optimize the a priori frame parameters. The experiment showed that the average accuracy of the target detection model reached 99.68%, and the detection speed reached 70.82 frames per second. It is concluded that the image recognition and classification model established based on deep learning has high accuracy and fast detection speed, which can effectively meet the automatic real-time detection of cigarette pack assembly lines.

Cite this article

Download Citations
DAN Wei-bo, ZHU Yong-jian, HUANG Yi. Cigarette Pack Recognition and Classification Based on Deep Learning[J]. Packaging Engineering. 2023(1): 133-140 https://doi.org/10.19554/j.cnki.1001-3563.2023.01.015
PDF(2148 KB)

Accesses

Citation

Detail

Sections
Recommended

/