四川大学学报（自然科学版）2024，Vol.61Issue(1)：119-128,10.DOI:10.19907/j.0490-6756.2024.013004
基于U-Net改进的内窥镜息肉图像分割算法
An improved endoscopic polyp image segmentation algorithm based on U-Net
摘要
Abstract
The segmentation of polyp images has extensive research and application value in the fields of clinical treatment and computer-aided diagnostic technology,but accurate polyp segmentation is still a challenge in terms of current research and application needs.In order to solve the problems that affect the segmentation quality of endoscopic polyp images,such as the unclear boundary between polyps and mucous membranes,and the large difference in the size and shape of polyps,this paper proposed an im-proved U-Net polyp segmentation algorithm.Firstly,the boundary feature enhancement module was in-troduced on the U-Net architecture.Considering the key clues of polyp boundary and internal area,this module used the high-level features of the encoder to generate additional boundary supplementary infor-mation,which is fused at the decoder stage to improve the ability of the model to process boundary fea-tures.Secondly,the decoder of the model adopts the method of gradually fusing features from the top to the bottom.After the output features of the encoder stage are passed through local emphasis module,the boundary features are gradually fused.This multi-scale feature fusion method effectively reduces the semantic gap between the encoder and the decoder.Finally,test-time augmentation was used in the post-processing stage to further refine the segmentation results.The model has been compared and abla-ted on five public datasets:CVC-300,CVC-ClinicDB,Kvasir-SEG,CVC-ColonDB and ETIS-LibPolyp-DB.The experimental results prove the effectiveness of the modified method,and it shows better seg-mentation performance and stronger stability in the endoscopic polyp image,which provides a new refer-ence for the processing and analysis of the polyp image.关键词
内窥镜息肉图像/息肉分割/U-Net/边界加强Key words
Endoscopic polyp image/Polyp segmentation/U-Net/Boundary strengthening分类
信息技术与安全科学引用本文复制引用
邓晓青,李征,王雁林..基于U-Net改进的内窥镜息肉图像分割算法[J].四川大学学报（自然科学版）,2024,61(1):119-128,10.基金项目
国家重点研发计划项目(2020YFA0714003) （2020YFA0714003）
国家重大项目(GJXM92579) （GJXM92579）
四川省科技厅重点研发项目(2021YFQ0059) （2021YFQ0059）
