Laser & Optoelectronics Progress, Volume. 60, Issue 6, 0617002(2023)

Improved YOLOv4 Model-Based Spinal Magnetic Resonance Imaging Image Detection

Ning Dai1、*, Yuhai Gu1, Zhicheng Zhang2, Yang Zhang2, and Zhan Xu1
Author Affiliations
  • 1Key Laboratory of Modern Measurement and Control Ministry of Education, Beijing Information Science and Technology University, Beijing 100192, China
  • 2Department of Orthopedics, PLA General Hospital, Beijing 100700, China
  • show less

    Aiming at the complex anatomical structure of the spine, a YOLOv4-disc algorithm for spinal magnetic resonance imaging image detection is proposed. First, aiming at the problem of small number of real case samples, the adaptive histogram equalization (CLAHE) data enhancement method with limited contrast is used to improve the generalization ability of the model. Second, K-means algorithm is used to cluster the size of real frames in the dataset to obtain the appropriate anchor frame size and determine the number of anchor frames. After that, depth separable convolution is used in CSPDarknet-53 backbone feature extraction network instead of ordinary convolution to reduce network parameters and reduce computation. Finally, the loss function of the native network is improved based on Focal loss to solve the problem that the proportion of positive and negative samples is seriously unbalanced in one-stage target detection. The experimental results show that the mean average precision (mAP) of the proposed YOLOv4-disc algorithm reaches 90.80%, which is 3.51 percentage points higher than that of the native YOLOv4 algorithm.

    Tools

    Get Citation

    Copy Citation Text

    Ning Dai, Yuhai Gu, Zhicheng Zhang, Yang Zhang, Zhan Xu. Improved YOLOv4 Model-Based Spinal Magnetic Resonance Imaging Image Detection[J]. Laser & Optoelectronics Progress, 2023, 60(6): 0617002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Medical Optics and Biotechnology

    Received: Nov. 25, 2021

    Accepted: Jan. 11, 2022

    Published Online: Mar. 16, 2023

    The Author Email: Dai Ning (18501301242@163.com)

    DOI:10.3788/LOP213059

    Topics