一种基于残差网络改进的牙齿颜色分类模型  

A Tooth Color Classification Model Improved by Residual Network

作　　者：刘博文 步扬[2] 邹多宏[3] 李建郎 LIU Bowen;BU Yang;ZOU Duohong;LI Jianlang(School of Optical-Electrical and Computer Engineering,University of Shanghai for Science and Technology,Shanghai 200093,China;Shanghai Institute of Optics and Fine Mechanics,Chinese Academy of Sciences,Shanghai 201899,China;Shanghai Ninth People's Hospital,Shanghai Jiao Tong University School of Medicine,Shanghai 200011,China)

机构地区：[1]上海理工大学光电信息与计算机工程学院,上海200093 [2]中国科学院上海光学精密机械研究所,上海201899 [3]上海交通大学医学院附属第九人民医院,上海200011

出　　处：《软件工程》2024年第3期52-57,共6页Software Engineering

基　　金：国家重点研发计划(2020YFB2007504);上海市地方高校能力建设项目(22010503200);国家自然科学基金面上项目(61975217)。

摘　　要：针对传统牙齿比色方法准确率低和效率低等问题,提出一种基于残差网络改进的牙齿颜色分类模型。该模型通过融合多层卷积结果以及引入压缩与激励注意力机制模块的方式,使网络能学习到更多的图像颜色特征。基于典型牙齿所建数据集进行颜色分类实验,在该数据集上对文中模型与GoogleNet、MobileNet-V1、ResNet-34和ResNet-50等模型进行颜色分类预测结果比较。实验结果表明,文中模型优于传统模型,预测分类准确度达到91.16%,有效提高了牙齿颜色分类准确率和效率。This paper proposes an improved tooth color classification model based on residual network to increase the accuracy and efficiency of traditional tooth colorimetric methods.This model enables the network to learn more image color features by fusing multilayer convolutional results and by introducing Squeeze-and-Excitation(SE)attention mechanism module.Color classification experiments are conducted on a typical teeth dataset,on which the color classification prediction results of the proposed model are compared with those of GoogleNet,MobileNet-V1,ResNet-34 and ResNet-50.The experimental results show that the proposed model is better than the traditional models,and the prediction classification accuracy reaches 91.16%,which effectively improves the accuracy and efficiency of tooth color classification.

关 键 词：牙齿比色 颜色分类 深度学习 ResNet-18网络 

分 类 号：TP751.2[自动化与计算机技术—检测技术与自动化装置]