Affiliations: [a]
College of Mechanical and Electrical Engineering, Northeast Forestry University, Harbin, Heilongjiang, China
| [b]
School of Electrical Engineering, Suihua University, Suihua, Heilongjiang, China
| [c]
School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan, China
Correspondence:
[*]
Corresponding author. Liangkuan Zhu, College of Mechanical and Electrical Engineering, Northeast Forestry University, Harbin, Heilongjiang, China.E-mail: [email protected].
Abstract: Multilevel thresholding segmentation of color images plays an important role in many fields. The pivotal procedure of this technique is determining the specific threshold of the images. In this paper, an improved mayfly algorithm (IMA)-based color image segmentation method is proposed. Tent mapping initializes the female mayfly population to increase population diversity. Lévy flight is introduced in the wedding dance iterative formulation to make IMA jump from the local optimal solution quickly. Two nonlinear coefficients were designed to speed up the convergence of the algorithm. To better verify the effectiveness, eight benchmark functions are used to test the performance of IMA. The average fitness value, standard deviation, and Wilcoxon rank sum test are used as evaluation metrics. The results show that IMA outperforms the comparison algorithm in terms of search accuracy. Furthermore, Kapur entropy is used as the fitness function of IMA to determine the segmentation threshold. 10 Berkeley images are segmented. The best fitness value, peak signal-to-noise ratio (PSNR), structural similarity index (SSIM), and other indexes are used to evaluate the effect of segmented images. The results show that the IMA segmentation method improves the segmentation accuracy of color images and obtains higher quality segmented images.
