Detection and classification of breast lesions with You Only Look Once version 5

Aim: To explore the ability of You Only Look Once version 5 (YOLOv5) to detect and classify breast lesions on dynamic contrast-enhanced MRI. Methods: Four YOLOv5 submodels were examined. A total of 2124 and 2226 images of benign and malignant lesions were obtained, respectively. Precision, recall rate and mean average precision were used to evaluate model performance. Results: The precision (0.916) and mean average precision _0.5 (0.894) of YOLOv5s were higher than those of YOLOv5m (0.832, 0.794), YOLOv5l (0.843, 0.803) and YOLOv5x (0.854, 0.821). In the validation set, YOLOv5s required 1.1 ms to detect lesions per image. Conclusion: YOLOv5s was the fastest and had the highest precision among the four YOLOv5 submodels for the detection and classification of breast lesions on dynamic contrast-enhanced MRI. It has a greater clinical application value.