[发明专利]一种施源器自动重建方法和装置

权利要求书

1.一种施源器自动重建方法，其特征在于，包括：

获取训练数据集，所述训练数据集中包含多组CT样本图像组，每组CT样本图像组中的施源器的类型不同，所述每组CT样本图像组中包含多个CT样本图像和每个CT样本图像对应的标签真值图；

将所述训练数据集输入初始施源器重建模型中，输出每个CT样本图像的预测真值图；

根据所有CT样本图像的预测真值图和每个CT样本图像对应的标签真值图确定损失值；

利用所述损失值对所述初始施源器重建模型的参数进行优化，得到施源器重建模型；

将采集的CT图像输入所述施源器重建模型，输出所述CT图像的施源器重建CT图像。

2.根据权利要求1所述施源器自动重建方法，其特征在于，所述获取训练数据集之后，还包括：

对所述训练数据集中的每个CT样本图像进行直方图均衡化处理；

将所述每个CT样本图像和每个CT样本图像对应的标签真值图的中心点作为裁剪中心点，按照预设尺寸进行裁剪；

对裁剪后的每个CT样本图像和每个CT样本图像对应的标签真值图进行数据增强处理。

3.根据权利要求1所述施源器自动重建方法，其特征在于，所述将所述训练数据集输入初始施源器重建模型中，输出每个CT样本图像的预测真值图，包括：

利用所述初始施源器重建模型从所述CT样本图像进行特征提取，得到特征数据；

将所述特征数据输入激活函数，输出所述CT样本图像的预测真值图。

4.根据权利要求1所述施源器自动重建方法，其特征在于，所述根据所有CT样本图像的预测真值图和每个CT样本图像对应的标签真值图确定损失值，包括：

根据每个CT样本图像的预测真值图和每个CT样本图像对应的标签真值图确定该CT样本图像的子损失值；

根据所有CT样本图像的子损失值、预设值确定所述训练数据集的损失值。

5.根据权利要求1所述施源器自动重建方法，其特征在于，所述利用所述损失值对所述初始施源器重建模型的参数进行优化，得到施源器重建模型，包括：

利用所述损失值对所述初始施源器重建模型的参数进行优化；

判断所述初始施源器重建模型是否收敛；

若否，则利用优化后的初始施源器重建模型对所述训练数据集进行重建，并重新计算损失值；

利用重新计算的损失值对所述优化后的初始施源器重建模型的参数进行优化；

直到所述初始施源器重建模型收敛，得到所述施源器重建模型。

6.一种施源器自动重建装置，其特征在于，所述装置包括：

采集模块，用于获取训练数据集，所述训练数据集中包含多组CT样本图像，每组CT样本图像中的施源器的类型不同，所述每组CT样本图像中包含多个CT样本图像和每个CT样本图像对应的标签真值图；

深度学习模块，用于将所述训练数据集输入初始施源器重建模型中，输出每个CT样本图像的预测真值图；

计算模块，用于根据所有CT样本图像的预测真值图和每个CT样本图像对应的标签真值图确定损失值；

所述深度学习模块，还用于利用所述损失值对所述初始施源器重建模型的参数进行优化，得到施源器重建模型；

所述深度学习模块，还用于将采集的CT图像输入所述施源器重建模型，输出所述CT图像的施源器重建CT图像。

7.根据权利要求6所述施源器自动重建装置，其特征在于，还包括：

处理模块，用于对所述训练数据集中的每个CT样本图像进行直方图均衡化处理；

所述处理模块，还用于将所述每个CT样本图像和每个CT样本图像对应的标签真值图的中心点作为裁剪中心点，按照预设尺寸进行裁剪；

所述处理模块，还用于对裁剪后的每个CT样本图像和每个CT样本图像对应的标签真值图进行数据增强处理。