基于轻量化YoloV5s的血癌细胞检测
Detection of Blood Cancer Cells Based on Lightweight YoloV5s
摘要: 为了解决人工检测血癌细胞耗费人力,且容易出现漏检、误检的情况,提出一种新型的检测血癌细胞检测算法——YoloV5s-FasterNet。该算法在原本的YoloV5s框架上,通过将主干网络中的C3检测层改进为FasterNet轻量化网络用来提取特征信息,降低整体网络模型的复杂程度,并在主干网络中增加坐标注意力(coordinate attention, CA)机制模型更好的定位和识别目标信息来提高检测精度。通过在血癌数据集上进行大量实验结果表明:与原模型相比,改进的模型平均精度提升2.2%,参数量减少25.7%,检测速度提高15%,实验结果良好,证明该算法对血癌细胞检测具有实用性。
Abstract: In order to solve the problem that manual detection of blood cancer cells is labor-intensive, and it is easy to miss and misdetect, a new detection algorithm for blood cancer cells is proposed- YoloV5s-FasterNet. Based on the original YoloV5s framework, this algorithm improves the C3 detection layer in the backbone network into a FasterNet lightweight network to extract feature information, reduce the complexity of the overall network model, and add coordinate attention to the backbone network. This model can better locate and identify target information to improve detection accuracy. A large number of experimental results on the blood cancer data set show that: compared with the original model, the average accuracy of the improved model is increased by 1.2%, the number of parameters is reduced by 25.7%, and the detection speed is increased by 15%. The experimental results are good, proving that the algorithm is effective in detecting blood cancer cells.
文章引用:乔冬, 何利文. 基于轻量化YoloV5s的血癌细胞检测[J]. 软件工程与应用, 2024, 13(2): 223-233. https://doi.org/10.12677/sea.2024.132023

参考文献

[1] Nazlibilek, S., Karacor, D., Ercan, T., et al. (2014) Automatic Segmentation, Counting, Size Determination and Classification of White Blood Cells. Measurement, 55, 58-65. [Google Scholar] [CrossRef
[2] Suryani, E., Palgunadi, S. and Pradana, T.P.N. (2017) Classification of Acute Myelogenous Leukemia (AML M2 and AML M3) Using Momentum Back Propagation from Watershed Distance Transform Segmented Images. Journal of Physics: Conference Series. Iop Publishing, 801, 012044. [Google Scholar] [CrossRef
[3] Shahin, A.I., Guo, Y., Amin, K.M., et al. (2019) White Blood Cells Identification System Based on Convolutional Deep Neural Learning Networks. Computer Methods and Programs in Biomedicine, 168, 69-80. [Google Scholar] [CrossRef] [PubMed]
[4] Liang, G.B., Hong, H.C., Xie, W.F., et al. (2018) Combining Convolutional Neural Network with Recursive Neural Network for Blood Cell Image Classification. IEEE Access, 6, 36188-36197. [Google Scholar] [CrossRef
[5] Zhang, D.D., Zhang, P.F. and Wang, L.S. (2019) Cell Counting Algorithm Based on YOLOv3 and Image Density Estimation. 2019 IEEE 4th International Conference on Signal and Image Processing, Wuxi, 19-21 July 2019, 920-924. [Google Scholar] [CrossRef
[6] Girshick, R., Donahue, J., Darrell, T., et al. (2014) Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation. 2014 IEEE Conference on Computer Vision and Pattern Recognition, Columbus, 23-28 June 2014, 580-587. [Google Scholar] [CrossRef
[7] Redmon, J., Divvala, S., Girshick, R., et al. (2016) You Only Look Once: Unified, Real-Time Object Detection. 2016 IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, 27-30 June 2016, 779-788. [Google Scholar] [CrossRef
[8] Redmon, J. and Farhadi, A. (2017) YOLO9000: Better, Faster, Stronger. 2017 IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, 21-26 July 2017, 6517-6525. [Google Scholar] [CrossRef
[9] Liu, W., Anguelov, D., Erhan, D., et al. (2016) SSD: Single Shot Multibox Detector. In: Leibe, B., Matas, J., Sebe, N. and Welling, M., Eds., Computer Vision-ECCV 2016, Lecture Notes in Computer Science, Vol. 9905, Springer, Cham, 21-37. [Google Scholar] [CrossRef
[10] Hou, Q.B., Zhou, D.Q. and Feng, J.S. (2021) Coordinate Attention for Efficient Mobile Network Design. 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, 20-25 June 2021, 13713-13722.
[11] Chen, J., Kao, S., He, H., et al. (2023) Run, Don’t Walk: Chasing Higher FLOPS for Faster Neural Networks. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, 17-24 June 2023, 12021-12031. [Google Scholar] [CrossRef
[12] Hu, J., Shen, L., Albanie, S., et al. (2018) Squeeze-and-Excitation Networks. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-23 June 2018, 7132-7141. [Google Scholar] [CrossRef
[13] Jaderberg, M., Simonyan, K. and Zisserman, A. (2015) Spatial Transformer Networks. Proceedings of the 28th International Conference on Neural Information Processing Systems, Montreal, 2017-2025.
[14] Woo, S., Park, J., Lee, J.Y., et al. (2018) CBAM: Convolutional Block Attention Module. In: Ferrari, V., Hebert, M., Sminchisescu, C. and Weiss, Y., Eds., Computer VisionECCV 2018. Lecture Notes in Computer Science, Vol. 11211, Springer, Cham, 3-19. [Google Scholar] [CrossRef
[15] Ren, S.Q., He, K.M., Girshick, R.B., et al. (2017) Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 1137-1149. [Google Scholar] [CrossRef