Research on Blood Cell Detection Algorithm Based on Improved YOLOv7

Authors

  • Yaxuan Wang
  • Cun Zhao
  • Zaichao Zhu
  • Jian Jia

DOI:

https://doi.org/10.54097/6yga3e60

Keywords:

Blood cell detection, YOLOv7, Vision Transformer, CondConv, SIoU.

Abstract

In the biomedical field, the detection of blood cells in microscopic images is crucial for assisting physicians in diagnosing blood-related diseases and plays a significant role in promoting the development of medicine towards more precise and efficient treatments. Traditional manual detection methods are time-consuming and prone to errors, and existing blood cell detection technologies face significant challenges in meeting the requirements of high precision and real-time performance. In light of this, this paper, from the perspective of image recognition and with the aid of deep learning, proposes an efficient and rapid detection model based on YOLOv7. Firstly, in order to further extract global features, this paper selects the advanced Vision Transformer module to be added to the backbone network. Then, the convolutional layer of the SPPCSPC layer in the YOLOv7 backbone network is replaced with a parameterized convolution, thus avoiding the shortcoming of traditional static convolutions where all samples share one convolution kernel. To directly add the global feature information learned in the small-scale feature layer to the maximum-scale feature layer, this paper adds an upsampling module between the minimum-scale feature layer and the maximum-scale feature layer. By using the SIoU loss function instead of the CIoU loss function, the convergence speed is further accelerated, and the precision is improved. Secondly, the experimental results validate the effectiveness of the improved YOLOv7 model. Compared to the results of YOLOv7, the mAP@0.5 value of this paper’s model has increased by 3.4% compared to the original YOLOv7, by 0.7% compared to the YOLOv5x with a larger number of parameters, and by 0.3% compared to the currently best-performing open-source model in this dataset-CST-YOLO.

Downloads

Download data is not yet available.

References

Patil P R, Sable G S, Anandgaonkar G. Counting of WBCs and RBCs from blood images using gray thresholding [J]. International Journal of Research in Engineering and Technology, 2014, 3 (4): 391-395.

Raina R, Gondhi N K, Chaahat, et al. A systematic review on acute leukemia detection using deep learning techniques [J]. Archives of Computational Methods in Engineering, 2023, 30 (1): 251-270.

Dhieb N, Ghazzai H, Besbes H, et al. An automated blood cells counting and classification framework using mask R-CNN deep learning model [C] // 2019 31st international conference on microelectronics (ICM). IEEE, 2019: 300-303.

Xia T, Fu Y Q, Jin N, et al. AI-enabled microscopic blood analysis for microfluidic COVID-19 hematology [C] // 2020 5th International Conference on Computational Intelligence and Applications (ICCIA). IEEE, 2020: 98-102.

Mao Y, Zhang H, Wu W, et al. DWS-YOLO: A Lightweight Detector for Blood Cell Detection [J]. Applied Artificial Intelligence, 2024, 38 (1): 2318673.

Shi C, Zhu D, Zhou C, et al. Gpmb-yolo: a lightweight model for efficient blood cell detection in medical imaging [J]. Health Information Science and Systems, 2024, 12 (1): 24.

Liu Z, Yuan D, Zhu G. Research on Blood Cell Detection and Counting Based on YOLO-BC Algorithm [J]. Available at SSRN 4676325, 2024.

Huang M, Wang B, Wan J, et al. Improved blood cell detection method based on YOLOv5 algorithm [C] // 2023 IEEE 6th Information Technology, Networking, Electronic and Automation Control Conference (ITNEC). IEEE, 2023, 6: 992-996.

Wang C Y, Bochkovskiy A, Liao H Y M. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors [C] // Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2023: 7464-7475.

Dosovitskiy A, Beyer L, Kolesnikov A, et al. An image is worth 16x16 words: Transformers for image recognition at scale [J]. arXiv preprint arXiv: 2010.11929, 2020.

Yang B, Bender G, Le Q V, et al. Condconv: Conditionally parameterized convolutions for efficient inference [J]. Advances in neural information processing systems, 2019, 32.

Rezatofighi H, Tsoi N, Gwak J Y, et al. Generalized intersection over union: A metric and a loss for bounding box regression [C] // Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2019: 658-666.

Zheng Z, Wang P, Liu W, et al. Distance-IoU loss: Faster and better learning for bounding box regression [C] // Proceedings of the AAAI conference on artificial intelligence. 2020, 34 (07): 12993-13000.

Gevorgyan Z. SIoU loss: More powerful learning for bounding box regression [J]. arXiv preprint arXiv: 2205.12740, 2022.

Kang M, Ting C M, Ting F F, et al. CST-YOLO: A Novel Method for Blood Cell Detection Based on Improved YOLOv7 and CNN-Swin Transformer [J]. arXiv preprint arXiv: 2306.14590, 2023.

Downloads

Published

30-06-2024

How to Cite

Wang, Y., Zhao, C., Zhu, Z., & Jia, J. (2024). Research on Blood Cell Detection Algorithm Based on Improved YOLOv7. Highlights in Science, Engineering and Technology, 105, 115-125. https://doi.org/10.54097/6yga3e60