Skip to main content
SpringerLink
Log in

Improved whale optimization algorithm for 2D-Otsu image segmentation with application in steel plate surface defects segmentation

  • Original Paper
  • Published:
Signal, Image and Video Processing Aims and scope Submit manuscript

Abstract

Since the steel plate surface defect image often has complicated background and lots of noise, the segmentation accuracy is low when using the single threshold Otsu method. Therefore, this paper introduces the whale optimization algorithm (WOA) to optimize the threshold of the dual-threshold image segmentation. To avoid the premature convergence, slow convergence speed and easy fall into the local optimum of the original WOA, an improved WOA is proposed. Firstly, the WOA is discretized by using round function; secondly, the sin mapping generation chaotic sequence is used to replace the randomly generated initial population in the initialization process of the WOA to enhance the multiformity of population; thirdly, the global search and local development capabilities are balanced and improved by nonlinear time-varying factors and inertia weights in the position updating mechanism; finally, the improved WOA is applied to the two-dimensional Otsu (2D-Otsu) algorithm to select the optimal threshold for image segmentation. The simulation results of 8 classic benchmark functions show that the improved WOA can obtain the optimal value of the function 0, − 12,569.5. The improved WOA can raise convergence speed and improve the global search ability and get rid of the local optimum. The experimental results show that the proposed algorithm outperforms the Otsu algorithm and can achieve more accurate segmentation of steel plate surface defect image. Compared with 2D-Otsu algorithm, the proposed algorithm reduces running time by 0.34 s and has the highest segmentation efficiency for rolled-in scale defects.

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

Access this article

Log in via an institution

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

Similar content being viewed by others

