一种轻量化光伏电池EL图像缺陷检测方法

本发明公开了一种轻量化光伏电池EL图像缺陷检测方法，首先对太阳能电池片EL图像进行采集，并形成只存在一种缺陷的分类数据集，以及包含多种缺陷的检测数据集。其次构建双残差掩码雪花高斯骨干网络DRMSG‑Net，使用分类头，结合分类数据集进行训练。然后构建分类剪枝头替换分类头，基于训练好的DRMSG‑Net权重，使用分类数据集对分类剪枝头进行训练，获得各特征通道重要度分数对特征通道进行剪枝。最后去除分类剪枝头，并构建特征融合网络和分类与回归网络，获得缺陷类型与目标框，通过检测数据集进行训练，获得缺陷检测结果。本发明能够学习到更加细致和复杂的表示，进行准确的光伏电池图像缺陷检测。