Simplify original tooth meshes to about 10,000 facets. Mesh simplification can be achieved using functions from Open3D or through implementations by MeshCNN, which utilize the bpy library.
DGCNN, TeethGNN, or others.
./scripts/geodesic_distance.py
Geodesic distance maps are extracted during the data preprocessing phase. Based on the ground truth and the coarse segmentation results, extract the segmentation boundaries (vertices). Then, compute the shortest geodesic distance from mesh vertices to the segmentation boundaries and save the results as '.txt' files.
Modify the config and dataset based on the path and filename.
...
python train.py
python test.py
@article{jin2024tsrnet,
title={TSRNet: A Dual-stream Network for Refining 3D Tooth Segmentation},
author={Jin, Hairong and Shen, Yuefan and Lou, Jianwen and Zhou, Kun and Zheng, Youyi},
journal={IEEE Transactions on Visualization and Computer Graphics},
year={2024},
publisher={IEEE}
}