Target detection network distillation method and device, electronic equipment and storage medium
The invention discloses a target detection network distillation method and device, electronic equipment and a storage medium, which are used for solving the technical problem of poor distillation effect of the existing target detection network distillation mode. The method comprises the steps of obt...
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 network distillation method and device, electronic equipment and a storage medium, which are used for solving the technical problem of poor distillation effect of the existing target detection network distillation mode. The method comprises the steps of obtaining a pre-training target detection network; pruning the pre-trained target detection network to obtain a student network; inputting a preset detection image into the pre-trained target detection network to obtain a first intermediate feature; inputting the preset detection image into the student network to obtain a second intermediate feature; calculating distillation loss according to the first intermediate characteristic and the second intermediate characteristic; and optimizing the student network according to the distillation loss to obtain a target detection network.
本发明公开了一种目标检测网络蒸馏方法、装置、电子设备及存储介质,用于解决现有的目标检测网络蒸馏方式蒸馏效果较差的技术问题。本发明包括:获取预训练目标检测网络;对所述预训练目标检测网络进行剪枝,得到学生网络;将预设检测图像输入所述预训练目标检测网络,得到第一中间特征;将所述预设检测图 |
---|