[发明专利]一种冠脉中心线提取模型的生成方法及装置

权利要求书

1.一种冠脉中心线提取模型的生成方法，其特征在于，包括：

对冠脉样本图像进行分割处理，得到冠脉样本分割图像；

对冠脉样本分割图像进行冠脉中心线标注，得到冠脉标注图像；

获取上一次训练的冠脉中心线训练模型中的首个生成器的输出图像，将冠脉样本分割图像与输出图像输入冠脉中心线训练模型中，得到冠脉中心线预测结果图像，所述冠脉中心线训练模型包括依次相连的多个生成器；

将冠脉中心线预测结果图像与对应的冠脉标注图像输入鉴别器，依据鉴别器的判定结果进行损失计算，若损失满足预先设置的损失阈值，依据满足损失阈值的冠脉中心线训练模型，得到所述冠脉中心线提取模型；

所述获取上一次训练的冠脉中心线训练模型中的首个生成器的输出图像，将冠脉样本分割图像与输出图像输入冠脉中心线训练模中，得到冠脉中心线预测结果图像，包括：

对于第一次训练，将冠脉样本分割图像与预先设置的冠脉中心线初始结果图像输入冠脉中心线训练模型中的第一生成器，得到冠脉中心线第一结果图像；

将冠脉样本分割图像与冠脉中心线第一结果图像输入冠脉中心线训练模型中的第二生成器，得到冠脉中心线第二结果图像；直至将冠脉样本分割图像与冠脉中心线训练模型中的倒数第二个生成器输出的冠脉中心线结果图像输入最后一个生成器，得到冠脉中心线预测结果图像；

对于其它次训练，将冠脉样本分割图像与上一次训练的冠脉中心线训练模型中的首个生成器的输出图像输入冠脉中心线训练模中，得到冠脉中心线预测结果图像，在一次训练过程中，每一生成器输入的冠 脉样本分割图像均相同。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

获取满足损失阈值的冠脉中心线训练模型的首个生成器的输出图像，得到所述冠脉中心线提取模型的第一输入图像。

3.根据权利要求2所述的方法，其特征在于，所述方法还包括：

获取冠脉图像，对冠脉图像进行分割处理，得到冠脉分割图像；

将冠脉分割图像与所述第一输入图像输入冠脉中心线提取模型中，得到冠脉中心线结果图像。

4.根据权利要求3所述的方法，其特征在于，所述对冠脉图像进行分割处理，得到冠脉分割图像，包括：

对冠脉图像进行一次分割，得到主动脉区域分割图像、心脏区域分割图像以及其他区域分割图像；

获取心脏区域分割图像的最小外接立方体；

对最小外接立方体进行下采样，得到冠脉分割图像。

5.根据权利要求4所述的方法，其特征在于，所述将冠脉分割图像与所述第一输入图像输入冠脉中心线提取模型中，得到冠脉中心线结果图像，包括：

将冠脉分割图像分别输入冠脉中心线提取模型中各生成器的第二输入端，将所述第一输入图像输入冠脉中心线提取模型中首个生成器的第一输入端；

将冠脉中心线提取模型中前一个生成器的输出图像输入后一个生成器的第一输入端，得到冠脉中心线提取模型输出的冠脉中心线结果图像。

6.根据权利要求5所述的方法，其特征在于，所述方法还包括：

对冠脉中心线提取模型输出的冠脉中心线结果图像进行上采样处理，以得到与心脏尺寸一致的冠脉中心线结果图像。