References

  1. Zhang, S., Zhang, Q., Gu, J., et al.: Visual inspection of steel surface defects based on domain adaptation and adaptive convolutional neural network. Mech. Syst. Signal Process. 109, 107541–107556 (2021)

    Article  Google Scholar 

  2. Fu, G., Sun, P., Zhu, W., et al.: A deep-learning-based approach for fast and robust steel surface defects classification. Opt. Lasers Eng. 121, 1–5 (2019)

    Article  Google Scholar 

  3. He, Y., Song, K., Meng, Q., et al.: An end-to-end steel surface defect detection approach via fusing multiple hierarchical features. IEEE Trans. Instrum. Meas. 69(4), 1493–1504 (2020)

    Article  Google Scholar 

  4. Huang, Z., Wu, J., Xie, F.: Automatic surface defect segmentation for hot-rolled steel strip using depth-wise separable U-shape network. Mater. Lett. 301, 130271 (2021)

    Article  Google Scholar 

  5. Niu, S., Li, B., Wang, X., et al.: Defect attention template generation cycle GAN for weakly supervised surface defect segmentation. Patt. Recognit. 123, 108396 (2022)

    Article  Google Scholar 

  6. Luo, Q., Fang, X., Liu, L., et al.: Automated visual defect detection for flat steel surface: a survey. IEEE Trans. Instrum. Meas. 69(3), 626–644 (2020)

    Article  Google Scholar 

  7. Andrés, T.G., Pierre, G., Laure, B.É.: Remote sensing image analysis by aggregation of segmentation-classification collaborative agents. Patt. Recognit. 73, 259–274 (2018)

    Article  Google Scholar 

  8. Upadhya, P., Chhabra, J.K.: Multilevel thresholding based image segmentation using new multistage hybrid optimization algorithm. J. Ambient. Intell. Humaniz. Comput. 12(1), 1081–1098 (2021)

    Article  Google Scholar 

  9. Cheng, D., Tian, F., Liu, L., et al.: Image segmentation based on multi-region multi-scale local binary fitting and Kullback-Leibler divergence. SIViP 12, 895–903 (2018)

    Article  Google Scholar 

  10. Yan, W., Wang, Y., Xia, M., et al.: Edge-guided output adaptor: highly efficient adaptation module for cross-vendor medical image segmentation. IEEE Signal Process. Lett. 26(11), 1593–1597 (2019)

    Article  Google Scholar 

  11. Kakhani, N., Mokhtarzade, M., Valadan Zouj, M.J.: Classification of very high-resolution remote sensing images by applying a new edge-based marker-controlled watershed segmentation method. SIViP 13, 1319–1327 (2019)

    Article  Google Scholar 

  12. Zhou, J., Wang, L., Chen, X.: Maximum two-dimensional entropy image segmentation based on improved whale optimization algorithm. Laser Technol. 45(03), 378–385 (2021)

    Google Scholar 

  13. Lei, X., Ouyang, H., Xu, L.: Image segmentation based on equivalent three-dimensional entropy method and artificial fish swarm optimization algorithm. Opt. Eng. 57(10), 103106 (2018)

    Article  Google Scholar 

  14. Anfal, T.H., Javad, R.: Multilevel thresholding of images with improved Otsu thresholding by black widow optimization algorithm. Multimed. Tools Appl. 80(18), 28217–28243 (2021)

    Article  Google Scholar 

  15. Bhuvan C., Bansal S., Gupta R. et al.: (2020) Computer based diagnosis of malaria in thin blood smears using thresholding based approach. In: 2020 7th International conference on signal processing and integrated networks (SPIN). pp 1132–1135.

  16. Huang, C., Li, X., Wen, Y.: An OTSU image segmentation based on fruitfly optimization algorithm. Alex. Eng. J. 60(1), 183–188 (2021)

    Article  Google Scholar 

  17. Cen, R., Li, J.: Application of improved PSO-2D Otsu in continuous casting slab defect image segmentation. China Test 4, 19–24 (2020)

    Google Scholar 

  18. Liu, J., Li, W.: Two dimensional Otsu automatic threshold segmentation method for gray image. J. Automat. 19(01), 101–105 (1993)

    Google Scholar 

  19. Wu, Y., Fan, J., Wu, S.: Improved fast iterative algorithm for threshold segmentation of two-dimensional Otsu method. J. Electron. Measure. Instrum. 25(03), 218–225 (2011)

    Article  Google Scholar 

  20. Sha, C., Hou, J., Cui, H.: A robust 2D Otsu’s thresholding method in image segmentation. J. Vis. Commun. Image Represent. 107, 339–351 (2016)

    Article  Google Scholar 

  21. Maruthi Nagendra Prasad, J., Vamsi, Krishna M.: Lung cancer segmentation in CT images using Fuzzy-C means clustering and artificial bee colony algorithm (Article). Int. J. Innovat. Technol. Explor. Eng. 8(10), 1737–1739 (2019)

    Article  Google Scholar 

  22. Jiang, Y., Ma, Y., Liang, Y., et al.: Otsu lung tissue segmentation algorithm based on fractional sparrow search optimization. Comput. Sci. 48(S1), 28–32 (2021)

    Google Scholar 

  23. Mirjalili, S., Lewis, A.: The whale optimization algorithm. Adv. Eng. Softw. 95(5), 51–67 (2016)

    Article  Google Scholar 

  24. El Aziz, E., Eweesc, A., Hassanien, A.: Whale optimization algorithm and moth-flame optimization for multilevel thresholding image segmentation (Article). Expert Systems with Applications. 83(C), 242–256 (2017)

    Article  Google Scholar 

  25. Hassan, G., Hassanien, A.: Retinal fundus vasculature multilevel segmentation using whale optimization algorithm. SIViP 12, 263–270 (2018)

    Article  Google Scholar 

  26. Zhang, D., Yang, Y., Chu, M., et al.: Image segmentation of steel surface defects based on 2-dimensional whale optimization weighted WGG-Otsu algorithm. J. Anhui Univ. Natl. Sci. Ed. 44(3), 6 (2020)

    Google Scholar 

  27. Liu Z.: Research on complex network community discovery under intelligent algorithm. Chongqing University. 45–57 (2017).

  28. Zou, K.H., Warfield, S.K., Bharatha, A., et al.: Statistical validation of image segmentation quality based on a spatial overlap index1: scientific reports. Acad Radiol 11(2), 178–189 (2004)

    Article  Google Scholar 

Download references

Acknowledgements

This work was supported in part by the Changsha University of Science and Technology Innovation Project under Grant SJCX202064, the Natural Science Foundation of Hunan Province under Grant 2021JJ30740 and Grant 2021JJ30732, the National Natural Science Foundation of China under Grant 62073342, and the Hunan Province 2011 Collaborative Innovation Center of Clean Energy and Smart Grid. The authors, hereby, gratefully appreciate their support.

Author information

Authors and Affiliations

  1. School of Electrical & Information Engineering, Changsha University of Science and Technology, Changsha, 410114, China

    Qiyue Xie, Lin Ma, Zhisheng Chen & Wanneng Wu

  2. School of Energy & Power Engineering, Changsha University of Science and Technology, Changsha, 410114, China

    Wenqian Zhou

  3. School of Automation, Central South University, Changsha, 410083, China

    Xiaoli Wang

Authors
  1. Qiyue Xie
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Wenqian Zhou
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Lin Ma
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Zhisheng Chen
    View author publications

    You can also search for this author in PubMed Google Scholar

  5. Wanneng Wu
    View author publications

    You can also search for this author in PubMed Google Scholar

  6. Xiaoli Wang
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Qiyue Xie.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Xie, Q., Zhou, W., Ma, L. et al. Improved whale optimization algorithm for 2D-Otsu image segmentation with application in steel plate surface defects segmentation. SIViP 17, 1653–1659 (2023). https://doi.org/10.1007/s11760-022-02375-0

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11760-022-02375-0

Keywords

Search

Navigation