针对宫颈异常细胞检测的 SER-DC YOLO  

SER-DC YOLO for the Detection of Abnormal Cervical Cells

作　　者：李超炜 杨晓娜 赵司琦 何勇军[2] LI Chaowei;YANG Xiaona;ZHAO Siqi;HE Yongjun(School of Computer Science and Technology,Harbin University of Science and Technology,Harbin 150080,China;School of Computer Science and Technology,Harbin Institute of Technology,Harbin 150006,China)

机构地区：[1]哈尔滨理工大学计算机科学与技术学院,哈尔滨150080 [2]哈尔滨工业大学计算机科学与技术学院,哈尔滨150006

出　　处：《哈尔滨理工大学学报》2024年第1期115-123,共9页Journal of Harbin University of Science and Technology

基　　金：国家自然科学基金面上项目(61673142);黑龙江省自然科学基金杰出青年项目(JJ2019JQ0013);哈尔滨市杰出青年人才基金(2017RAYXJ013);哈尔滨理工大学杰出青年人才项目(20200203).

摘　　要：由于宫颈细胞样本的液基薄层细胞学检测(thin prep cytologic test,TCT)图像内容复杂,背景颜色丰富多样,而且不同女性的宫颈细胞具有一定程度的天然差异,这给宫颈异常细胞的检测带来了很大的困难。为解决这一难题,提出了一种名为基于特征压缩与激发和可变形卷积(SE-ResNet-deformable convolution you only look once,SER-DC YOLO)的目标检测网络。该网络在YOLOv5的Backbone中融合注意力机制,添加了SE-ResNet模块,然后改进了SPP层的网络结构,并且使用可变形卷积来替换普通卷积,最后修改了边界框的损失计算函数,将广义交并比(generalized intersection over union,GIoU)改为α-IOU Loss。实验表明,该网络与YOLOv5网络相比,在宫颈图片数据集上召回率提高了19.94%,精度提高了3.52%,平均精度均值提高了7.19%。相关代码链接:https://github.com/sleepLion99/SER-DC_YOLO。Due to the complex content of Thin Prep Cytology Test(TCT)images of cervical cell samples with rich and diverse background colors and a certain degree of natural variation of cervical cells among different women,this poses a great difficulty in the detection of abnormal cervical cells.To solve this challenge,a target detection network called SE-ResNet-Deformable Convolution You Only Look Once(SER-DC YOLO)is proposed.The network incorporates the attention mechanism in YOLOv5s Backbone,adds the SE-ResNet module,then improves the network structure of the SPP layer and replaces the normal convolution with deformable convolution,and finally modifies the loss calculation function of the bounding box by replacing the Generalized Intersection over Union(GIoU)toα-IOU Loss.Experiments show that the network improves recall by 19.94%,precision by 3.52%,and average precision by 7.19%on the cervical image dataset compared with the YOLOv5 network.Link to related code:https://github.com/sleepLion99/SER-DC_YOLO.

关 键 词：SER-DC YOLO YOLOv5 目标检测 注意力机制 可变形卷积 

分 类 号：TP315.69[自动化与计算机技术—计算机软件与理论]