摘要
本发明公开了一种基于Mamba的轻量级眼底血管图像分割模型,包括以下步骤:编码器和解码器包括至少四层SlimPVM Block;输入特征图经过最大池化操作逐步下采样;CCA Bottleneck模块通过在特征图的高度和宽度维度上计算跨通道的自注意力;编码器和解码器通过选择性跳跃连接进行特征融合,选择性跳跃连接通过通道注意力机制对特征进行加权选择;解码器利用SlimPVM Block对上采样后的特征进行精炼,模型通过卷积转置操作恢复丢失的空间维度,并结合来自选择性跳跃连接的编码特征;解码器最后输出通过一个1x1卷积层生成最终的分割掩码,分割掩码能够精准描绘医学图像中的目标区域边界。专用于视网膜血管分割任务,减少模型参数,降低计算开销,同时保持甚至提升分割性能。