Highly efficient industrial sorting lines require fast and reliable classification methods. Various types of sensors are used to measure the features of an object to determine which output class it belongs to. One technique involves the use of an RGB camera and a machine learning classifier. The paper is focused on protecting the sorting process against prohibited and dangerous items potentially present in the sorted material that pose a threat to the sorting process or the subsequent metallurgical process. To achieve this, a convolutional neural network classifier was applied under real-life conditions to detect forbidden elements in copper-based metal scrap. A laboratory stand simulating the working conditions in a high-speed scrap sorting line was prepared. Using this custom stand, training and test sets for machine learning were gathered and labeled. An image preprocessing algorithm was designed to increase the robustness of the resulting forbidden element detector system. The performance of multiple neural network architectures and data set augmentations was analyzed. The highest accuracy of 98.03% and F1-score of 97.16% were achieved with a DenseNet-based classifier. The results of this paper show the feasibility of using the presented solution on a high-speed industrial line.
Read full abstract