Abstract
In this paper, a novel wear particle online images multi-label classification based on semantic segmentation (WPC-SS) is proposed. In this model, both semantic labels and class labels are applied to guide network training, which make the regions with wear particles attain more attention during the process of network training. It solves the problem that it is difficult to classify the small wear particles when they are compared with the background in the online image. In addition, chain channel attention and class attention unit are added to optimize the network to improve the recognition accuracy. The important channels of the network are monitored by chain channel attention, so that the extracted features can be better prepared for the subsequent classification work. Class attention unit can refine the segmentation results and further optimize the classification results. Comparison experiments are executed with the standard image classification method (multi-CNN). The results of experiments show that WPC-SS model surpasses the standard image classification methods in solving the problem of multi-label classification of online wear particle images.
Similar content being viewed by others
Data availability
The manuscripts have no associated data or the data will not be deposited.
References
Kumar, M., Mukherjee, P.S., Misra, N.M.: Advancement and current status of wear debris analysis for machine condition monitoring: a review. Ind. Lubr. Tribol. 65(1), 3–11 (2013)
Cao, W., Dong, G., Xie, Y.B., et al.: Prediction of wear trend of engines via on-line wear debris monitoring. Tribol. Int. 120(01), 510–519 (2018)
Wang, S., Wu, T., Yang, L., et al.: Three-dimensional reconstruction of wear particle surface based on photometric stereo. Measurement 133(10), 350–360 (2019)
Zheng, L., Huan, H., Zeng, Y., Song, S., Cheng, S., Zhang, C.: A study on the failure mechanism and wear loss of impregnated diamond bits during machining process of armor ceramics. J. Mech. Sci. Technol. 32(1), 261–268 (2018)
Stachowiak, G.P., Stachowiak, G.W., Podsiadlo, P.: Automated classification of wear particles based on their surface texture and shape features. Tribol. Int. 41(1), 34–43 (2008)
Yuan, W., Chin, K.S., Hua, M., et al.: Shape classification of wear particles by image boundary analysis using machine learning algorithms. Mech. Syst. Signal Process. 72–73(10), 346–358 (2016)
Myshkin, N.K., Kong, H., Grigoriev, A.Y., et al.: The use of color in wear debris analysis. Wear 250(2), 1218–1226 (2001)
Asgari, S., Mehrnia, A., Moussavi, M.: Automatic detection of atrial fibrillation using stationary wavelet transform and support vector machine. Comput. Biol. Med. 60(3), 132–142 (2015)
Peng, Y., Tonghai, Wu., Cao, G., et al.: A hybrid search-tree discriminant technique for multivariate wear debris classification. Wear 392–393(9), 152–158 (2017)
Wang, J., Wang, X.: A wear particle identification meth-od by combining principal component analysis and grey relational analysis. Wear 304(1–2), 96–102 (2013)
Xu, B., Wen, G., Zhang, Z., et al.: Wear particle classification using genetic programming evolved features. Lubr. Sci. 30(5), 229–246 (2018)
Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. Commun. ACM 60(6), 84–90 (2017)
Simonyan K., Zisserman A.: Very deep convolutional networks for large-scale image recognition. Computer Science, (2014). arXiv:1409.1556
Szegedy C et al, Going deeper with convolutions. In: The IEEE Conference on Computer Vision and Pattern Recognition, (2015) pp. 1–9.
He K., Zhang X., Ren S., Sun J.: Deep residual learning for image recognition. In: The IEEE Conference on Computer Vision and Pattern Recognition pp. 770–778 (2016)
Zhang X., Zhou X., Lin M., et al, ShuffleNet: an extremely efficient convolutional neural network for mobile devices. In: The IEEE Conference on Computer Vision and Pattern Recognition, pp. 6848–6856 (2018)
Wang, S., Wu, T.H., Shao, T., et al.: Integrated model of BP neural network and CNN algorithm for automatic wear debris classification. Wear 426–427(12), 1761–1770 (2019)
Peng, P., Wang, J.: Wear particle classification considering particle overlapping. Wear 422–423(01), 119–127 (2019)
Peng, P., Wang, J.: FECNN: a promising model for wear particle recognition. Wear 432–433, 202968 (2019)
Peng, Y., Cai, J., Wu, T., et al.: A hybrid convolutional neural network for intelligent wear particle classification. Tribol. Int. 138(05), 166–173 (2019)
Wang, J., Liu, X., Wu, M., Wu, M., Wang, X., et al.: Direct detection of wear conditions by classification of ferrograph images. J. Braz. Soc. Mech. Sci. Eng. 42(4–5), 857–862 (2020)
Peng, Y., Cai, J., Wu, T., Cao, G., Kwok, N., Peng, Z.: WP-DRnet: a novel wear particle detection and recognition network for automatic ferrograph image analysis. Tribol. Int 151, 106379 (2020)
Wang XY., Xie Y., Cui Y.: A real-time semantic encoder-decoder segmentation networks for autonomous driving[A]. In: Proceedings of the 4th International Symposium on Application of Materials Science and Energy Materials, Shanghai, 8 (2020)
Badrinarayanan, V., Kendall, A., Cipolla, R.: SegNet: a deep convolutional encoder-decoder architecture for image segmentation. IEEE Trans. Pattern Anal. Mach. Intell. 39(12), 2481–2495 (2017)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. arXiv:1505.04597
Lin, G., Milan, A., Shen, C, Reid, I.: RefineNet: multipath refinement networks for high-resolution semantic segmentation", CVPR (2017)
Chen, L et al.: Semantic image segmentation with deep convolutional nets and fully connected CRFs. arXiv: Computer Vision and Pattern Recognition (2014)
Chen, L et al.: DeepLab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected CRFs. arXiv: Computer Vision and Pattern Recognition (2016)
Chen, L et al.: Rethinking atrous convolution for semantic image segmentation. arXiv: Computer Vision and Pattern Recognition (2017)
Chen, L, et al.: Encoder-decoder with atrous separable convolution for semantic image segmentation. In: European conference on computer vision 833–851 (2018)
Zhao, H, et al.: Pyramid scene parsing network. In: Computer vision and pattern recognition 6230–6239 (2017)
Lin, M et al.: Network in network, (2013). arXiv:1312.4400
.Wang, J et al.: CNN-RNN: a unified framework for multi-label image classification. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2285–2294 (2016)
Gong Y., Jia Y.,Leung, T.,Toshev, A.,Ioffe, S.: Deep convolutional ranking for multilabel image annotation. CoRR, arxiv pre-print, vol.abs/1312.4894, pp.1–9, 2013.
Turpin, A., Scholer F.: User performance versus precision measures for simple search tasks. In: Proceedings of the 29th annual international ACM SIGIR conference on Research and development in information retrieval, pages 11–18. ACM, (2006)
Chen, L.C., Zhu, Y., Papandreou, G., et al.: Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation. Springer, Cham (2018)
Funding
This paper is sponsored by the National Study Abroad Fund of China, Fundamental Research Funds for the Central Universities (FRF-GF-20-16B) and supported by The National Key Research and Development Program of China (2017YFB1002304).
Author information
Authors and Affiliations
Contributions
SF and TZ performed the data analyses, the experiment and wrote the manuscript. AW contributed significantly to analysis and manuscript preparation. XG and YZ helped perform the analysis with constructive discussion.
Corresponding authors
Ethics declarations
Conflict interests
We declare that we have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Fan, S., Zhang, T., Guo, X. et al. WPC-SS: multi-label wear particle classification based on semantic segmentation. Machine Vision and Applications 33, 43 (2022). https://doi.org/10.1007/s00138-022-01287-w
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s00138-022-01287-w