基于双重路径规划的密度峰值聚类三维牙齿分割算法
A 3D Tooth Segmentation Algorithm Based on Density Peak Clustering with Dual Path Planning
-
摘要: 针对当前数字牙科软件中牙齿分割不准确和交互烦琐的问题, 提出双重路径规划下的分段式密度峰值聚类三维牙齿分割算法, 采用凹点匹配原则对牙齿模型进行分割, 简化用户交互操作的同时在一定程度上提高了模型分割的准确率. 首先设计基于双重路径规划算法, 为后续步骤提供稳定的牙齿-牙龈轮廓线; 然后在牙齿-牙龈轮廓线中检测凹点, 根据人类牙齿形态将检测出的凹点分为磨牙附近区域的凹点和切牙附近区域的凹点, 并结合该凹点所处路径上的位置提出分段式的密度峰值聚类对其进行匹配; 对于已配对的凹点执行分裂操作, 按照特定的顺序将分裂后的点重新组合, 生成单颗牙齿的分割线; 最后通过单颗牙齿分割线的最长直径和最短直径的比值, 筛选出不符合条件的单颗牙齿分割线, 利用3DSnake修正牙齿分割线完成牙齿分割. 实验采用400套自采数据进行评估, mIoU达到92.03%. 同时, 在随机抽取的100套Teeth3DS数据集样本中进行测试, 总体失败率为4%. 结果表明, 所提方法在各类牙齿模型上的分割效果优异, 有效减少了用户交互操作的复杂性, 能够在平均20.10秒内完成牙齿分割, 并表现出较强的鲁棒性.Abstract: To address the issues of inaccurate tooth segmentation and cumbersome interactions in current digital dental software, a segmented density peak clustering 3D tooth segmentation algorithm under dual-path planning is proposed. This method employs a concavity matching principle to segment the tooth model, simplifying user interaction while improving segmentation accuracy to a certain extent. First, a dual path planning algorithm is designed to provide stable tooth-gum contour lines for subsequent steps. Then, concave points are detected within the tooth-gum contour lines and classified into two categories based on human dental morphology: concave points near the molar region and those near the incisor region. Combining the location of these concave points along their paths, a segmented density peak clustering approach is introduced for matching. Splitting operations are performed for the paired concave points, and the split points are recombined in a specific sequence to generate single-tooth segmentation lines. Finally, single-tooth segmentation lines are screened based on the ratio of their longest shortest and diameters. Lines not meeting the criteria are corrected using 3DSnake to complete the tooth segmentation. The experiment evaluated 400 self-collected datasets, achieving a mIoU of 92.03%. Additionally, 100 samples were randomly selected from the Teeth3DS dataset for testing, with an overall failure rate of 4%. The results demonstrate that the proposed method achieves excellent segmentation performance across various dental models, significantly reduces the complexity of user interactions, completes dental segmentation in an average of 20.10 seconds, and exhibits strong robustness.