China Safety Science Journal ›› 2023, Vol. 33 ›› Issue (S1): 134-137.doi: 10.16265/j.cnki.issn1003-3033.2023.S1.5005

• Safety engineering technology • Previous Articles     Next Articles

Video detection algorithm based on small sample for foreign objects in conveyor belt

JIA Shangfeng(), WANG Zhiqiang, HAO Xinyu   

  1. Guoneng Zhunneng Group Co., Ltd., Ordos Inner Mongolia 010300, China
  • Received:2023-03-28 Revised:2023-05-11 Online:2023-06-30 Published:2023-12-31

Abstract:

Since the data sets of foreign objects in the conveyor belt collected by the video detection algorithm based on deep learning for foreign objects in the conveyor belt were fewer, this paper proposed a detection algorithm based on YOLOX for foreign objects in the conveyor belt. First, MobileNet V3 was used to replace the backbone feature extraction network CSPDarknet in YOLOX to reduce the calculation time of the backbone features, and then the convolutional network in the enhanced feature extraction network in YOLOX was replaced with a deep separable convolutional neural network to reduce the calculation time of the enhanced feature extraction network. Finally, a convolutional block attention module (CBAM) was added before the deep separable convolutional neural network in the enhanced feature extraction network to improve detection accuracy. Experiments have verified that the algorithm used in this paper can complete the training and achieve a recognition accuracy of 96.5% when using a small number of samples. At the same time, the average time of this algorithm is only 0.019 s, and the recognition frame rate can reach 40 FPS.

Key words: small sample, foreign object detection, transfer learning, YOLOX, convolutional block attention module (CBAM)