一、算法
1. 概述
肾结石在全球公共卫生中面临着重大挑战,准确及时的诊断和有效的治疗对患者康复至关重要。传统的肾结石检测方法在识别小型或不规则形状的目标仍然面临着众多挑战。因此,本研究提出了一种基于改进的YOLOv8肾结石小目标检测网络,简称STD-YOLOv8s。
实验评估表明,STD-YOLOv8s模型在TEZ_ROI_AUG数据集上的召回率、mAP50和mAP50-95相比YOLOv8s基线模型分别提升了8.91%、8.7%和3.13%。STD-YOLOv8s网络为肾结石的临床诊断和治疗提供更高效的方法,优化了肾结石的临床诊断和治疗策略。
2. 算法测试效果
图1:比较不同网络对小目标肾结石检测的输出目标框结果:(a)-(d)STD-YOLOv8s;(e)-(h)YOLOv8s;(i)-(l)YOLOv6s;(m)-(p)YOLOv5s.
图2:本文提出的SPD-YOLOv8s网络与YOLOv8基线网络的主干网络检测输出特征热力图对比结果
图3 训练阶段各模型的mAP50曲线对比图:
(a)全过程曲线 和 (b)第60 epoch到100 epoch的曲线
图4 紧密度-召回率曲线对比结果图
二、 基于Flask项目Web应用端搭建
1. 硬件需求
类别 | 基本要求 |
服务器端 | CPU-2核 2G 内存或2G以上;硬盘剩余空间不低于40G; |
客户端 | 4G及以上内存,硬盘空间16G及以上 |
2. 软件需求
类别 | 名 称 | 基本环境 |
服务器端 | 操作系统 | Windows10 |
数据库软件 | MySQL8.0 | |
客户端 | 操作系统 | Windows操作系统 |
3. 登录系统
本程序是本着易学、易用、模块化及共用性的原则进行设计及开发的。系统具有较好的安全保密机制,使用者在使用系统前必须先登录,用户登录时需要通过注册新用户时提供的用户名称和密码进行登录。
登录页面是程序的主入口,可以输入邮箱、登录密码,即可进入系统界面。
在登录界面输入邮箱、登录密码,点击“登录”按钮,即可进入系统界面。
如下图所示:
4. 系统界面
4.1 单图像检测
登录成功后,进入软件系统界面。
在界面的左方显示有“单目标检测”、“多目标检测”、“性能监测”、“个人信息”4个模块,点击后便进入对应的模块。
具体如下图所示:
在界面的左上方点击收起收起符号后收起(三条横线符合)菜单,
具体如下图所示:
点击“单目标检测”,进入单目标检测模块,可以上传图片和显示图片。
点击“上传图片”按钮,即可将图片上传至系统中,显示在“显示图片”模块中。点击“开始检测”按钮,即可将上传的图片进行检测处理。
具体如下图所示:
点击“Dark Mode”按钮,能够切换背景颜色,由白色切换为灰色,方便图像的查看与识别。
具体如下图所示:
4.2 多目标检测
点击界面上的“多目标检测”,进入对应的模块。在该模块内,用户可以看到全部多目标检测信息,可以上传文件、检测文件和刷新表格。
具体如下图所示:
点击“Dark Mode”按钮,能够切换背景颜色,由白色切换为灰色,方便图像的查看与识别。
具体如下图所示:
点击“上传”按钮,即可批量上传肾结石CT图像。上传后会将图片显示在屏幕下方。
效果如下图所示:
点击“检测”按钮,即可查看肾结石CT图像检测结果。结果会以图片和统计表的形式显示在界面中。
效果如下图所示:
下拉界面即可查全部检测结果,包括图片、标签、精度、检查框等信息。
效果如下图所示:
双击图片即可将图片进行放大处理,方便观察肾结石CT图像。
效果如下图所示:
4.3 性能检测
点击界面上的“性能检测”,进入对应的模块。在该模块内,用户可以看到全部任务管理信息,主要包括性能检测曲线图等信息。
如下图所示:
“性能检测”模块以折线图、饼状图、曲线图等模式展示性能数据结果,包含CPU1、CPU2、NETWORK THROUGHPUT以及Memory USAGE等数据。
效果如下图所示点击下拉界面,即可查看更多的曲线图详情信息。
效果如下图所示:
4.4 个人信息
点击界面上的“个人信息”,进入对应的模块。在该模块内,用户可以看到全部个人信息信息,主要包括个人资料、修改密码、退出登录。
如下图所示: