一种基于注意力和特征金字塔的肺结核检测系统

本发明公开了一种基于注意力和特征金字塔的肺结核检测系统，包括：图像特征获取模块，用于获取肺结核CT胸部图像特征；数据预处理模块，用于对获取的肺结核CT胸部图像特征进行数据预处理；检测模型模块，用于对预处理后的肺结核CT胸部图像特征进行检测分类；其中，检测模型模块采用Faster RCNN模型进行检测分类，模型中特征提取网络模块采用ResNet50网络，ResNet50网络的残差结构嵌入SE注意力模块，并通过特征金字塔模块构建多尺度特征图。本发明在两阶段检测网络Faster RCNN的基础上添加了SE注意力模块来增强模型对PTB小病灶的检测精确度，再通过特征金字塔结构生成多尺度特征图来更好地捕捉目标的不同细节和位置，使得目标检测的定位更加精准。