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