一种基于DoseTransNet模型的三维放疗剂量分布预测方法

本发明提供一种基于DoseTransNet模型的三维放疗剂量分布预测方法。该方法包括：获取数据集；数据集包括头颈癌数据集和宫颈癌数据集；对数据集进行预处理；按照预设比例将预处理后的头颈癌、宫颈癌数据集划分为对应的训练集、验证集和测试集；搭建DoseTransNet模型；步骤S5：构造DoseTransNet模型的损失函数；将头颈癌数据集或宫颈癌数据集对应的训练集输入DoseTransNet模型，并基于损失函数训练DoseTransNet模型；将头颈癌数据集或宫颈癌数据集中对应的验证集输入训练后的DoseTransNet模型进行优化；将头颈癌数据集或宫颈癌数据集中对应的测试集输入最终优化后的DoseTransNet模型，得到患者的三维放疗剂量分布的预测结果。本发明实现能够精确捕捉和表达CT图像、PTV和OARs之间的复杂关系，实现高精度的剂量分布预测。