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