摘要
本发明公开了一种基于端到端关键点检测的玻璃绝缘子片定位方法,包括:1)电力巡检玻璃绝缘子实例分割数据集的构建与标注;2)利用数据增强算法进行数据扩充;3)训练得到实例分割模型并将玻璃绝缘子所在区域的最小外接多边形图像切割下来,作为下一步关键点检测的数据集;4)标注关键点检测数据集并做数据扩充;5)设计一个端到端的关键点检测模型,不断调优训练;6)将训练好的实例分割模型和关键点检测模型串联起来工作:将待检测的玻璃绝缘子图片输入训练好的实例分割模型,将分割结果的区域切割下来输入到训练好的关键点检测模型,得到图中各玻璃绝缘子片所在位置关键点的坐标值。本发明可提高玻璃绝缘子片定位的速度和精确度。
- 单位