Network compression method and device, computer equipment and storage medium

The invention relates to a network compression method and device. The method comprises the following steps: marking normal equipment and defective equipment in a collected image to obtain a sample image; training the target detection network through the sample image to obtain a pre-trained target de...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: ZHAI RUICONG, LU SONG, ZHENG YAOHUA, DONG CHENGXI, TAN HAI'AO, PAN QITONG, ZHANG JIAWEI, YUAN HANJIE, LU LIN, CHEN LIANG, LI RU, YE GUANGXIAN, JIANG TIANHANG, LIAO GAOFENG, HE YONG
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 relates to a network compression method and device. The method comprises the following steps: marking normal equipment and defective equipment in a collected image to obtain a sample image; training the target detection network through the sample image to obtain a pre-trained target detection network; performing pruning processing on the pre-trained target detection network according to the importance of each network parameter in the pre-trained target detection network to obtain a pruned target detection network; performing sparse distillation processing on the pruned target detection network to obtain a distilled target detection network; fusing the network parameters of the pruned target detection network and the network parameters of the distilled target detection network to obtain a fused target detection network; and according to the importance degree of each filter in the fused target detection network, carrying out filter pruning on the fused target detection network to obtain a compress