Professional Documents
Culture Documents
分割模型训练方法、曲面确定方法、装置、设备及介质
分割模型训练方法、曲面确定方法、装置、设备及介质
(12)发明专利申请
(10)申请公布号 CN 116310651 A
(43)申请公布日 2023.06.23
(21)申请号 202310311200 .8
(71)申请人 杭州朝厚信息科技有限公司
地址 311400 浙江省杭州市富阳区银湖街
道中国智谷富春园区13号楼13楼1302
室
(72)发明人 马成龙 徐乾
(74)专利代理机构 北京超凡宏宇专利代理事务
所(特殊普通合伙) 11463
专利代理师 曹瑞敏
(51)Int .Cl .
G06V 10/774 (2022 .01)
G06V 20/64 (2022 .01)
G06V 20/70 (2022 .01)
G06V 10/26 (2022 .01)
权利要求书2页 说明书10页 附图5页
(54)发明名称
分割模型训练方法、曲面确定方法、装置、设
备及介质
(57)摘要
本申请提供一种牙齿分割模型的训练方法、
牙齿曲面确定方法、 装置、设备及存储介质,涉及
牙齿模型技术领域。该牙齿分割模型的训练方法
包括: 获取预设牙齿三维模型;为多个初始面片
标注属性信息, 构建训练样本集;根据训练样本
集, 对预设牙齿分割模型进行训练, 得到目标牙
齿分割模型, 目标牙齿分割模型用于确定待检测
牙齿三维模型中各面片的类别和位置。用本申请
的训练方法训练出的牙齿分割模型来判断牙齿
面片类别和位置, 可以提高判断的准确度,并且,
用该模型可以实现对牙齿面片类别的自动判断,
节省了工作人员的时间。
CN 116310651 A
CN 116310651 A 权 利 要 求 书 1/2 页
2
CN 116310651 A 权 利 要 求 书 2/2 页
分割模型。
6 .一种牙齿曲面确定方法,其特征在于, 包括:
获取待检测牙齿三维模型中的多个面片;
根据所述多个面片, 采用目标牙齿分割模型进行处理, 确定所述多个面片的类别;
根据所述多个面片的类别和所述待检测牙齿三维模型, 生成牙齿曲面模型。
7 .如权利要求6所述的方法,其特征在于, 所述根据所述多个面片的类别和所述待检测
牙齿三维模型, 生成牙齿曲面模型, 包括:
将所述待检测牙齿三维模型中类别为牙齿的各第二目标面片进行连接, 得到初始曲面
模型;
去除所述初始曲面模型中类别为非牙齿面片的各第三目标面片, 得到所述牙齿曲面模
型。
8 .一种牙齿分割模型的训练装置, 其特征在于,包括:
获取模块,用于获取预设牙齿三维模型, 所述预设牙齿三维模型包括:多个初始面片,
多个所述初始面片的类别不同;
标注模块, 用于为多个所述初始面片标注属性信息, 构建训练样本集,所述属性信息用
于指示多个所述初始面片的类别、 以及多个所述初始面片的位置;
训练模块, 用于根据所述训练样本集, 对预设牙齿分割模型进行训练, 得到目标牙齿分
割模型, 所述目标牙齿分割模型用于确定待检测牙齿三维模型中各面片的类别。
9 .一种计算机设备,其特征在于, 包括:处理器、存储介质和总线,所述存储介质存储有
所述处理器可执行的程序指令, 当计算机设备运行时, 所述处理器与所述存储介质之间通
过总线通信, 所述处理器执行所述程序指令, 以执行如权利要求1至7任一所述的牙齿分割
模型的训练方法的步骤。
10 .一种计算机可读存储介质, 其特征在于,所述存储介质上存储有计算机程序, 所述
计算机程序被处理器运行时执行如权利要求1至7任一所述的牙齿分割模型的训练方法的
步骤。
3
CN 116310651 A 说 明 书 1/10 页
分割模型训练方法、曲面确定方法、装置、设备及介质
技术领域
[0001] 本申请涉及牙齿模型技术领域,具体而言,
涉及一种牙齿分割模型的训练方法、
牙
齿曲面确定方法、 装置、
设备及存储介质。
背景技术
[0002] 牙齿建模是临床诊断中一项十分重要的技术手段,该技术采集目标病例的口腔三
维信息, 并确定采集到的口腔三维信息中的牙齿信息,从而根据牙齿信息对目标病例的牙
齿进行建模, 便于工作人员用牙齿模型来实现对牙齿问题的诊断。
[0003] 通常,对牙齿三维模型中牙齿位置的判断由工作人员凭借肉眼观察以及经验进
行, 这种判断方式极大程度地依赖工作人员的经验, 且人眼判断容易有误差,
容易导致对牙
齿位置判断的准确度低, 同时也需要花费技术人员的大量时间。
发明内容
[0004] 本申请的目的在于, 针对上述现有技术中的不足, 提供一种牙齿分割模型的训练
方法、 牙齿曲面确定方法、装置、设备及存储介质,以便解决现有技术中存在的问题。
[0005] 为实现上述目的,本申请实施例采用的技术方案如下:
[0006] 第一方面,本申请实施例提供了一种牙齿分割模型的训练方法, 包括:
[0007] 获取预设牙齿三维模型, 所述预设牙齿三维模型包括: 多个初始面片,多个所述初
始面片的类别不同;
[0008] 为多个所述初始面片标注属性信息, 构建训练样本集, 所述属性信息用于指示多
个所述初始面片的类别、 以及多个所述初始面片的位置;
[0009] 根据所述训练样本集, 对预设牙齿分割模型进行训练, 得到目标牙齿分割模型,所
述目标牙齿分割模型用于确定待检测牙齿三维模型中各面片的类别和位置。
[0010] 在可能的实现示例中, 所述为多个所述初始面片标注属性信息, 构建训练样本集,
包括:
[0011] 对所述预设牙齿三维模型进行采样, 确定每个初始面片的属性信息; 所述属性信
息包括: 多个所述初始面片的中心位置、 法向量、
以及面片类别, 所述面片类别用于指示多
个所述初始面片为牙齿面片或非牙齿面片, 所述中心位置和所述法向量用于指示多个所述
初始面片的坐标;
[0012] 为多个所述初始面片标注所述属性信息, 生成样本面片,构建所述训练样本集。
[0013] 在可能的实现示例中, 所述对所述预设牙齿三维模型进行采样, 确定每个初始面
片的属性信息之前, 所述方法还包括:
[0014] 确定所述预设牙齿三维模型中各初始面片的尺寸是否满足预设尺寸条件;
[0015] 若所述各初始面片中存在满足所述预设尺寸条件的第一目标面片, 则去除所述预
设牙齿三维模型中的所述第一目标面片;
[0016] 所述对所述预设牙齿三维模型进行采样, 确定每个初始面片的属性信息,包括:
4
CN 116310651 A 说 明 书 2/10 页
5
CN 116310651 A 说 明 书 3/10 页
所述存储介质之间通过总线通信, 所述处理器执行所述程序指令,以执行上述实施例提供
的牙齿分割模型的训练方法的步骤。
[0038] 第五方面,本申请实施例提供了一种计算机可读存储介质,所述存储介质上存储
有计算机程序, 所述计算机程序被处理器运行时执行上述实施例提供的牙齿分割模型的训
练方法的步骤。
[0039] 本申请的有益效果是:本申请提供了一种牙齿分割模型的训练方法, 先获取预设
牙齿三维模型, 再为多个初始面片标注属性信息,构建训练样本集,最后根据训练样本集,
对预设牙齿分割模型进行训练, 得到目标牙齿分割模型,目标牙齿分割模型即可用于确定
待检测牙齿三维模型中各面片的类别和位置。相比于现有技术中用人眼来判断面片的类
别,用本申请的训练方法训练出的牙齿分割模型来判断牙齿面片类别, 可以提高判断的准
确度, 并且,
用该模型可以实现对牙齿面片类别的自动判断,节省了工作人员的时间。
附图说明
[0040] 为了更清楚地说明本申请实施例的技术方案, 下面将对实施例中所需要使用的附
图作简单地介绍, 应当理解, 以下附图仅示出了本申请的某些实施例,因此不应被看作是对
范围的限定, 对于本领域普通技术人员来讲, 在不付出创造性劳动的前提下,还可以根据这
些附图获得其他相关的附图。
[0041] 图1为本申请实施例提供的一种牙齿分割模型的训练方法的流程示意图;
[0042] 图2为本申请一实施例提供的对预设牙齿曲面样本进行预处理的方法流程示意图
之一;
[0043] 图3为本申请一实施例提供的对预设牙齿曲面样本进行预处理的方法流程示意图
之二;
[0044] 图4为本申请一实施例提供的对目标牙齿分割模型进行验证的方法的流程示意
图;
[0045] 图5为本申请实施例提供的一种牙齿分割模型的应用方法的流程示意图;
[0046] 图6为本申请一实施例提供的对目标牙齿分割模型的输出结果进行修正的方法的
流程示意图;
[0047] 图7为本申请一实施例提供的初始曲面模型的示意图之一;
[0048] 图8(a)和图8(b)为本申请一实施例提供的初始曲面模型的示意图之二;
[0049] 图9为本申请一实施例提供的牙齿分割模型的训练装置的结构示意图;
[0050] 图10为本申请一实施例提供的计算机设备的结构示意图。
具体实施方式
[0051] 为使本申请实施例的目的、技术方案和优点更加清楚,下面将结合本申请实施例
中的附图, 对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是
本申请一部分实施例, 而不是全部的实施例。
[0052] 因此,以下对在附图中提供的本申请的实施例的详细描述并非旨在限制要求保护
的本申请的范围, 而是仅仅表示本申请的选定实施例。基于本申请中的实施例,本领域普通
技术人员在没有作出创造性劳动前提下所获得的所有其他实施例, 都属于本申请保护的范
6
CN 116310651 A 说 明 书 4/10 页
围。
[0053] 在本申请的描述中, 需要说明的是, 若出现术语“上”、“下”、等指示的方位或位置
关系为基于附图所示的方位或位置关系, 或者是该申请产品使用时惯常摆放的方位或位置
关系, 仅是为了便于描述本申请和简化描述, 而不是指示或暗示所指的装置或元件必须具
有特定的方位、 以特定的方位构造和操作, 因此不能理解为对本申请的限制。
[0054] 此外, 本申请的说明书和权利要求书及上述附图中的术语“第一”、 “第二”等是用
于区别类似的对象, 而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在
适当情况下可以互换, 以便这里描述的本申请的实施例能够以除了在这里图示或描述的那
些以外的顺序实施。此外, 术语“包括”和“具有”以及他们的任何变形, 意图在于覆盖不排他
的包含, 例如, 包含了一系列步骤或单元的过程、 方法、系统、产品或设备不必限于清楚地列
出的那些步骤或单元, 而是可包括没有清楚地列出的或对于这些过程、 方法、产品或设备固
有的其它步骤或单元。
[0055] 需要说明的是, 在不冲突的情况下, 本申请的实施例中的特征可以相互结合。
[0056] 本申请实施例提供了一种牙齿分割模型的训练方法, 该牙齿分割模型的训练方法
可由任一集成有预设牙齿分割模型生成算法的计算机设备生成, 计算机设备例如可以是面
向终端的计算机设备, 也可以是后端的服务器。
[0057] 以下结合附图通过多个示例对本申请实施例提供的牙齿分割模型的训练方法、 装
置、 设备及存储介质进行具体的示例说明。
[0058] 本申请一实施例提供了一种牙齿分割模型的训练方法的可能实施方式, 图1为本
申请实施例提供的一种牙齿分割模型的训练方法的流程示意图, 如图1所示, 该方法包括:
[0059] S101、
获取预设牙齿三维模型。
[0060] 在对牙齿分割模型进行训练之前, 需要先获取预设牙齿三维模型, 预设牙齿三维
模型中包括多个初始面片, 预设牙齿三维模型用于构建训练样本集。
[0061] 其中 , 每个初始面片的类别不同 (面片的类别例如可以包括牙齿面片、非牙齿面
片) , 预设牙齿三维模型是用三维口腔扫描仪重建而来的口腔模型, 需要说明的是, 初始面
片的数量例如可以为4096个, 4096是基于对训练速度和训练质量综合得出的数值,用4096
个样本面片对预设牙齿分割模型进行训练, 不仅可以保证训练速度, 还能确保训练得到的
目标牙齿分割模型的质量。
[0062] S102、
为多个初始面片标注属性信息, 构建训练样本集。
[0063] 根据步骤S101得到预设牙齿三维模型后, 可以在预设牙齿三维模型中标注出初始
面片的属性信息, 并将预设牙齿三维模型和标注的属性信息记录在训练样本集中, 从而构
建出训练样本集。训练样本集例如可以为一个文本文件。
[0064] 属性信息用于指示多个初始面片的类别、 以及多个初始面片的位置, 具体地,属性
信息例如可以包括: 多个初始面片的中心位置、 法向量、 以及面片类别, 面片类别用于指示
多个初始面片为牙齿面片或非牙齿面片, 中心位置和法向量用于指示多个初始面片的坐
标。
[0065] 在本实施例中, 例如可以预先对预设牙齿三维模型进行采样, 得到预设数量个初
始面片(4096个) , 再标注出多个初始面片的类别、中心位置以及法向量, 从而生成样本面
片。初始面片的中心位置和法向量可以 用矩阵形式表达矩阵的每一行表示一个面片的坐
7
CN 116310651 A 说 明 书 5/10 页
标, 例如(x,y,z,nx,ny,
nz) ,
x、y、z表示面片的中心位置的坐标, nx、 ny、
nz表示面片的单位
法向量坐标。
[0066] 可选的, 在得到训练样本集之后, 为使训练样本集中的数据与模型的方向无关, 还
可以同时对(x, y,z)和(nx,ny,nz)乘一个随机的旋转矩阵, 也即对数据做旋转增广; 在旋转
增广后, 为使数据与模型位置无关, 可以将(x ,y ,z)归一化,即x ,y ,z∈[0 ,1];
最后,为使数
据与面片排列无关, 可以对矩阵的行进行随机排列。
[0067] S103、根据训练样本集, 对预设牙齿分割模型进行训练, 得到目标牙齿分割模型。
[0068] 将步骤S102得到的训练样本集代入到预设牙齿分割模型中, 即可对预设牙齿分割
模型进行训练。 由于预设牙齿曲面样本是由多个样本面片组成, 且, 对每个样本面片的样本
类别、 中心位置以及法向量都进行了标注, 因此, 训练后的预设牙齿分割模型即为目标牙齿
分割模型, 可以用于确定待检测牙齿三维模型中各面片的类别和位置。其中, 预设牙齿分割
模型例如可以为DGCNN语义分割网络, 当然,也可以选择其他类型的语义分割网络, 本申请
对此不做限定。
[0069] 综上, 本实施例提供了一种牙齿分割模型的训练方法, 本申请提供了一种牙齿分
割模型的训练方法, 先获取预设牙齿三维模型, 再为多个初始面片标注属性信息, 构建训练
样本集, 最后根据训练样本集, 对预设牙齿分割模型进行训练, 得到目标牙齿分割模型, 目
标牙齿分割模型即可用于确定待检测牙齿三维模型中各面片的类别和位置。相比于现有技
术中用人眼来判断面片的类别, 用本实施例的训练方法训练出的牙齿分割模型来判断牙齿
面片类别, 可以提高判断的准确度, 并且,用该模型可以实现对牙齿面片类别的自动判断,
节省了工作人员的时间。
[0070] 本申请一实施例还提供了对样本面片进行预处理的方法的可能实施方式, 图2为
本申请一实施例提供的对样本面片进行预处理的方法流程示意图之一, 如图2所示, 在对预
设牙齿三维模型进行采样, 确定每个初始面片的属性信息之前, 还可以包括:
[0071] S201、确定预设牙齿三维模型中各初始面片的尺寸是否满足预设尺寸条件。
[0072] 由于牙齿三维模型是用三维口腔扫描仪重建而来, 其不可避免地会伴随着噪声,
如扫描时用的底座、气泡、补洞等非牙齿三维模型的面片, 因此, 在对预设牙齿三维模型进
行采样之前, 可以先确定预设牙齿三维模型中各初始面片的尺寸是否满足预设尺寸条件,
满足预设条件的初始面片可以被认为是非牙齿三维模型的面片, 即为需要从样本面片中去
除的面片。
[0073] S202、若各初始面片中存在满足预设尺寸条件的第一目标面片, 则去除预设牙齿
三维模型中的第一目标面片。
[0074] 在根据步骤S202获取到满足预设尺寸条件的初始面片后, 将得到的初始面片作为
第一目标面片, 并在预设牙齿三维模型中去除第一目标面片, 即可完成对样本面片的预处
理。
[0075] 则上述实施例中, 对预设牙齿三维模型进行采样, 得到多个样本面片的中心位置
以及法向量, 包括:
[0076] S203、对去除第一目标面片后的预设牙齿三维模型进行采样, 得到多个样本面片
的中心位置以及法向量。
[0077] 在步骤S202完成了对样本面片的预处理之后, 可以根据去除第一目标面片后的预
8
CN 116310651 A 说 明 书 6/10 页
9
CN 116310651 A 说 明 书 7/10 页
10
CN 116310651 A 说 明 书 8/10 页
齿三维模型中的牙齿面片进行判断 , 只需将待检测牙齿三维模型代入目标牙齿分割模型
中, 即可实现自动判断, 提高判断准确率的同时也节省了工作人员的时间。
[0105] 图6为本申请一实施例提供的对目标牙齿分割模型的输出结果进行修正的方法的
流程示意图, 如图6所示,步骤S503中,根据多个面片的类别和待检测牙齿三维模型, 生成牙
齿曲面模型, 包括:
[0106] S601、将待检测牙齿三维模型中类别为牙齿的各第二目标面片进行连接, 得到初
始曲面模型。
[0107] 在确定多个面片的类别之后, 目标牙齿分割模型可以将类别为牙齿的各面片作为
第二目标面片, 并将第二目标面片进行连接, 得到初始曲面模型, 具体的方法为: a)搜索所
有第二目标面片的邻接面片, 将这些邻接面片加入到第二目标面片邻域中; b)将所有的邻
域作为新的第二目标面片集合, 重复a)步骤直到满足设定的膨胀次数, 即可得到初始曲面
模型, 得到的初始曲面模型例如可以如图7所示。
[0108] 如图8(a)所示,根据步骤a)和b)得到的初始曲面模型中, 第二目标面片的邻域可
能会有多个, 则上述步骤还可以包括: c)计算所有区域两两之间的距离, 计算各区域的最近
点对, 分别得出各区域的最相邻区域(一个区域的中心点与除该区域的其他区域中的最近
点, 称为一个最近点对) ; d)计算两个点之间的最短路径; e)按照步骤a)和b)的方法进行膨
胀; f)合并膨胀后的路径和它连接的两个区域, 重复上述步骤直到只剩下一个区域。根据步
骤a)‑f),即可得到如图8(b)所示的初始曲面模型。
[0109] 对初始模型进行修正后, 即可得到牙齿曲面模型, 修正的方法如步骤S602。
[0110] S602、去除初始曲面模型中类别为非牙齿面片的各第三目标面片, 得到牙齿曲面
模型。
[0111] 目标牙齿分割模型可以将类别为非牙齿的各面片作为第三目标面片, 并去除初始
曲面模型中类别为非牙齿面片的各第三目标面片, 从而得到牙齿曲面模型。具体的方法为:
对膨胀后的邻域点集做一次标签筛选, 如果顶点的邻域中被判定为牙齿的面片数量很少,
则丢弃该面片及其邻域。
[0112] 通过本实施例的方法, 可以将类别为牙齿的各面片连接成一个整体, 并且, 去除了
类别为非牙齿的各面片, 进一步实现了对目标牙齿分割模型的输出结果进行修正。
[0113] 如下继续对执行本申请上述任一实施例提供的牙齿分割模型的训练方法和牙齿
曲面确定方法的装置、 设备及存储介质进行相应的解释, 其具体的实现过程以及产生的技
术效果与前述对应的方法实施例相同, 为简要描述, 本实施例中未提及部分, 可参考方法实
施例中的相应内容。
[0114] 图9为本申请一实施例提供的牙齿分割模型的训练装置的结构示意图 , 如图9所
示, 本申请一实施例还提供了一种牙齿分割模型的训练装置, 包括:
[0115] 获取模块901,用于获取预设牙齿三维模型。其中, 预设牙齿三维模型包括: 多个初
始面片, 多个初始面片的类别不同。
[0116] 标注模块902, 用于为多个初始面片标注属性信息, 构建训练样本集, 属性信息用
于指示多个初始面片的类别、 以及多个初始面片的位置;
[0117] 训练模块903,用于根据训练样本集, 对预设牙齿分割模型进行训练, 得到目标牙
齿分割模型, 目标牙齿分割模型用于确定待检测牙齿三维模型中各面片的类别。
11
CN 116310651 A 说 明 书 9/10 页
12
CN 116310651 A 说 明 书 10/10 页
13
CN 116310651 A 说 明 书 附 图 1/5 页
图1
图2
14
CN 116310651 A 说 明 书 附 图 2/5 页
图3
图4
15
CN 116310651 A 说 明 书 附 图 3/5 页
图5
图6
图7
16
CN 116310651 A 说 明 书 附 图 4/5 页
图8(a)
图8(b)
图9
17
CN 116310651 A 说 明 书 附 图 5/5 页
图10
18