基于YOLOv7网络的高分辨率结核杆菌识别方法研究

Research on High-resolution Mycobacterium Tuberculosis Identification Method Based on YOLOv7 Network

结核病已经被列为全球十大主要死亡原因之一,针对痰涂片图像背景复杂,结核杆菌目标尺寸小等难题,提出基于痰涂片图像的单阶段YOLOv7结核杆菌检测网络。骨干网络中,分别在不同尺度的高效聚合网络后嵌入CBAM注意力模块,以提取不同尺度的特征,在颈部网络中引入分离合并操作以改进MPConv结构,减少深层网络因卷积下采样引起的图像特征损失,并在头部网络中引进高斯分布距离来避免因边界框重叠而引起的小目标漏检,得到一种精度高、检测速度快的结核杆菌方法。实验表明,改进模型的均值平均精度达到了87.8%,相比基线网络模型提升5.1%,且优于同类算法,对推进结核杆菌的智能化检测具有重要意义。

Tuberculosis is one of the ten leading causes of death worldwide.To address the complex background and small target size of Mycobacterium tuberculosis in sputum smear images,in this paper,we put forward a YOLOv7 detection network for detecting Mycobacterium tuberculosis using sputum smear images.In the backbone network,CBAM attention modules are embedded after efficient aggregation networks at different scales to extract features at different scales.We introduce separation and merging operations in neck net-works to improve MPConv structure and reduce image feature loss due to convolutional downsampling in deep networks,and introduce Gaussian distribution distances in the head mesh to avoid small target misses due to overlapping bounding boxes.This technique is exceedingly precise and rapid,rendering it ideal for identifying Mycobacterium tuberculosis.Experiments indicate that the mean average accuracy achieved by the improved model is 87.8%,a 5.1%enhancement compared to the baseline network model,and better than similar algorithms,which is important for advancing intelligent detection of Mycobacterium tuberculosis.