[发明专利]一种骨抑制图像的生成方法、装置、存储介质及电子设备

权利要求书

1.一种骨抑制图像的生成方法，其特征在于，包括：

获取目标对象的胸部原始图像；

基于预先训练的骨抑制模型对所述胸部原始图像进行处理，得到所述目标对象的初始骨抑制图像和骨分割图像，其中，所述预先训练的骨抑制模型中包括骨抑制分支和骨分割分支；

根据所述骨分割图像中骨骼位置、所述胸部原始图像和所述初始骨抑制图像生成目标骨抑制图像。

2.根据权利要求1所述的方法，其特征在于，所述骨分割图像中包括各像素点的分类概率，其中，根据所述骨分割图像中骨骼位置、所述胸部原始图像和所述初始骨抑制图像生成目标骨抑制图像，包括：

根据所述骨分割图像中各像素点的分类概率确定骨骼位置和非骨骼位置；

提取所述胸部原始图像中所述非骨骼位置对应的第一像素点；

提取所述初始骨抑制图像中所述骨骼位置对应的第二像素点；

将所述非骨骼位置对应的第一像素点与所述骨骼位置对应的第二像素点进行组合，生成所述目标骨抑制图像。

3.根据权利要求1所述的方法，其特征在于，所述骨分割图像中包括各像素点的分类概率，其中，根据所述骨分割图像中骨骼位置、所述胸部原始图像和所述初始骨抑制图像生成目标骨抑制图像，包括：

根据所述骨分割图像中各像素点的分类概率确定像素点的所述胸部原始图像的权重和所述初始骨抑制图像的权重；

根据所述胸部原始图像中第一像素值、所述初始骨抑制图像中的第二像素值、所述胸部原始图像的权重和所述初始骨抑制图像的权重进行加权计算，得到所述目标骨抑制图像。

4.根据权利要求1所述的方法，其特征在于，所述骨抑制模型包括特征提取模块、所述骨抑制分支和所述骨分割分支，其中，所述骨抑制分支和所述骨分割分支分别与所述特征提取模块连接，所述特征提取模块中包括编码器和解码器，所述编码器和所述解码器中的卷积块横向连接。

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

创建初始骨抑制模型；

将样本数据输入至所述初始骨抑制模型，得到预测骨抑制图像和预测骨分割图像；

基于所述预测骨抑制图像和标准骨抑制图像对所述初始骨抑制模型进行训练，以及，基于所述预测骨分割图像和标准骨分割图像对所述初始骨抑制模型进行训练。

6.根据权利要求5所述的方法，其特征在于，所述基于所述预测骨抑制图像和标准骨抑制图像对所述初始骨抑制模型进行训练，包括：

根据所述预测骨抑制图像和标准骨抑制图像生成第一损失函数；

根据所述骨分割图像生成掩码矩阵；

根据所述掩码矩阵对所述第一损失函数进行处理，生成第二损失函数；

将所述第二损失函数反向输入至所述骨抑制分支，对所述初始骨抑制模型进行训练。

7.根据权利要求5所述的方法，其特征在于，所述基于所述预测骨分割图像和标准骨分割图像对所述初始骨抑制模型进行训练，包括：

根据所述预测骨分割图像和所述标准骨分割图像生成第三损失函数；

将所述第三损失函数反向输入至所述骨分割分支，对所述初始骨抑制模型进行训练。

8.一种骨抑制图像的生成装置，其特征在于，包括：

原始图像获取模块，用于获取目标对象的胸部原始图像；

图像处理模块，用于基于预先训练的骨抑制模型对所述胸部原始图像进行处理，得到所述目标对象的初始骨抑制图像和骨分割图像，其中，所述预先训练的骨抑制模型中包括骨抑制分支和骨分割分支；

目标骨抑制图像确定模块，用于根据所述骨分割图像中骨骼位置、所述胸部原始图像和所述初始骨抑制图像生成目标骨抑制图像。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如权利要求1-7中任一项所述的一种骨抑制图像的生成方法。

10.一种包含计算机可执行指令的存储介质，其特征在于，所述计算机可执行指令在由计算机处理器执行时实现如权利要求1-7中任一项所述的一种骨抑制图像的生成方法。