Model training method, image recognition method, device, and storage medium

Abstract

The application provides a model training method, an image recognition method, equipment and a storage medium. The method comprises the following steps: inputting a first sample image into a first image recognition model, and outputting a first prediction classification result and a first prediction mask; inputting a target feature image into a degradation prediction network, and outputting a degradation parameter prediction result; calculating a first loss function value based on the first prediction classification result and the first prediction mask; calculating a second loss function value based on the degradation parameter and the degradation parameter prediction result; and adjusting model parameters of the first image recognition model based on the two loss functions, and training the first image recognition model. In the embodiment of the application, the network parameters are optimized by the loss functions of the prediction classification result and the prediction mask and the degradation parameter loss function, so that the image features of classification and mask are simultaneously paid attention to in the process of model training, and the image features that can be restored to the image features before degradation under different degradation degrees can be extracted, thereby improving the recognition accuracy of the model.

Images