[发明专利]牙齿图像的处理方法及装置、电子设备和存储介质

权利要求书

1.一种牙齿图像的处理方法，其特征在于，包括：

从待处理图像的多个待处理像素中，预测目标牙实例的中心像素，其中，所述待处理像素表示所述待处理图像中未被预测为属于任一牙实例的像素，所述目标牙实例表示当前预测的牙实例；

确定多个所述待处理像素中的第二像素所指向的牙实例中心的预测坐标，其中，所述第二像素表示多个所述待处理像素中的任一像素，所述第二像素所指向的牙实例中心的预测坐标表示基于所述第二像素预测的所述第二像素所属的牙实例的中心像素的坐标；

预测所述目标牙实例对应的聚类参数，其中，所述聚类参数用于表示所述目标牙实例的中心像素的预测坐标的离散程度；

根据所述第二像素所指向的牙实例中心的预测坐标，所述目标牙实例的中心像素的坐标，以及所述目标牙实例对应的聚类参数，预测所述第二像素属于所述目标牙实例的中心的概率；

根据所述第二像素属于所述目标牙实例的中心的概率，从多个所述待处理像素中预测属于所述目标牙实例的像素集合，得到所述目标牙实例包含的像素集合的预测结果；

根据所述待处理图像中的多个牙实例包含的多个像素集合的预测结果，得到所述待处理图像的牙实例分割结果，其中，一个牙实例对应于一颗牙齿，所述牙实例分割结果包括所述待处理图像中的像素所属的牙实例的信息；

基于所述牙实例分割结果进行牙位定位，得到所述待处理图像的牙位定位结果。

2.根据权利要求1所述的方法，其特征在于，所述从待处理图像的多个待处理像素中，预测目标牙实例的中心像素，包括：

从待处理图像的多个待处理像素中，确定位于牙实例中心的概率最大的第一像素；

在所述第一像素位于牙实例中心的概率大于或等于第一预设值的情况下，将所述第一像素预测为目标牙实例的中心像素。

3.根据权利要求2所述的方法，其特征在于，所述在所述第一像素位于牙实例中心的概率大于或等于第一预设值的情况下，将所述第一像素预测为目标牙实例的中心像素，包括：

在多个所述待处理像素中、位于牙实例中心的概率大于或等于所述第一预设值的像素数大于或等于第二预设值，且所述第一像素位于牙实例中心的概率大于或等于第一预设值的情况下，将所述第一像素预测为目标牙实例的中心像素。

4.根据权利要求1所述的方法，其特征在于，所述确定多个所述待处理像素中的第二像素所指向的牙实例中心的预测坐标，包括：

确定多个所述待处理像素中的第二像素到所述第二像素所属的牙实例的中心像素的预测偏移量；

根据所述第二像素的坐标，以及所述第二像素到所述第二像素所属的牙实例的中心像素的预测偏移量，确定所述第二像素所指向的牙实例中心的预测坐标。

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

将所述待处理图像输入第一神经网络，经由所述第一神经网络得到所述第二像素到所述第二像素所属的牙实例的中心像素的预测偏移量，所述第二像素所属的牙实例的聚类参数，以及所述第二像素位于牙实例中心的概率。

6.根据权利要求5所述的方法，其特征在于，所述第一神经网络包括第一解码器和第二解码器；

所述将所述待处理图像输入第一神经网络，经由所述第一神经网络得到所述第二像素到所述第二像素所属的牙实例的中心像素的预测偏移量，所述第二像素所属的牙实例的聚类参数，以及所述第二像素位于牙实例中心的概率，包括：

将所述待处理图像输入第一神经网络，经由所述第一解码器得到所述第二像素到所述第二像素所属的牙实例的中心像素的预测偏移量，以及所述第二像素所属的牙实例的聚类参数，经由所述第二解码器得到所述第二像素位于牙实例中心的概率。