Detection of whole body bone fractures based on improved YOLOv7

被引:0
|
作者
Zou, Junting [1 ]
Arshad, Mohd Rizal [1 ]
机构
[1] Univ Sains Malaysia USM, Sch Elect & Elect Engn, Engn Campus, Nibong Tebal 14300, Pulau Pinang, Malaysia
关键词
Deep learning; Bone fracture; Detection; YOLOv7; CLASSIFICATION; INTELLIGENCE; FUTURE;
D O I
10.1016/j.bspc.2024.105995
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
In the field of medical imaging, early and accurate detection of bone fractures can significantly improve the treatment effect of patients. In this study, we conduct a comparative study of one -stage and two -stage deep learning architectures, with a particular focus on their ability to autonomously and accurately localize four different fracture morphologies in the whole body: angle fractures, normal fractures, line fractures, and messedup angle fractures. Using well -annotated datasets, we explore the capabilities of the frontier models, with a special focus on the YOLO variants (v4, v5, v7, v8 and our improved v7 model), SSD, Faster-RCNN, and MaskRCNN. To further improve the detection accuracy, we introduce an Enhanced Intersection of Unions (EIoU) loss function to refine the positional differences between the predicted bounding box and the ground truth bounding box. We measure the performance of the models by precision, recall, mAP, and IoU metrics. Our analysis illuminates the strengths and limitations of each model for bone fracture detection and highlights the advances made by integrating the attention mechanism into YOLOv7. Most notably, our customized YOLOv7-ATT model incorporating the attention mechanism significantly outperforms the baseline metrics of the pre -trained model, achieving a mAP of 80.2%. It exhibits excellent generalization on the FracAtlas dataset, achieving a mAP of 86.2%, which is significantly better than the other models. This study provides researchers with a foundational resource aimed at optimizing and deploying deep learning models for fracture detection in clinical settings.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] Mask wearing detection based on improved YOLOv7
    Fu Hui-chen
    Gao Jun-wei
    Che Lu-yang
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2023, 38 (08) : 1139 - 1147
  • [2] Helmet Detection Algorithm Based on Improved YOLOv7
    Yajie Yaermaimaiti Yilihamu
    Lingfei Liu
    Ruohao Xi
    undefined Wang
    Automatic Control and Computer Sciences, 2024, 58 (6) : 642 - 655
  • [3] Underwater Target Detection Based on Improved YOLOv7
    Liu, Kaiyue
    Sun, Qi
    Sun, Daming
    Peng, Lin
    Yang, Mengduo
    Wang, Nizhuan
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (03)
  • [4] Ship Detection and Recognition Based on Improved YOLOv7
    Wu, Wei
    Li, Xiulai
    Hu, Zhuhua
    Liu, Xiaozhang
    CMC-COMPUTERS MATERIALS & CONTINUA, 2023, 76 (01): : 489 - 498
  • [5] Underwater Target Detection Based on Improved YOLOv7
    Fu, Junshang
    Tian, Ying
    IAENG International Journal of Computer Science, 2024, 51 (04) : 422 - 429
  • [6] A Flame Detection Algorithm Based on Improved YOLOv7
    Yan, Guibao
    Guo, Jialin
    Zhu, Dongyi
    Zhang, Shuming
    Xing, Rui
    Xiao, Zhangshu
    Wang, Qichao
    APPLIED SCIENCES-BASEL, 2023, 13 (16):
  • [7] Driver fatigue detection based on improved YOLOv7
    Li, Xianguo
    Li, Xueyan
    Shen, Zhenqian
    Qian, Guangmin
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (03)
  • [8] Efficient Object Detection and Recognition of Body Welding Studs Based on Improved YOLOv7
    Huang, Hong
    Peng, Xiangqian
    Hu, Xiaoping
    Ou, Wenchu
    IEEE ACCESS, 2024, 12 : 41531 - 41541
  • [9] Night target detection algorithm based on improved YOLOv7
    Bowen, Zheng
    Huacai, Lu
    Shengbo, Zhu
    Xinqiang, Chen
    Hongwei, Xing
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [10] Mask wearing detection algorithm based on improved YOLOv7
    Luo, Fang
    Zhang, Yin
    Xu, Lunhui
    Zhang, Zhiliang
    Li, Ming
    Zhang, Weixiong
    MEASUREMENT & CONTROL, 2024, 57 (06): : 751 - 762