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

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: ZOU PENGYU, WANG KAN, DONG PEI, HU SHUPING, TAN HUAN, PANG JIANXIN
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 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