A Lightweight CNN Model for Fast Classification of Wear Particle Images
Aiming at the disadvantages of CNN model for wear particle analysis,such as too many parameters,slow oper-ation speed,and difficulty in practical application,research on lightweight of the CNN model for wear particle image classi-fication was carried out.The parameters,computation and pruning sensitivity of each layer in the CNN model were ana-lyzed,which determined the convolution layer 4 and 5 as the filter pruning targets.The importance of all filters in convolu-tion layer 4 and 5 was calculated,and the filters were sorted by their importance.A lightweight model was obtained after re-moving the filters with low importance at 75％pruning rate and retraining.The experimental results show that after light-weight processing,the number of theoretical parameters and amount of memory usage are reduced by more than 50％,and the operation speed is increased by more than 20％while the accuracy barely drops.This study provides an idea for the ap-plication of CNN model on the portable and mobile ferrography equipment.
万方数据
维普
