Strip Tobacco Identification Based on Improved YOLOv5

LIU Yunfei, YANG Xudong, SUN Dong

Packaging Engineering ›› 2024 ›› Issue (5) : 144-150.

PDF(4886 KB)
PDF(4886 KB)
Packaging Engineering ›› 2024 ›› Issue (5) : 144-150. DOI: 10.19554/j.cnki.1001-3563.2024.05.017

Strip Tobacco Identification Based on Improved YOLOv5

  • LIU Yunfei1, YANG Xudong1, SUN Dong2
Author information +
History +

Abstract

To solve the problems of picking errors of strip tobacco by sorting machine and manual sorting in tobacco logistics center, the work aims to propose a strip tobacco identification model with faster convergence speed and higher accuracy based on YOLOv5s algorithm from the perspective of ensuring the real-time performance and recognition accuracy. Firstly, CA attention module was integrated into the YOLOv5s network architecture to better extract features and improve the accuracy of target acquisition. Then, the nearest neighbor in the original network was changed to lightweight general upper sampling operator CARAFE to obtain a larger feeling field. Next, the Ghost module was embedded in the backbone network to lightweight the network. Finally, the tobacco image acquisition system was built in the tobacco logistics center to establish the tobacco image data set. Compared with YOLOv5s, the proposed optimization algorithm was reduced by 45.8%, mAP@0.5 reached 99.3%, and the recognition rate was about 99.9% on the strip tobacco error correction system. The optimization algorithm proposed can meet the requirements of high-speed strip tobacco sorting and identification with high accuracy.

Cite this article

Download Citations
LIU Yunfei, YANG Xudong, SUN Dong. Strip Tobacco Identification Based on Improved YOLOv5[J]. Packaging Engineering. 2024(5): 144-150 https://doi.org/10.19554/j.cnki.1001-3563.2024.05.017
PDF(4886 KB)

Accesses

Citation

Detail

Sections
Recommended

/