[发明专利]标志点检测模型生成方法及标志点检测方法

说明书

本发明提供了一种标志点检测模型生成方法及标志点检测方法。所述生成方法包括：构建含有预先标记的标志点的三维模型；获取三维模型多个不同视角的二维图像，并将标志点的三维位置数据转换为二维位置数据；利用深度学习方法，将每个视角的二维图像及标志点的二维位置数据作为输入分别训练得到多个神经网络模型；获取具标志点的第二响应图，并对所述第一响应图进行修正得到修正的第一响应图，将修正的第一响应图上标志点的位置数据作为映射层输入，通过映射关系得到标志点的预测三维位置数据；根据原始三维位置数据和预测三维位置数据计算得到损失值，响应损失值满足预设条件得到训练完成的标志点检测模型。本发明提供的方法具有性能好的优点。

【技术领域】

本发明涉及医学图像处理技术领域，尤其涉及一种标志点检测模型生成方法及标志点检测方法。

【背景技术】

在临床医学手术应用、生物科学研究以及形态学识别等领域，标志点的准确检测起着重要作用；同时标志点也是其他研究领域的基础，如配准、分割等。深度学习为特征学习提供了一个有效的工具。近年来，深度学习模型以及用于三维形状特征表示取得了很大的成果。

相关技术中利用深度学习处理三维数据主要分为两种类型。第一类是基于三维体数据进行标志点的检测。体数据是由一张一张CT(Computed Tomography，电子计算机断层扫描)切片组成，数据从三个维度长度、宽度、高度出发，训练的网络使用的是二维图像，训练一般直接对标志点进行学习，通过取image patch(图像块)来解决有限医学图像数据的问题。第二类是基于训练数据转换。Haggai等人提出的检测方法：先基于体数据建模，然后对建好的三维模型进行标记，接着将三维数据转化为二维数据进行训练，转换过程中会出现某些区域不连续的问题。第一类方法直接对原始体数据进行学习，省去了建模的部分，数据处理简单化，但是在训练阶段，由于体数据的计算量比较大，所以计算复杂度是一个很大的问题。第二类方法在操作过程中，每个数据的处理过程很耗时，而且为了处理转换过程中模型的不连续性问题，转换的得到的二维数据就会比原始数据量要大，训练时间自然也跟着增加了。除此之外有些研究对于模型也有要求，要求三维模型是球状的才能进行数据转换。

因此，实有必要提供一种改进的标志点检测模型生成方法及用于检测标志点的方法以解决上述问题。

【发明内容】

本发明将二维图像作为深度学习工具的输入进行训练得到神经网络模型，再通过修正(增强)标志点的像素值及二维至三维的映射关系得到标志点的三维预测位置数据，相比基于三维的CNN来说，在保证实际可用的前提下，性能更好，时间效率更高。

为实现上述目的，本发明提供一种标志点检测模型生成方法，所述方法包括：

构建三维模型，所述三维模型上包含有预先标记的标志点，所述标志点对应的位置数据为原始三维位置数据；

基于所述三维模型，获取多个不同视角的二维图像，并根据所述原始三维位置数据和与视角对应的视点位置数据计算得到所述标志点的二维位置数据，所述多个不同视角包括至少一个全局视角和一个局部视角，所述二维图像包括深度图像和表面参考图像；

利用深度学习方法，将每个视角的所述深度图像及所述标志点的二维位置数据作为输入，将优化得到的响应图作为输出，分别训练得到多个神经网络模型，每个不同视角的二维图像对应一个神经网络模型，其中输出的响应图包括对应全局视角的第一响应图和对应局部视角的第二响应图；

基于所述表面参考图像获取具标志点的第二响应图，将所述具标志点的第二响应图对所述第一响应图进行修正得到修正的第一响应图，将所述修正的第一响应图上标志点的位置数据作为映射层输入，基于二维到三维的映射关系得到标志点的预测三维位置数据；

根据所述原始三维位置数据和所述预测三维位置数据计算得到损失值，响应所述损失值满足预设条件得到训练完成的所述标志点检测模型。