一种基于扩散模型的胎儿脑核磁体积图像的生成方法

本发明公开了一种基于扩散模型的胎儿脑核磁体积图像的生成方法，该方法包括下述步骤：对胎儿脑三维体积的图像数据集进行脑组织提取得到真实三维胎儿脑体积和对应的图像掩码；构建扩散模型；对真实三维胎儿脑体积按时间步添加高斯噪声得到含噪体积，时间步经线性层处理嵌入得到时间步编码，基于含噪体积、时间步编码和图像掩码对扩散模型进行训练；初始化含噪体积，基于时间步进行迭代去噪，基于训练后的扩散模型输出预测噪声，计算下一时间步的含噪体积；在设定的时间步加入保真优化过程，将保真优化的结果体积替换原来时间步对应的体积；迭代去噪后输出完全去噪后的生成体积。本发明提升了三维胎儿脑核磁图像体积的重建质量。