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

被引:3
|
作者
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 条
  • [1] Few-shot Object Detection via Feature Reweighting
    Kang, Bingyi
    Liu, Zhuang
    Wang, Xin
    Yu, Fisher
    Feng, Jiashi
    Darrell, Trevor
    2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 8419 - 8428
  • [2] Few-Shot Object Detection via Transfer Learning and Contrastive Reweighting
    Wu, Zhen
    Li, Haowei
    Zhang, Dongyu
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PT VII, 2023, 14260 : 78 - 87
  • [3] FRDet: Few-shot object detection via feature reconstruction
    Chen, Zhihao
    Mao, Yingchi
    Qian, Yong
    Pan, Zhenxiang
    Xu, Shufang
    IET IMAGE PROCESSING, 2023, 17 (12) : 3599 - 3615
  • [4] Few-Shot Object Detection via Variational Feature Aggregation
    Han, Jiaming
    Ren, Yuqiang
    Ding, Jian
    Yan, Ke
    Xia, Gui-Song
    THIRTY-SEVENTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 37 NO 1, 2023, : 755 - 763
  • [5] Few-shot Object Detection via Improved Classification Features
    Jiang, Xinyu
    Li, Zhengjia
    Tian, Maoqing
    Liu, Jianbo
    Yi, Shuai
    Miao, Duoqian
    2023 IEEE/CVF WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2023, : 5375 - 5384
  • [6] Meta-Feature Relearning with Embedded Label Semantics and Reweighting for Few-Shot Object Detection
    Li P.-F.
    Liu F.
    Li L.-L.
    Liu X.
    Feng Z.-X.
    Jiao L.-C.
    Xiong Y.-M.
    Jisuanji Xuebao/Chinese Journal of Computers, 2022, 45 (12): : 2561 - 2575
  • [7] Efficient Feature Enhancement for Few-Shot Object Detection
    Li, Lin
    Lei, Zhou
    Chen, Shengbo
    Xu, Qingguo
    2022 IEEE 6TH ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2022, : 1206 - 1210
  • [8] Few-Shot Object Detection Based on Contrastive Class-Attention Feature Reweighting for Remote Sensing Images
    Miao, Wang
    Zhao, Zihao
    Geng, Jie
    Jiang, Wen
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 2800 - 2814
  • [9] Few-Shot Object Detection with Local Feature Enhancement and Feature Interrelation
    Lai, Hefeng
    Zhang, Peng
    ELECTRONICS, 2023, 12 (19)
  • [10] Few-Shot Object Detection via Sample Processing
    Xu, Honghui
    Wang, Xinqing
    Shao, Faming
    Duan, Baoguo
    Zhang, Peng
    IEEE ACCESS, 2021, 9 (09): : 29207 - 29221