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 条
  • [21] MCA-YOLOv7: An Improved UAV Target Detection Algorithm Based on YOLOv7
    Qin, Zhiyong
    Chen, Dike
    Wang, Hongyuan
    IEEE ACCESS, 2024, 12 : 42642 - 42650
  • [22] YOLOv7-SN: Underwater Target Detection Algorithm Based on Improved YOLOv7
    Zhao, Ming
    Zhou, Huibo
    Li, Xue
    SYMMETRY-BASEL, 2024, 16 (05):
  • [23] YOLOv7-SiamFF: Industrial defect detection algorithm based on improved YOLOv7
    Yi, Feifan
    Zhang, Haigang
    Yang, Jinfeng
    He, Liming
    Mohamed, Ahmad Sufril Azlan
    Gao, Shan
    COMPUTERS & ELECTRICAL ENGINEERING, 2024, 114
  • [24] Tea Buds Detection in Complex Background Based on Improved YOLOv7
    Meng, Junquan
    Kang, Feng
    Wang, Yaxiong
    Tong, Siyuan
    Zhang, Chenxi
    Chen, Chongchong
    IEEE ACCESS, 2023, 11 : 88295 - 88304
  • [25] Improved YOLOv7 Underwater Object Detection Based on Attention Mechanism
    Fu, Junshang
    Tian, Ying
    ENGINEERING LETTERS, 2024, 32 (07) : 1377 - 1384
  • [26] A detection method for dead caged hens based on improved YOLOv7
    Yang, Jikang
    Zhang, Tiemin
    Fang, Cheng
    Zheng, Haikun
    Ma, Chuang
    Wu, Zhenlong
    Computers and Electronics in Agriculture, 2024, 226
  • [27] An Apricot Detection Algorithm in Complex Environments Based on Improved YOLOv7
    Guo, Qiang
    Ma, Chi
    Hu, Hui
    IAENG International Journal of Computer Science, 2024, 51 (12) : 2135 - 2144
  • [28] Pedestrian Detection Method in Infrared Image Based on Improved YOLOv7
    Liu, Zhengyan
    Dai, Chaoyue
    Li, Xu
    Proceedings of 2023 IEEE 3rd International Conference on Information Technology, Big Data and Artificial Intelligence, ICIBA 2023, 2023, : 946 - 954
  • [29] Detection Algorithm of Laboratory Personnel Irregularities Based on Improved YOLOv7
    Yang, Yongliang
    Xu, Linghua
    Luo, Maolin
    Wang, Xiao
    Cao, Min
    CMC-COMPUTERS MATERIALS & CONTINUA, 2024, 78 (02): : 2741 - 2765
  • [30] Detection of Famous Tea Buds Based on Improved YOLOv7 Network
    Wang, Yongwei
    Xiao, Maohua
    Wang, Shu
    Jiang, Qing
    Wang, Xiaochan
    Zhang, Yongnian
    AGRICULTURE-BASEL, 2023, 13 (06):