基于改进RT-DETR的光伏电池可见光图像缺陷检测
DOI:
CSTR:
作者:
作者单位:

长沙理工大学人工智能学院 长沙 410114

作者简介:

通讯作者:

中图分类号:

TP391;TN36

基金项目:


Defect detection of photovoltaic cells based on improved RT-DETR for visible images
Author:
Affiliation:

School of Artificial Intelligence, Changsha University of Science and Technology, Changsha 410114, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为提升光伏电池缺陷检测在多类别、小目标等复杂场景下的准确性,基于RT-DETR提出了一种轻量化改进模型。首先,设计了一种轻量化的双路特征提取模块用于替代ResNet的基本模块,在减少模型参数的同时,使得模型具备局部与全局特征建模能力;其次,将基于注意力的同尺度交互模块(AIFI)中的多头注意力模块进行改良,删除部分注意力头,对剩下的注意力头进行线性变换,以缓解多头注意力机制中的冗余问题;最后,将网络中的卷积下采样模块替换为一种基于小波变换的下采样模块,改善下采样过程中存在的边缘信息丢失问题。实验结果表明,改进后的模型参数量相较原模型下降了47.3%,计算量下降了39.7%。在私有数据集上,改进模型的精确率、召回率、mAP@50相较基准模型分别提高了2.2%、4.1%、2.4%,这验证了改进模型的有效性;在公开数据集PVEL-AD上进行泛化性实验,改进模型的mAP@50达到了70.6%,比基准模型高6.3%,初步验证了其在未见数据上的泛化能力。

    Abstract:

    To enhance the accuracy of photovoltaic cell defect detection in complex scenarios such as multiple categories and small targets, this paper proposes a lightweight improved model based on RT-DETR. Firstly, a lightweight dual-path feature extraction module was designed to replace the basic module of ResNet. This not only reduced the model parameters but also enabled the model to possess the ability of local and global feature modeling. Secondly, the multi-head attention module in (attention-based intra-scale feature interaction,AIFI) was improved by deleting some attention heads and performing linear transformations on the remaining attention heads to alleviate the redundancy problem in the multi-head attention mechanism. Finally, the convolutional downsampling module in the network was replaced by a downsampling module based on wavelet transformation to improve the problem of edge information loss during downsampling. The experimental results show that the number of parameters of the improved model has decreased by 47.3% compared to the original model, and the computational cost has decreased by 39.7%. On the private dataset, the improved model outperformed the baseline with increases of 2.2%, 4.1% and 2.4% in precision, recall and mAP@50, respectively, demonstrating its effectiveness. In addition, a generalization experiment was conducted on the public PVEL-AD dataset, where the improved model achieved an mAP@50 of 70.6%, which is 6.3% higher than the baseline model, providing initial evidence of its generalization ability on unseen data.

    参考文献
    相似文献
    引证文献
引用本文

陈志洪,范必双,陈梦圆,张浩.基于改进RT-DETR的光伏电池可见光图像缺陷检测[J].电子测量技术,2026,49(12):120-129

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-09-04
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告