4 个月前

使用特征上下文激励模块检测儿童腕部骨折的X射线图像

使用特征上下文激励模块检测儿童腕部骨折的X射线图像

摘要

儿童在日常生活中经常遭受腕部创伤,通常需要放射科医生分析和解读X光图像,然后由外科医生进行手术治疗。深度学习的发展使得神经网络可以作为计算机辅助诊断(CAD)工具,帮助医生和专家进行医学影像诊断。由于YOLOv8模型在目标检测任务中取得了令人满意的效果,因此已被应用于各种骨折检测。本研究介绍了四种Feature Contexts Excitation-YOLOv8(FCE-YOLOv8)模型的变体,每种变体分别集成了不同的FCE模块(即Squeeze-and-Excitation(SE)、Global Context(GC)、Gather-Excite(GE)和Gaussian Context Transformer(GCT)模块),以提升模型性能。实验结果表明,在GRAZPEDWRI-DX数据集上,我们提出的YOLOv8+GC-M3模型将mAP@50值从65.78%提高到66.32%,优于当前最先进的(SOTA)模型,并且减少了推理时间。此外,我们提出的YOLOv8+SE-M3模型达到了最高的mAP@50值67.07%,超过了SOTA性能。本研究的实现代码已发布在 https://github.com/RuiyangJu/FCE-YOLOv8。

代码仓库

ruiyangju/fce-yolov8
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
fracture-detection-on-grazpedwri-dxYOLOv8+SE
AP50: 67.07
F1-score: 0.66
fracture-detection-on-grazpedwri-dxYOLOv8+GCT
AP50: 65.67
F1-score: 0.64
fracture-detection-on-grazpedwri-dxYOLOv8+GC
AP50: 66.32
F1-score: 0.66
fracture-detection-on-grazpedwri-dxYOLOv8+GE
AP50: 65.99
F1-score: 0.64

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用特征上下文激励模块检测儿童腕部骨折的X射线图像 | 论文 | HyperAI超神经