-
题名改进的YOLOv5积水检测算法
- 1
-
-
作者
杨庆江
信威
马仲甜
李钊枢
吕丛生
赵斌
-
机构
黑龙江科技大学
国网黑龙江省电力有限公司黑河供电公司
-
出处
《黑龙江科技大学学报》
CAS
2024年第2期297-301,共5页
-
基金
黑龙江省“揭榜挂帅”科技攻关项目(2021ZXJ02A02)
国家电网科学技术项目(SGHLHH00SZJS2310193)。
-
文摘
为了提升YOLOv5算法的积水检测精准度,提出一种改进YOLOv5网络结构的积水检测算法。通过解耦头代替初始YOLOv5的耦合头,使分类任务与回归任务两者互不干扰,在网络训练到300轮次左右时表现出了良好的收敛效果;利用CIoU损失函数代替GIoU,降低积水检测的误差,提高积水检测精准度。结果表明,相比初始YOLOv5算法的精准度、召回率、m_(AP1)和m_(AP2)分别提升了1.7%、1.8%、1%和0.5%。文中的积水检测算法比初始算法减少了训练损失,收敛速度更快,精准度更高,可为积水区域的检测提供算法支撑。
-
关键词
积水检测
YOLOv5
解耦头
CIoU
-
Keywords
water accumulation detection
YOLOv5
decoupling head
CIoU
-
分类号
TP18
[自动化与计算机技术—控制理论与控制工程]
TP391.41
[自动化与计算机技术—控制科学与工程]
-
