MODEL TRAINING METHOD AND APPARATUS, STORAGE MEDIUM, AND ELECTRONIC DEVICE

摘要

A model training method and apparatus, a storage medium, and an electronic device, the method comprising: acquiring a sample image set; inputting the sample image set into a deep neural network for training; if a classification image is inputted, calculating a loss value on the basis of a first loss function; and if a target detection image is inputted, calculating a loss value on the basis of the first loss function and a second loss function; and performing backpropagation on the basis of the loss value to update a network parameter until convergence, so as to obtain an image recognition model. The described method can improve the accuracy of image category prediction and target detection by a deep neural network.