Target detection model training method, training device and electronic equipment
The invention discloses a target detection model training method and device, electronic equipment and a storage medium. The method comprises the following steps: predicting unlabeled training data through a teacher model and a student model respectively to obtain a first prediction result output by...
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 a target detection model training method and device, electronic equipment and a storage medium. The method comprises the following steps: predicting unlabeled training data through a teacher model and a student model respectively to obtain a first prediction result output by the teacher model and a second prediction result output by the student model; according to the confidence coefficient in the first prediction result, determining a target false label category to which the first prediction result belongs, different false label categories corresponding to different false label loss functions; according to the first prediction result, the second prediction result and a pseudo label loss function corresponding to the target pseudo label category, calculating the pseudo label loss of this time; and updating the student model based on the current pseudo tag loss, and updating the teacher model based on the updated student model. And repeatedly executing the steps until a preset training |
---|