Material State Identification of the Conveyor Belt Based on Inception-v4

Jinliang Tang,Zhenyu Cai,Chao Xu,Tehuan Chen
DOI: https://doi.org/10.1109/cac53003.2021.9728211
2021-01-01
Abstract:For the uneven phenomenon of materials during material transportation on the factory assembly line, this paper proposes an improved model based on Inception-v4 to realize the identification and classification of three material states (more materials, normal materials, and fewer materials). In the image processing stage, the Farneback algorithm is used to capture the moving area, which makes the training of the model focus on effective information. In addition, considering the color bias of materials, the RGB enhancement strategy is proposed. By processing the image data in RGB channels, different weights are given to realize the enhancement of image features. In the network training and testing stage, we use Inception-v4 as the baseline model. The output of the model is modified for the classification task, and a new loss function is designed. In addition, considering the limited sample size, we pre-trained the model on ImageNet. Through experimental comparison, the improved algorithm proposed in this paper has better performance.
What problem does this paper attempt to address?