一种基于多任务学习的肺叶分割模型训练及肺叶分割方法
申请号:CN202410825110
申请日期:2024-06-25
公开号:CN118823036A
公开日期:2024-10-22
类型:发明专利
摘要
本申请提供了一种基于多任务学习的肺叶分割模型训练及肺叶分割方法,所述肺叶分割模型由编码器和解码器基于U型结构搭建而成;所述训练方法,包括:设置所述肺叶分割模型的批处理大小、最大训练次数、优化算法、以及所述编码器对各肺部结构的学习率;将不同肺部结构对应的训练集交替输入至所述肺叶分割模型,调用该肺部结构的解码器执行该肺部结构的分割任务,并对该肺部结构的解码器的权重进行更新;根据所述编码器对该肺部结构的学习率对所述编码器的权重进行更新;基于更新权重后的该肺部结构的解码器和所述编码器,得到最优的所述肺叶分割模型;本申请的肺叶分割模型训练方法减轻了数据标注压力,提高了肺叶分割的精度。
技术关键词
编码器
分割模型训练方法
血管分割
肺叶分割方法
分割肺部
训练集
解码器执行
多任务
分割算法
网络
U型结构
区域生长算法
数据
图像缩放
参数
层间距