A coarse-to-fine segmentation frame for polyp segmentation via deep and classification features

Abstract

Accurate polyp segmentation is of great significance for the diagnosis and treatment of colon cancer. Deep convolution network can extract the common high level features of the target. However, most network models ignore some individual features, so the sample prediction results in complex space are fuzzy and lack of regularity. In this paper, a coarse-to-fine segmentation frame for polyp segmentation via deep and classification features is proposed. Firstly, batch schatten-p norms maximization is introduced into a network model to strengthen the predict map. Then, an automatic two classification mechanism is constructed and the prediction map is classified into two categories: simple and complex samples. Since the CNN prediction maps of simple samples are close to binary images, the prediction maps are not processed. Finally, an active contour model segmentation algorithm for saliency detection of complex samples is proposed. Experiments on Kvasir-SEG, CVC-300, CVC-ClinincDB, CVC-ColonDB and ETIS-LaribPolypDB datasets using multiple models verify the effectiveness of the framework. Code is available at https://doi.org/10.24433/CO.7821162.v1.