公开(公告)号：CN118212417A

公开(公告)日：2024-06-18

申请号：CN202410428615.8

申请日：2024-04-10

Applicant: 重庆大学

Inventor： 毛玉星 ,  陆航 ,  柳林均 ,  朱琳 ,  陈春旭 ,  宁澜嘉

IPC: G06V10/26 ,  G06V10/774 ,  G06V10/776 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/084 ,  G16H30/20

Abstract: 本发明公开了一种基于轻量化注意力模块的医学图像分割模型及其训练方法，基于UNet网络构建分割模型，分割模型包括4～6个上下采样阶段，每个上下采用阶段包括一个卷积采样模块，下采样层采用池化操作，上采样层采用插值操作，并且每个卷积层后连接着批量归一化层；在下采样层中串联加入由大核空洞卷积组合形成的局部空间注意力模块以及通道之间交互关注的通道注意力模块；在跳跃连接和上采样层中加入卷积形式的门控注意力模块；在网络瓶颈处设置了全局注意力模块。本发明中基于卷积结构的注意力模块在局部‑全局层面增强了模型对于图像的多尺度、多方位的学习和理解，其分割的准确度更高，且鲁棒性强，训练方法更加简单，运行速率更快。