An Improved Few-Shot Object Detection via Feature Reweighting Method for Insulator Identification

被引:4
|
作者
Wu, Junpeng [1 ,2 ]
Zhou, Yibo [2 ]
机构
[1] Northeast Elect Power Univ, Key Lab Modern Power Syst Simulat & Control & Rene, Minist Educ, Jilin 132012, Peoples R China
[2] Northeast Elect Power Univ, Sch Elect Engn, Jilin 132012, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 10期
关键词
insulator; few-shot object detection; feature extraction network; attention mechanism; non-maximum suppression algorithm;
D O I
10.3390/app13106301
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
To address the issue of low accuracy in insulator object detection within power systems due to a scarcity of image sample data, this paper proposes a method for identifying insulator objects based on improved few-shot object detection through feature reweighting. The approach utilizes a meta-feature transfer model in conjunction with the improved YOLOv5 network to realize insulator recognition under conditions of few-shot. Firstly, the feature extraction module of the model incorporates an improved self-calibrated feature extraction network to extract feature information from multi-scale insulators. Secondly, the reweighting module integrates the SKNet attention mechanism to facilitate precise segmentation of the mask. Finally, the multi-stage non-maximum suppression algorithm is designed in the prediction layer, and the penalty function about confidence is set. The results of multiple prediction boxes are retained to reduce the occurrence of false detection and missing detection. For the poor detection results due to a low diversity of sample space, the transfer learning strategy is applied in the training to transfer the entire trained model to the detection of insulator targets. The experimental results show that the insulator detection mAP reaches 29.6%, 36.0%, and 48.3% at 5-shot, 10-shot, and 30-shot settings, respectively. These findings serve as evidence of improved accuracy levels of the insulator image detection under the condition of few shots. Furthermore, the proposed method enables the recognition of insulators under challenging conditions such as defects, occlusion, and other special circumstances.
引用
收藏
页数:18
相关论文
共 50 条
  • [21] Few-Shot Video Object Detection
    Fan, Qi
    Tang, Chi-Keung
    Tai, Yu-Wing
    COMPUTER VISION, ECCV 2022, PT XX, 2022, 13680 : 76 - 98
  • [22] Incremental few-shot object detection via knowledge transfer
    Feng, Hangtao
    Zhang, Lu
    Yang, Xu
    Liu, Zhiyong
    PATTERN RECOGNITION LETTERS, 2022, 156 : 67 - 73
  • [23] Few-shot object detection via message transfer mechanism
    Lv, Wen
    Shi, Hongbo
    Tan, Shuai
    Song, Bing
    Tao, Yang
    JOURNAL OF ELECTRONIC IMAGING, 2024, 33 (02)
  • [24] Few-Shot Object Detection of drones
    Zou Weibao
    Liu Xindi
    Yang Jitao
    Qu Wei
    INTERNATIONAL CONFERENCE ON ELECTRICAL, COMPUTER AND ENERGY TECHNOLOGIES (ICECET 2021), 2021, : 1030 - 1034
  • [25] Efficient Few-Shot Object Detection via Knowledge Inheritance
    Yang, Ze
    Zhang, Chi
    Li, Ruibo
    Xu, Yi
    Lin, Guosheng
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2023, 32 : 321 - 334
  • [26] Few-Shot Object Detection via Understanding Convolution and Attention
    Tong, Jiaxing
    Chen, Tao
    Wang, Qiong
    Yao, Yazhou
    PATTERN RECOGNITION AND COMPUTER VISION, PT I, PRCV 2022, 2022, 13534 : 674 - 687
  • [27] Feature reconstruction and metric based network for few-shot object detection
    Li, Yuewen
    Feng, Wenquan
    Lyu, Shuchang
    Zhao, Qi
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2023, 227
  • [28] GFENet: Generalization Feature Extraction Network for Few-Shot Object Detection
    Ke, Xiao
    Chen, Qiuqin
    Liu, Hao
    Guo, Wenzhong
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (12) : 12741 - 12755
  • [29] Multi-Stage Feature Redistribution for Few-Shot Object Detection
    Liu L.
    He Z.
    Ma Z.
    Liu B.
    Wang L.
    Dianzi Keji Daxue Xuebao/Journal of the University of Electronic Science and Technology of China, 2023, 52 (01): : 116 - 124
  • [30] Few-Shot Object Detection via Disentangling Class-Related Factors in Feature Distribution
    Wei, Lili
    Tang, Xiaofen
    Dang, Jin
    PATTERN RECOGNITION AND COMPUTER VISION, PT XIII, PRCV 2024, 2025, 15043 : 77 - 92