Image segmentation model training method and device, equipment and storage medium
The invention discloses an image segmentation model training method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence. The method comprises the following steps: inputting a training image into an untrained initial image segmentation model to ob...
Gespeichert in:
Hauptverfasser: | , , , , |
---|---|
Format: | Patent |
Sprache: | chi ; eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | The invention discloses an image segmentation model training method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence. The method comprises the following steps: inputting a training image into an untrained initial image segmentation model to obtain an output prediction segmentation image and prediction category data; determining an edge mapping matrix, a category loss matrix and a distance incidence matrix based on the predicted segmented image, the predicted category data, the standard segmented image and the standard category data; determining a segmentation loss function based on the edge mapping matrix, the category loss matrix and the distance incidence matrix; and based on the segmentation loss function, adjusting model parameters of the initial image segmentation model until the segmentation loss function converges, and obtaining a trained target image segmentation model. According to the embodiment, the problem that the network architecture of th |
---|