一种基于改进U-Net模型的结肠息肉分割方法

本发明公开了一种基于改进U‑Net模型的结肠息肉分割方法，包括：S1：将初始的结肠息肉数据集通过随机旋转、随机裁剪、随机亮度和对比度变化等方式进行数据增强；S2：在U‑Net模型的编码器4个层中加入残差连接，并在模型的跳跃连接部分通过通道注意力增强重要特征通道获取每个通道的全局信息，将注意力权重与原始特征图相乘，得到加权后的特征图；并在U‑Net模型的下采样部分加入FEM特征增强模块，其包含3×3与1×1的卷积层；S3：对模型进行训练完成结肠息肉分割任务。本发明利用残差连接补充初始图片语义信息，引入通道注意力获取图像全局信息，并结合特征增强凸出息肉纹理，提高了结肠息肉分割的准确率，为结肠息肉的图像分割方面提供技术支持。