一种基于深度学习的牙齿比色光谱成像方法及系统

本发明公开了一种基于深度学习的牙齿比色光谱成像方法及系统，采集牙齿和不同厚度陶瓷片样本在多光源条件下的RGB图像和光谱图像，并进行预处理；基于预处理得到的RGB图像和光谱图像，构建训练集和测试集；构建基于U‑Net框架的光谱重建网络模型，所述光谱重建网络模型主体包括多个串联的注意力模块，每个注意力模块均包含通道自注意力和空间注意力两种机制，所述光谱重建网络模型的输入为线性RGB图像，输出为重建的光谱图像；利用所述训练集和测试集对光谱重建网络模型进行训练和测试，利用训练的光谱重建网络模型进行实时牙齿光谱图像重建。本发明能更准确地恢复牙齿的真实反射光谱，显著提高比色精度。