AUTHOR=Hao Zhiqiang , Wang Zhigang , Bai Dongxu , Tong Xiliang TITLE=Surface Defect Segmentation Algorithm of Steel Plate Based on Geometric Median Filter Pruning JOURNAL=Frontiers in Bioengineering and Biotechnology VOLUME=10 YEAR=2022 URL=https://www.frontiersin.org/journals/bioengineering-and-biotechnology/articles/10.3389/fbioe.2022.945248 DOI=10.3389/fbioe.2022.945248 ISSN=2296-4185 ABSTRACT=

Problems such as redundancy of detection model parameters make it difficult to apply to factory embedded device applications. This paper focuses on the analysis of different existing deep learning model compression algorithms and proposes a model pruning algorithm based on geometric median filtering for structured pruning and compression of defect segmentation detection networks on the basis of structured pruning. Through experimental comparisons and optimizations, the proposed optimization algorithm can greatly reduce the network parameters and computational effort to achieve effective pruning of the defect detection algorithm for steel plate surfaces.