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

The invention provides an image retrieval model training method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence, in particular to a computer vision and deep learning technology. The method comprises the following steps: acquiring a training i...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: WU BINGHONG, SUN QINPEI, YANG DALU, ZHOU WENSHUO, HUANG HAIFENG, YANG YEHUI
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 provides an image retrieval model training method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence, in particular to a computer vision and deep learning technology. The method comprises the following steps: acquiring a training image set comprising at least one training image pair, wherein the training image pair comprises a first image and a second image; for each training image pair in the training image pairs, respectively extracting a first feature of the first image and a second feature of the second image by using a first encoder of the initial image retrieval model, and respectively extracting a third feature of the first image and a fourth feature of the second image by using a second encoder of the initial image retrieval model; and updating parameters of the initial image retrieval model based on a first contrast loss of the first feature and the third feature and a second contrast loss of the second feature and the fourth featur