基于改进YOLOv5的眼底出血点检测算法

作者：吕辉; 吕卫峰

来源：广西师范大学学报(自然科学版), 2023, 1-10.

DOI：10.16088/j.issn.1001-6600.2023053101

• 眼底图像
• 糖尿病视网膜病变
• 目标检测
• 出血病灶
• 特征融合

糖尿病视网膜病变眼底图像中出血点病灶尺寸小且分布密集，导致现有算法难以实现该病灶的精确检测及定位，故本文提出一种RCA-YOLO出血病灶检测算法。该算法在YOLOv5s基础上，首先在主干网络中使用RCA-Net模块，使网络能够在获得各通道间联系的同时保留目标病灶的位置信息，增强网络对出血区域的特征提取及定位能力；在特征融合阶段采用轻量化特征金字塔网络Tiny-BiFPN，在减少网络参数量的同时，实现高效率的多尺度特征融合；最后提出小目标特征增强模块，提升算法对小出血点病灶的检测精度。实验结果表明，改进后的RCA-YOLO算法能够准确地检测并定位出血点病灶，平均检测准确率(mAP)可达79.3%，较YOLOv5s算法提高了9.5个百分点，其检测结果同样优于Faster R-CNN、YOLOv6s、YOLOv7和YOLOv8s等主流算法。

• 出版日期2023
• 单位河南理工大学; 自动化学院