Detection of whole body bone fractures based on improved YOLOv7

被引:4
作者
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
相关论文
共 59 条
  • [1] FracAtlas: A Dataset for Fracture Classification, Localization and Segmentation of Musculoskeletal Radiographs
    Abedeen, Iftekharul
    Rahman, Md. Ashiqur
    Prottyasha, Fatema Zohra
    Ahmed, Tasnim
    Chowdhury, Tareque Mohmud
    Shatabda, Swakkhar
    [J]. SCIENTIFIC DATA, 2023, 10 (01)
  • [2] COMPUTER VISION
    ALOIMONOS, Y
    ROSENFELD, A
    [J]. SCIENCE, 1991, 253 (5025) : 1249 - 1259
  • [3] Medical Image Analysis using Convolutional Neural Networks: A Review
    Anwar, Syed Muhammad
    Majid, Muhammad
    Qayyum, Adnan
    Awais, Muhammad
    Alnowami, Majdi
    Khan, Muhammad Khurram
    [J]. JOURNAL OF MEDICAL SYSTEMS, 2018, 42 (11)
  • [4] Medical Imaging: From Roentgen to the Digital Revolution, and Beyond
    Bercovich, Eyal
    Javitt, Marcia C.
    [J]. RAMBAM MAIMONIDES MEDICAL JOURNAL, 2018, 9 (04):
  • [5] Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934]
  • [6] A Learning Health Care System Using Computer-Aided Diagnosis
    Cahan, Amos
    Cimino, James J.
    [J]. JOURNAL OF MEDICAL INTERNET RESEARCH, 2017, 19 (03)
  • [7] Towards Accurate One-Stage Object Detection with AP-Loss
    Chen, Kean
    Li, Jianguo
    Lin, Weiyao
    See, John
    Wang, Ji
    Duan, Lingyu
    Chen, Zhibo
    He, Changwei
    Zou, Junni
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 5114 - 5122
  • [8] Shaping the future through innovations: From medical imaging to precision medicine
    Comaniciu, Dorin
    Engel, Klaus
    Georgescu, Bogdan
    Mansi, Tommaso
    [J]. MEDICAL IMAGE ANALYSIS, 2016, 33 : 19 - 26
  • [9] Deng L, 2013, INT CONF ACOUST SPEE, P8604, DOI 10.1109/ICASSP.2013.6639345
  • [10] LONGITUDINAL STRESS FRACTURES - ANOTHER VARIETY SEEN IN LONG BONES
    DEVAS, MB
    [J]. JOURNAL OF BONE AND JOINT SURGERY-BRITISH VOLUME, 1960, 42 (03): : 508 - 514