Vol.65, No.1, 2020, pp.563-578, doi:10.32604/cmc.2020.09806
OPEN ACCESS
ARTICLE
Ore Image Segmentation Method Based on U-Net and Watershed
  • Hui Li1, Chengwei Pan2, 3, Ziyi Chen1, Aziguli Wulamu2, 3, *, Alan Yang4
1 School of Automation and Electrical Engineering, University of Science and Technology Beijing, Beijing, 100083, China.
2 School of Computer and Communication Engineering, University of Science and Technology Beijing, Beijing, 100083, China.
3 Beijing Key Laboratory of Knowledge Engineering for Materials Science, Beijing, 100083, China.
4 Amphenol AssembleTech, Houston, 77070, USA.
* Corresponding Author: Aziguli Wulamu. Email: aziguli@ustb.edu.cn.
Received 19 January 2020; Accepted 04 May 2020; Issue published 23 July 2020
Abstract
Ore image segmentation is a key step in an ore grain size analysis based on image processing. The traditional segmentation methods do not deal with ore textures and shadows in ore images well Those methods often suffer from under-segmentation and over-segmentation. In this article, in order to solve the problem, an ore image segmentation method based on U-Net is proposed. We adjust the structure of U-Net to speed up the processing, and we modify the loss function to enhance the generalization of the model. After the collection of the ore image, we design the annotation standard and train the network with the annotated image. Finally, the marked watershed algorithm is used to segment the adhesion area. The experimental results show that the proposed method has the characteristics of fast speed, strong robustness and high precision. It has great practical value to the actual ore grain statistical task.
Keywords
Image segmentation, ore grain size, U-Net, watershed algorithm.
Cite This Article
Li, H., Pan, C., Chen, Z., Wulamu, A., Yang, A. (2020). Ore Image Segmentation Method Based on U-Net and Watershed. CMC-Computers, Materials & Continua, 65(1), 563–578.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.