Image segmentation model training method and device, equipment and storage medium

Abstract

The invention provides an image segmentation model training method and device, equipment and a storage medium. The method comprises the following steps: training an initial image segmentation model byadopting a source domain sample to obtain a pre-trained image segmentation model; extracting a prediction segmentation result of the source domain image and a prediction segmentation result of the target domain image through the pre-trained image segmentation model; training the first discriminator by adopting the predicted segmentation result of the source domain image and the predicted segmentation result of the target domain image; training a second discriminator by adopting the predicted segmentation result of the source domain image and the standard segmentation result of the source domain image; according to the loss function of the pre-trained image segmentation model, the confrontation loss function of the first discriminator and the confrontation loss function of the second discriminator, re-training the pre-trained image segmentation model, and performing iterative loop training until convergence to obtain a trained image segmentation model, so that the segmentation result of the target domain image is more accurate.

Description

Technical field [0001] The embodiments of the present application relate to the field of image recognition technology, and in particular, to a training method, device, device, and storage medium of an image segmentation model. Background technique [0002] Image segmentation refers to classifying each pixel in the image and marking the target area. Image segmentation can be applied to medical image analysis, unmanned vehicle driving, geographic information system, underwater object detection and other fields. For example, in the field of medical image analysis, image segmentation can be used to realize tasks such as the positioning of tumors and other lesions, the measurement of tissue volume, and the study of anatomical structures. [0003] The traditional image segmentation method relies on a large number of labeled images, and the premise of this method is that the data distribution of the training image set (ie the source domain image) and the test image set (ie the target dom...

AI Technical Summary

Problems solved by technology

[0005] The embodiment of the present application provides a training method, device, equipment and storage medium of an image segmentation model, which can be used to solve the problem of inaccurate image segmentation results in related technologies

Images