高级检索

基于RV1126B端侧部署的尺度自适应NWD-STAL-YOLO26配网线路隐患检测方法

Transmission-Line Hazard Detection Using Scale-Adaptive NWD-STAL-YOLO26 for RV1126B Edge Deployment

  • 摘要: 本文针对配网线路监拍图像中小目标占比高,基于交并比的候选匹配对微小位置偏差敏感,以及YOLO26s在瑞芯微RV1126B平台上部署适配困难等问题,提出一种面向端侧部署的尺度自适应NWD-STAL YOLO26s检测方法。首先,构建包含11000幅图像的配网线路隐患目标数据集,覆盖施工机械、烟火及导线异物等多类隐患目标。其次,在训练阶段的正样本分配分支中,根据真实框面积自适应融合完全交并比与归一化Wasserstein距离:对于小尺度目标,增大归一化Wasserstein距离的权重,以减小候选排序对微小位置偏移的敏感性;对于较大目标,则采用完全交并比进行匹配。在端侧部署方面,完成多尺度原始输出图构建、INT8量化、无DFL边界框解码、分类logit阈值映射以及RGA图像预处理。实验结果表明,与YOLO26s基线相比,改进模型的召回率和mAP@0.5分别提高1.75%和0.96%,转换后的RKNN模型已在RV1126B平台完成4K图像功能验证。上述结果表明,所提方法在不增加推理结构开销的条件下,能够兼顾配网线路隐患检测性能与端侧部署需求。

     

    Abstract: This paper addresses the high proportion of small targets in distribution-line monitoring images, the sensitivity of IoU-based candidate matching to slight localization deviations, and the difficulties in adapting YOLO26s for deployment on the RV1126B platform. A scale-adaptive NWD-STAL YOLO26s detection method for edge deployment is proposed. First, a distribution-line hazard dataset containing 11,000 images is constructed, covering multiple categories of hazardous targets, including construction machinery, smoke and fire, and foreign objects on conductors. Second, in the positive-sample assignment branch during training, Complete Intersection over Union (CIoU) and normalized Wasserstein distance (NWD) are adaptively fused according to the ground-truth box area. For small-scale targets, the weight of NWD is increased to reduce the sensitivity of candidate ranking to slight localization deviations, whereas CIoU is used for matching larger targets. For edge deployment, a multi-scale raw-output graph is constructed, and INT8 quantization, DFL-free bounding-box decoding, classification-logit threshold mapping, and RGA-based image preprocessing are implemented. Experimental results show that, compared with the YOLO26s baseline, the proposed model improves recall and mAP@0.5 by 1.75% and 0.96%, respectively. The converted RKNN model has also been functionally verified on 4K images using the RV1126B platform. The results demonstrate that the proposed method achieves a balance between distribution-line hazard detection performance and edge-deployment requirements without introducing additional inference-structure overhead.

     

/

返回文章
返回