• Laser & Optoelectronics Progress
  • Vol. 60, Issue 6, 0617002 (2023)
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
    DOI: 10.3788/LOP213059 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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