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...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: SU DONGNAN, WANG TONG, WU JIPENG, WEI FENG, SHANG PENG
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
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