[发明专利]一种基于改进的YOLOv7和Swin-Unet的异常细胞检测方法

说明书

本发明公开了一种基于改进的YOLOv7和Swin‑Unet的异常细胞检测方法，包括：收集病理细胞涂片图像，制作异常细胞检测数据集和分割数据集；搭建改进的YOLOv7模型并进行训练；搭建检测结果筛选模块，对检测网络输出图像中的细胞进行分类；搭建Swin‑Unet模型并进行训练，用于重叠细胞团图像的分割：以Unet模型为基础，针对多尺度下的细胞图像的局部关系与全局关系，引用Swin‑Transformer模块进行采样；使用改进的YOLOv7和Swin‑Unet模型进行异常细胞检测。本发明充分利用细胞检测时的上下文信息，有效处理难检的细胞团，可在保证检测速率的前提下大大提高准确率和召回率。

技术领域

本发明属于计算机视觉技术领域，具体涉及一种基于改进的YOLOv7和Swin-Unet的异常细胞检测方法。

背景技术

病理细胞学检查指的是通过采取细胞学的标本，比如痰脱落细胞、液基细胞等，经过涂片、病理切片制片技术等方式做成病理细胞学切片，然后通过显微镜观察细胞的种类、类型等情况，对疾病做出诊断。比如乳腺癌、宫颈癌等疾病的筛查，多是应用病理细胞学检查而进行诊断的，其中宫颈癌作为世界范围最常见的妇科恶性肿瘤，严重威胁女性的生命。2016年世界卫生组织报道：每年全球有超过50万新发的宫颈癌病例，作为发展中国家的中国约占其中的28％，癌症的早期治疗效果好、费用低、难度小，但是无明显症状且不易发现，中国以细胞学(包括传统巴氏涂片)作为宫颈癌等常见女性癌症的主要筛查手段，可是总体筛查水平不高，主要是由于国内有经验的细胞病理学医师及辅助人员稀缺，因此利用计算机对病理细胞进行辅助检测是十分有必要和有价值的。

现有技术的检测方法，主要是基于深度学习方法，包括基于目标检测的方法以及基于实例分割的方法。中国专利申请(CN202111048528.2)“一种基于引入注意力机制的异常细胞检测方法”，采用先进的目标检测网络RetinaNet对可疑细胞进行筛选，然后通过利用引入注意力机制的Mean-Teacher网络进行分类。该方法有效地实现了检测过程中的假阳性抑制，并提高了检测精度，但当检测样本噪声增加，重叠细胞团增加时，其表现不佳。主要表现为：(1)对重叠的异常细胞难以检测；(2)样本中包含组织液等非细胞单位时，检测精度明显下降；(3)引入的注意力机制没有充分结合多尺度的信息，检测性能有待提高。

发明内容

本发明的目的在于克服现有技术的缺陷，提供一种基于改进的YOLOv7和Swin-Unet的异常细胞检测方法，对难以检测的重叠异常细胞进行分阶段检测，利用目标检测对大目标性能好与实例分割的高精度的特性可有效检测难检的细胞样本，不易发生漏检与误检现象；在检测头部分采用最先进的动态头注意力机制，充分融合尺度、空间、任务三个维度的注意力，可大大提高检测精度。

为解决上述技术问题，本发明采用以下技术方案。

一种基于改进的YOLOv7和Swin-Unet的异常细胞检测方法，包括以下步骤：

步骤1、收集病理细胞学检查中的细胞涂片图像，制作异常细胞检测数据集和异常细胞分割数据集；

步骤2、搭建改进的YOLOv7模型并进行训练，用于异常细胞与重叠细胞团的检测，以最新的YOLOv7模型为基础，针对多维度特征信息的提取问题进行改进，得到改进的YOLOv7模型结构，包含异常细胞检测数据预处理模块Process、主干网络Backbone、颈部网络Neck和检测网络Head；

步骤3、搭建一个检测结果筛选模块，对检测网络输出图像中的细胞进行分类，异常细胞图像作为输出，重叠细胞团图像作为分割模型的输入；

步骤4、搭建Swin-Unet模型并进行训练，用于重叠细胞团图像的分割：以医学领域最常用的Unet模型为基础，针对多尺度下的细胞图像的局部关系与全局关系，引用Swin-Transformer模块进行采样；Swin-Unet模型结构包含编码器Encoder、颈部网络Neck、解码器Decoder；