面向变电设备金属锈蚀检测的分层嵌套标注方法

摘要

为解决变电设备锈蚀数据集标注过程中经常遭遇的模糊和不确定问题,提出了一种基于分层嵌套的训练样本标注方法.首先,采用较大的矩形框对锈蚀区域进行大面积标注,将视觉上连续的、相邻的或不能清晰划分的锈蚀区域用一个矩形框标注;然后,在标注框内对特征明显并具有相对独立性的区域进行二次标注,形成第2层内部嵌套标注.为了验证分层嵌套方法的有效性,与常用标记方法进行对比实验.结果表明,采用分层嵌套标注方法后,YOLOv5模型的召回率由50.79%提升至59.41%,Faster R-CNN+VGG16模型的召回率由66.50%提升至78.94%,Faster R-CNN+Res101模型的召回率由78.32%提升至84.61%.由此可见,通过分层嵌套标注可以有效提升主流模型在金属锈蚀方面的检测能力.