Fully convolution neural network combined with K-means clustering algorithm for image segmentation

Bing He; FengXiang Qiao; Weijun Chen; Ying Wen

doi:10.1117/12.2502814

9 August 2018 Fully convolution neural network combined with K-means clustering algorithm for image segmentation

Bing He, FengXiang Qiao, Weijun Chen, Ying Wen

Proceedings Volume 10806, Tenth International Conference on Digital Image Processing (ICDIP 2018); 108062O (2018) https://doi.org/10.1117/12.2502814
Event: Tenth International Conference on Digital Image Processing (ICDIP 2018), 2018, Shanghai, China

Abstract

Image segmentation is an important part of many computer vision tasks such as image recognition and image understanding. Traditional image segmentation algorithms are susceptible to the influence of complex backgrounds such as illumination, shading and occlusion, thus the application of convolution neural network to image segmentation becomes a hot spot of current research. But in the process of image convolution, as the convolution goes further, the image will lose some edge information, resulting in the blurring of the final partition edge. To overcome this problem, we propose an image segmentation algorithm combining the fully convolution neural network and K-means clustering algorithm. By conducting pixel matching between the coarse segmentation result obtained by using the convolution neural network and the segmentation results obtained by using K-means, the algorithm enhances the classification of pixels on the edge to improve segmentation accuracy. The proposed algorithm adopts two-stage training method to train and optimize the model. The experimental results on VOC2012 set validate the effectiveness of the proposed method.

Citation Download Citation

Bing He, FengXiang Qiao, Weijun Chen, and Ying Wen "Fully convolution neural network combined with K-means clustering algorithm for image segmentation", Proc. SPIE 10806, Tenth International Conference on Digital Image Processing (ICDIP 2018), 108062O (9 August 2018); https://doi.org/10.1117/12.2502814

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available