-
公开(公告)号:CN118470296A
公开(公告)日:2024-08-09
申请号:CN202410435196.0
申请日:2024-04-11
IPC分类号: G06V10/25 , G06V10/44 , G06V10/774 , G06V10/80 , G06V10/82 , G06N3/0464
摘要: 本发明提供一种改进YOLOv5的地下排水管道缺陷自动判读方法,包括:S1、采用CCTV检测器获取管道缺陷视频,从管道缺陷视频中找出管道缺陷的图像;对图像进行数据标注,制作数据集;S2、对数据集进行划分,得到训练数据集、验证数据集和测试数据集;S3、在原始YOLOv5颈部网络的预测层引入重写后的ECBAM注意力机制,在YOLOv5主干网络的部分C3模块中引入可变空洞卷积SAC模块,再将算法损失函数CIOU替换成SIOU,得到改进的YOLOv5算法;S4、将训练数据集放入到改进的YOLOv5模型中进行训练,训练得到城市地下排水管道缺陷检测模型;S5、将上述模型用于真实工况下地下排水管道的缺陷检测。本发明对地下排水管道缺陷识别精度具有显著的提升效果,可用于真实工况下地下排水管道缺陷识别。