The Detection of Impurity Content in Machine-Picked Seed Cotton Based on Image Processing and Improved YOLO V4

The detection of cotton impurity rates can reflect the cleaning effect of cotton impurity removal equipment, which plays a vital role in improving cotton quality and economic benefits. Therefore, several studies are being carried out to improve detection accuracy. Image processing technology is increasingly used in cotton impurity detection, in which deep learning technology based on convolution neural networks has shown excellent results in image classification, segmentation, target detection, etc. However, most of these applications focus on detecting foreign fibers in lint, which is of little significance to the parameter adjustment of cotton impurity removal equipment. For this reason, our goal was to develop an impurity detection system for seed cotton. In image segmentation, we propose a multi-channel fusion segmentation algorithm to segment the machine-picked seed cotton image. We collected 1017 images of machine-picked seed cotton as a dataset to train the detection model and tested and recognized 100 groups of samples, with an average recognition rate of 94.1%. Finally, the image segmented by the multi-channel fusion algorithm is input into the improved YOLOv4 network model for classification and recognition, and the established V–W model calculates the content of all kinds of impurities. The experimental results show that the impurity content in machine-picked cotton can be obtained effectively, and the detection accuracy of the impurity rate can increase by 5.6%.