[发明专利]一种基于三维可变形transformer的肺结节检测方法

摘要

本发明提供了一种基于三维可变形transformer的肺结节检测方法,包括获取三维肺部CT数据集，对数据集进行预处理；构建基于三维可变形transformer的肺结节检测网络模型；裁剪预处理后的数据；以序列进行位置编码后输入到3D可变形transformer模块进行数据处理；引用3D RPN网络进行肺结节检测的分类和回归；使用验证集输入到已构建的网络模型中，输出肺结节检测结果。本发明采用三维卷积神经网络，使用3D卷积核学习结节立体特征，充分利用了CT序列的三维空间信息；针对肺结节检测这一典型的小目标检测问题，采用多尺度融合模块，充分挖掘上下文信息；采用了可变形transformer模块，提升了检测效果，其次与其他transformer结构相比，可变形transformer只关注采样点附近，提升了检测效率，且训练收敛速度更快。