基于深度学习的口腔疾病图像识别系统(UI界面+改进算法+数据集+训练代码)

摘要口腔疾病是影响人类健康的常见疾病传统的口腔疾病诊断依赖于专业医生的经验判断存在效率低、主观性强等问题。随着深度学习技术的快速发展基于计算机视觉的智能检测方法为口腔疾病的早期筛查和辅助诊断提供了新的解决方案。本文设计并实现了一个基于YOLOv8深度学习算法的口腔疾病智能检测系统旨在提高口腔疾病检测的准确性和效率项目简介基于YOLOv8深度学习算法的口腔疾病智能检测系统支持6种常见口腔疾病的自动识别与可视化分析。系统概述本系统采用YOLOv8目标检测算法作为核心检测引擎能够识别牙结石Calculus、龋齿Caries、牙龈炎Gingivitis、牙齿缺失 Hypodontia、口腔溃疡Mouth Ulcer和牙齿变色Tooth Discoloration等6种常见口腔疾病。在系统架构上后端采用Django框架构建RESTful API实现用户认证、图像上传、模型推理和 结果管理等功能前端采用Vue.js框架开发提供友好的用户交互界面支持图像上传、检测结果可视化、历史记录查询和数据统计分 析等功能。系统采用MySQL数据库存储用户信息和检测记录通过JWT令牌实现安全的身份认证机制。深度学习的口腔疾病检测系统采用YOLOv8算法实现了六类常见口腔疾病的自动识别与定位。系统具备口腔图像智能检测功能 支持实时上传图像并返回检测结果用户可选择YOLOv8或YOLOv8-CBAM模型进行检测。检测结果通过可视化方式展示在原图上标 注疾病位置、类别和置信度直观呈现病灶区域。系统提供完善的历史记录管理功能支持检测记录的查询、筛选和删除方便用 户追踪诊断历史。数据统计分析模块提供疾病分布、检测趋势等可视化图表辅助医生进行数据分析和决策。系统实现了用户权限 管理机制区分管理员和普通用户角色管理员可管理用户账户和系统数据。系统采用DjangoVue.js前后端分离架构通过RESTf ul API实现前后端通信支持多用户并发访问为口腔疾病的早期筛查和辅助诊断提供了高效、准确的技术支持。实验结果表明该系统能够准确识别多种口腔疾病具有良好的检测性能和用户体验可为口腔疾病的早期筛查和辅助诊断提供有效的技术支持具有一定的实用价值和推广前景。系统架构本系统采用经典的架构设计图1 深度学习的口腔疾病图像识别系统核心亮点本系统以 YOLOv8 目标检测为核心融合深度学习推理、数据管理与可视化界面实现了口腔疾病识别、结果分析和历史追踪的一体化智能诊断。算法特点本算法以 YOLOv8 目标检测为基础融合疾病类别、置信度、检测框位置、病灶数量及类别分布等多维特征并结合数据统计分析 完成口腔健康评估具有检测与分析一体化的特点。性能突破本文在口腔疾病检测数据集上开展实验数据集共包含 3488 张图像和 3616 个标注框其中训练集 2441 张、验证集 698 张、测试集 349 张。通过 150 轮训练YOLOv8 模型在验证集上达到 99.4% 的 mAP0.5 和 97.9% 的 mAP0.5:0.95能够准确完成六类口腔疾病检测任务为临床辅助诊断提供了有效支撑。图2 基线模型性能分析图核心技术YOLOv8 轻量级目标检测模型结合高效特征提取与多尺度特征融合技术在口腔疾病数据集3,488 张图像3,616 个标注框上训练 150 轮实现对牙结石、龋齿、牙龈炎、牙齿缺失、口腔溃疡、牙齿变色六类疾病的高精度智能识别并结合 Web 可视化界面、历史记录管理和数据统计分析提升口腔健康辅助诊断能力算法详解YOLOv8 是 Ultralytics 推出的新一代目标检测模型在网络结构、检测头设计、损失函数和训练策略等方面对 YOLOv5 进行了优化。该模型提供 N、S、M、L、X 等多种尺度版本以适应不同场景需求。相比 YOLOv5YOLOv8 将 C3 模块替换为 C2f 模块增强了特征提取能力采用解耦头和 Anchor-Free 机制提高了检测精度与收敛效率在损失计算中引入 TaskAlignedAssigner 和 Distribution Focal Loss增强了边界框回归能力同时在训练后期关闭 Mosaic 数据增强进一步提升模型精度与泛化性能。图3 YOLOv8网络架构图基于CBAM注意力机制的模型改进为进一步提升模型对关键特征的关注能力本研究在YOLOv8基础上引入了CBAMConvolutional Block Attention Module注意力机制。CBAM通过串联通道注意力Channel Attention和空间注意力Spatial Attention两个子模块实现对特征图的自适应加权。其中通道注意力模块采用平均池化和最大池化双路径提取全局特征描述 符经过共享MLP网络后生成通道权重空间注意力模块则通过对特征图进行通道维度的池化操作利用7×7卷积核捕获空间位置的 重要性。在网络结构设计上CBAM模块被策略性地插入到Backbone的P3层输出后layer 5以及Neck的三个检测尺度特征层P3、P4对应的layer 17和21共计4个位置在保持模型轻量化的同时总参数量3.02M计算量8.2 GFLOPs有效增强了模型对口腔病变区域的特 征表达能力。训练过程采用SGD优化器初始学习率0.01动量0.937权重衰减0.0005共训练150轮。技术优势分析CBAM注意力机制的引入为模型带来了显著的技术优势首先通过通道注意力和空间注意力的串联设计实现了”关注什么特征”和 “关注哪里位置”的双重自适应筛选相比单一注意力机制能更精准地捕获口腔病变的关键区域其次采用稀疏部署策略仅在Ba ckbone P3层和Neck的三个检测尺度关键位置插入CBAM模块新增参数量仅3,366个在保持模型轻量化总参数3.02M和实时性 的同时实现了从浅层纹理到深层语义的多尺度特征增强此外CBAM模块采用即插即用设计无需修改主干网络结构通过YAML 配置文件即可灵活集成具有良好的可扩展性和工程实用性。系统功能本系统集成用户登录注册、口腔图像上传检测、六类疾病智能识别、检测结果可视化、历史记录管理、数据统计分析及用户权限管理等功能。功能概述本系统面向口腔疾病智能筛查与辅助诊断需求基于 YOLOv8 目标检测模型实现对口腔图像中牙结石、龋齿、牙龈炎、牙齿缺失、 口腔溃疡、牙齿变色六类疾病的识别、定位与可视化展示系统提供用户登录注册与权限管理功能能够对检测结果进行病灶定位 标注并结合历史数据分析完成口腔健康状况评估同时将检测记录与分析数据进行存储和展示在数据统计模块中实现检测结果 的多维度可视化呈现为口腔医疗机构智能化诊断和患者健康管理提供一体化支持。系统流程图系统采用 Django Vue.js 前后端分离架构开发后端基于 Python 3.12 和 Ultralytics YOLOv8 框架实现口腔疾病目标检测使用 Django REST Framework 提供 API 接口通过 JWT 实现身份认证前端基于 Vue.js 3 构建响应式界面结合 Element Plus 和 ECharts 实现数据可视化系统采用 MySQL 数据库存储用户信息和检测记录。系统面向口腔疾病智能筛查需求集成用户登录注册、图像检测、疾病识别、结果可视化、历史 管理与数据统计等功能为口腔医疗智能化诊断提供一体化技术支持。图4 系统总流程图系统优势本系统基于 YOLOv8 目标检测模型在验证集上达到 99.4% 的 mAP0.5能够准确完成六类口腔疾病识别与病灶定位并结合历史数据分析实现口腔健康状况评估具备检测与分析一体化的特 点。相比传统人工诊断方式系统具有自动化程度高、检测效率快推理速度 1.3ms/图、结果客观性强等优势。与此同时系统采用 Django Vue.js 前后端分离架构支持多用户并发访问、JWT 身份认证和角色权限管理并结合 Element Plus 和 ECharts 构建响应式数据可视化界面具备良好的交互性、可扩展性和跨平 台特性可为口腔医疗机构智能化诊断和患者健康管理提供有效支持。运行展示系统界面采用响应式布局设计包含顶部导航栏、侧边功能菜单和中部内容展示区域。系统能够完成口腔图像上传、六类疾病智能检测并对检测结果进行可视化标注展示同时支持历史记录查询、检测详情查看、数据统计分析及用户权限管理等功能具有界面简洁、交互友好和操作便捷等特点。检测效果展示登录界面图5 登录主界面用户登录界面展示系统入口图6 注册主界面用户注册界面新用户创建账号系统运行模块图7 首页概览图8 图片检测图9 图片检测牙结石图10 图片检测牙龈炎图11 图片检测牙缺失图12 图片检测牙齿变色图13 图片检测口腔溃疡图14 图片检测龋齿图15 图片CBAM注意力机制检测龋齿图16 图片CBAM注意力机制检测牙结石图17 图片CBAM注意力机制检测牙龈炎图18 图片CBAM注意力机制检测牙缺失图19 图片CBAM注意力机制检测口腔溃疡图20 图片CBAM注意力机制检测牙齿变色图21 历史记图22 历史记录-详情图23 历史记录-查询图24 历史记录-删除图25 算法对比图26 个人中心图27 用户管理数据集与训练本文构建了口腔疾病检测数据集共包含 3488 张图像与 3616 个标注框涵盖牙结石、龋齿、牙龈炎、牙齿缺失、口腔溃疡、牙齿变色六类疾病其中训练集 2441 张、验证集 698 张、测试集 349 张。模型训练采用 YOLOv8n 目标检测框架模型参数量为 3.01M输入图像尺寸设为 640×640batch size 为 8优化器为 SGD学习率 0.01动量 0.937训练轮数为 150采用 Mosaic 数据增强、随机翻转和 HSV 颜色变换等策略。在 NVIDIA GeForce RTX 4070 Ti SUPER 显卡环境下完成 1.67 小时训练后模型在验证集上取得了优异的检测效果精确率 99.7%召回率 99.5%mAP0.5 达到 99.4%mAP0.5:0.95 达到 97.9%推理速度 1.3ms/图能够满足口腔疾病智能识别与临床辅助诊断任务需求。数据集构建本文构建并使用口腔疾病检测数据集开展实验研究。该数据集以口腔医学影像中的病灶区域为目标对象涵盖牙结石、龋齿、牙龈 炎、牙齿缺失、口腔溃疡、牙齿变色六类常见口腔疾病包含不同拍摄设备、光照条件和拍摄角度下采集的图像数据具有较好的 临床代表性。数据集共包含 3488 张图像和 3616 个标注框其中训练集 2441 张、验证集 698 张、测试集 349 张对应标注框数量分别为 2533、720 和 363。该数据集为 YOLOv8 模型训练、性能验证以及后续系统部署提供了可靠的数据基础。图24 数据集划分及类别信息统计示意图数据集增强训练阶段采用Mosaic拼接、随机翻转、HSV色彩扰动、随机擦除等数据增强策略提升模型对不同光照、角度和遮挡条件下的泛化能力。图25 图像增强训练流程模型训练采用端到端的方式首先加载训练集和验证集进行数据预处理然后加载 YOLOv8n 预训练权重进行模型初始化接着使用 SGD 优化器进行 150 轮迭代训练每轮训练后在验证集上评估性能指标系统自动保存验证集上性能最佳的模型权重最终输出完整 的性能指标和训练曲线。图26 模型训练流程训练流程1. 开始训练 → 加载训练集和验证集进行数据预处理2. 模型初始化 → 加载YOLOv8预训练权重yolov8n.pt使用标准YOLOv8架构3. 模型训练 → 使用SGD优化器进行150轮迭代训练应用数据增强技术4. 模型验证 → 每轮训练后在验证集上评估性能指标Precision, Recall, mAP0.5, mAP0.5:0.955. 最佳模型保存 → 系统自动监控验证性能保存验证集上性能最佳的模型权重best.pt6. 训练完成 → 输出完整的性能指标报告和训练曲线图训练配置硬件环境:软件环境训练超参数学习率调度策略学习率调度策略采用线性衰减方式前3个epoch进行warmup预热学习率从0线性增长到初始学习率0.01之后按线性方式从0.01逐步衰减到最终学习率0.0001。训练结果训练曲线分析下图展示了模型在150轮训练过程中的完整性能变化包括损失函数曲线和精度指标曲线图27 训练曲线分析图中展示了10个关键指标的训练过程训练损失box/cls/dfl、验证损失box/cls/dfl、精确率、召回率、mAP50和mAP50-951损失函数曲线从训练结果可以看出模型在训练过程中各项损失函数均呈现稳定下降趋势说明 YOLOv8 模型能够较好地学习目标特征。其中train/box_loss 由约 1.45 快速下降至约 0.25train/cls_loss 由约 3.5 快速下降至约 0.2train/dfl_loss 由约 1.7 下降至约 0.85验证集上的 val/box_loss 由约 1.50 下降至约 0.23val/cls_loss 由约 4.0 下降至约 0.15val/dfl_loss 由约 2.0 下降至约 0.82验证集损失曲线与训练集基本一致整体曲线变化平滑、波动较小 表明模型训练过程较为稳定未出现明显的过拟合现象。2精度指标曲线从精度指标变化情况来看模型各项性能指标均随训练轮次增加而持续提升。Precision 曲线由初始约 0.55 快速提升至约 1.0接近100%并保持稳定Recall 曲线由初始约 0.32 稳步上升至约 1.0接近100%mAP0.5 由初始约 0.15 快速提升最终达到约 0.99599.5%mAP0.5:0.95 由初始约 0.10 持续增长至约 0.9999%。从整体趋势看模型训练过程大致经历了快速提升前20-30轮、稳定优化和收敛趋稳三个阶段最终在验证集上 取得了接近完美的检测效果说明所构建的 YOLOv8 模型在该目标检测任务中具有极高的精度和良好的收敛性。3Precision-Recall 曲线图28 Precision-Recall 曲线展示模型在不同置信度阈值下的精确率和召回率关系4混淆矩阵归一化图29 归一化混淆矩阵归一化混淆矩阵展示模型的分类准确性最佳模型选择在模型训练过程中系统依据验证集性能指标自动保存最优模型权重并以验证集mAP0.5:0.95作为最佳模型判定标准。当该指标达到当前训练过程中的最高值时对应模型权重将被保存为best.pt。本次实验中最优模型出现在第150轮最后一轮其验证集mAP0.5:0.95为0.97997.9%同时该轮次的其他指标也表现优异Precision为0.997Recall为0.995mAP0.5为0.994。模型文件保存于runs/train/yolov8/weights/best.pt。训练稳定性分析收敛速度 前20轮快速收敛14.95% → 84.55%20-40轮稳定提升40轮后缓慢优化过拟合控制训练集与验证集损失走势一致无过拟合训练稳定性损失曲线平滑学习率逐步衰减训练稳定最终状态第150轮达最优mAP0.5:0.95为97.94%项目资源我们提供项目的完整技术资源包括源代码、训练脚本、配置文件、数据集和模型权重等全部内容。代码采用模块化设计结构清晰注释完善支持完全复现论文中的所有实验结果。项目提供详细的文件清单和技术架构说明(网页已经提供)帮助用户快速理解项目结构便于二次开发和功能扩展。所有资源均已开源遵循AGPL-3.0协议用户可自由使用、修改和分发。关于项目原创论文原创论文基于深度学习的口腔疾病图像识别系统 注意需要另外付费购买作者信息作者Bob (张家梁)原创声明本项目为原创作品