SDE-YOLO: a novel method for blood cell detection

Yonglin Wu, Dongxu Gao, Yinfeng Fang, Xue Xu, Hongwei Gao, Zhaojie Ju

Research output: Contribution to journalArticlepeer-review

4 Downloads (Pure)

Abstract

This paper proposes an improved target detection algorithm, SDE-YOLO, based on the YOLOv5s framework, to address the low detection accuracy, misdetection, and leakage in blood cell detection caused by existing single-stage and two-stage detection algorithms. Initially, the Swin Transformer is integrated into the back-end of the backbone to extract the features in a better way. Then, the 32 × 32 network layer in the path-aggregation network (PANet) is removed to decrease the number of parameters in the network while increasing its accuracy in detecting small targets. Moreover, PANet substitutes traditional convolution with depth-separable convolution to accurately recognize small targets while maintaining a fast speed. Finally, replacing the complete intersection over union (CIOU) loss function with the Euclidean intersection over union (EIOU) loss function can help address the imbalance of positive and negative samples and speed up the convergence rate. The SDE-YOLO algorithm achieves a mAP of 99.5 95.3 and 93.3 red blood cells, and platelets, respectively, which is an improvement over other single-stage and two-stage algorithms such as SSD, YOLOv4, and YOLOv5s. The experiment yields excellent results, and the algorithm detects blood cells very well. The SDE-YOLO algorithm also has advantages in accuracy and real-time blood cell detection performance compared to the YOLOv7 and YOLOv8 technologies.
Original languageEnglish
Article number404
Number of pages15
JournalBiomimetics
Volume8
Issue number5
DOIs
Publication statusPublished - 1 Sept 2023

Keywords

  • blood cell testing
  • Swin Transformer
  • PAN
  • depth-separable convolution
  • EIOU

Cite this