公开(公告)号：CN118365628A

公开(公告)日：2024-07-19

申请号：CN202410603317.8

申请日：2024-05-15

Applicant: 东南大学

Inventor： 敬宛亭 ,  陈从颜

IPC: G06T7/00 ,  G06V10/25 ,  G06V10/764 ,  G06V10/80 ,  G06V10/82 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了一种类别失衡的光伏组件EL缺陷检测方法，实现了在样本失衡的情况下，对目前常见的10种小目标EL缺陷的检测和定位，包括以下步骤：对光伏数据集进行预处理和标注；将YOLOv5网络算法的骨干网络中的C3模块替换成GhostNet结构，在骨干网最后一层加入注意力机制CA模块，在颈部网络引入特征金字塔网络BiFPN模块，得到优化后的YOLOv5网络算法；针对样本类别不均衡问题，使用Focal Loss函数替换原有的损失函数；训练优化后的YOLOv5网络算法；使用训练好的模型对实时输入的图片进行目标检测，标注缺陷位置和种类。