DOI: 10.17587/prin.11.330-334

Features of the Development of a Neural Network to Automate the Recognition of Steel Defects

R. A. Karelova, riya2003@mail.ru, E. E. Ignatov, levia4119@gmail.com, Nizhny Tagil Technological Institute (branch) of the Ural Federal University named after the first President of Russia B. N. Yeltsin, Nizhny Tagil, 622031, Russian Federation

Corresponding author: Karelova Riya A., PhD, Associate Professor, Nizhny Tagil Technological Institute (branch) of the Ural Federal University named after the first President of Russia B. N. Yeltsin, Nizhny Tagil, 622031, Russian Federation, E-mail: riya2003@mail.ru

The article presents an embodiment of an artificial neural network for recognizing defects in images of steel sheets. Several stages of solving the problem are described: the choice of a development environment, a programming language, and libraries necessary for the implementation; features of data analysis, graphing, histograms, finding dependencies; the selection of a suitable neural network, the choice of neural network architecture, the selection of an algorithm for assessing quality and accuracy; neural network spelling; training and checking accuracy and quality, checking for overfitting (retraining). As development tools, Python language, PyTorch library, Jupyter development environment, convolutional neural network architecture — Unet are proposed. Features of the analysis of input images of steel sheets, features of the implementation of the neural network itself are described. The function of binary cross entropy was chosen as a criterion for assessing accuracy, since it seeks to bring the distribution of the network forecast to the target, fine not only for erroneous predictions, but also for uncertain ones. For additional evaluation, the DICE method was also used. The accuracy of the resulting model is 84 %. The proposed solution can become part of a hardware-software system for automating the recognition of defects on metal sheets.

Keywords: production digitalization, neural network, steel defects, image recognition, production automation, python, pytorch, unet architecture

pp. 330–334

For citation:
Karelova R. A., Ignatov E. E. Features of the Development of a Neural Network to Automate the Recognition of Steel Defects, Programmnaya Ingeneria, 2020, vol. 11, no. 6, pp. 330—334