Skip to main content
Log in

WPC-SS: multi-label wear particle classification based on semantic segmentation

  • Original Paper
  • Published:
Machine Vision and Applications Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6

Similar content being viewed by others

Data availability

The manuscripts have no associated data or the data will not be deposited.

References

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

  7. Myshkin, N.K., Kong, H., Grigoriev, A.Y., et al.: The use of color in wear debris analysis. Wear 250(2), 1218–1226 (2001)

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Article  Google Scholar 

  11. Xu, B., Wen, G., Zhang, Z., et al.: Wear particle classification using genetic programming evolved features. Lubr. Sci. 30(5), 229–246 (2018)

    Article  Google Scholar 

  12. Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. Commun. ACM 60(6), 84–90 (2017)

    Article  Google Scholar 

  13. Simonyan K., Zisserman A.: Very deep convolutional networks for large-scale image recognition. Computer Science, (2014). arXiv:1409.1556

  14. Szegedy C et al, Going deeper with convolutions. In: The IEEE Conference on Computer Vision and Pattern Recognition, (2015) pp. 1–9.

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

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

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

    Article  Google Scholar 

  18. Peng, P., Wang, J.: Wear particle classification considering particle overlapping. Wear 422–423(01), 119–127 (2019)

    Article  Google Scholar 

  19. Peng, P., Wang, J.: FECNN: a promising model for wear particle recognition. Wear 432–433, 202968 (2019)

    Article  Google Scholar 

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

    Article  Google Scholar 

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

    Google Scholar 

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

    Article  Google Scholar 

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

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

    Article  Google Scholar 

  25. Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. arXiv:1505.04597

  26. Lin, G., Milan, A., Shen, C, Reid, I.: RefineNet: multipath refinement networks for high-resolution semantic segmentation", CVPR (2017)

  27. Chen, L et al.: Semantic image segmentation with deep convolutional nets and fully connected CRFs. arXiv: Computer Vision and Pattern Recognition (2014)

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

  29. Chen, L et al.: Rethinking atrous convolution for semantic image segmentation. arXiv: Computer Vision and Pattern Recognition (2017)

  30. Chen, L, et al.: Encoder-decoder with atrous separable convolution for semantic image segmentation. In: European conference on computer vision 833–851 (2018)

  31. Zhao, H, et al.: Pyramid scene parsing network. In: Computer vision and pattern recognition 6230–6239 (2017)

  32. Lin, M et al.: Network in network, (2013). arXiv:1312.4400

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

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

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

  36. Chen, L.C., Zhu, Y., Papandreou, G., et al.: Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation. Springer, Cham (2018)

    Book  Google Scholar 

Download references

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

Authors

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

Correspondence to Taohong Zhang or Aziguli Wulamu.

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

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s00138-022-01287-w

Keywords

Navigation