为解决传统太阳能电池片表面缺陷检测方法存在的检测精度低、速度慢、检测缺陷类型单一的问题,提出一种基于改进YOLOv7的太阳能电池片表缺陷检测算法。首先,在YOLOv7网络模型的基础上引入Swin Transformer模块,以增强模型的全局信息建模...为解决传统太阳能电池片表面缺陷检测方法存在的检测精度低、速度慢、检测缺陷类型单一的问题,提出一种基于改进YOLOv7的太阳能电池片表缺陷检测算法。首先,在YOLOv7网络模型的基础上引入Swin Transformer模块,以增强模型的全局信息建模;其次,引入Shuffle Attention机制,有效融合空间注意力和通道注意力机制,以增强模型的特征提取能力;再次,使用SIoU(Scylla Intersection over Union)损失函数替换原模型中的CIoU(Complete Intersection over Union)损失函数,提高模型的收敛速度与效率;最后,采用K-means++聚类算法优化先验框。实验结果表明,改进算法具有较好的检测效果,在测试集上的mAP@50%达到86.6%,相较于原始YOLOv7提升了4.8%,且检测速度并未大幅降低,能较为快速、准确地对太阳能电池片表面缺陷进行检测。展开更多
文摘为解决传统太阳能电池片表面缺陷检测方法存在的检测精度低、速度慢、检测缺陷类型单一的问题,提出一种基于改进YOLOv7的太阳能电池片表缺陷检测算法。首先,在YOLOv7网络模型的基础上引入Swin Transformer模块,以增强模型的全局信息建模;其次,引入Shuffle Attention机制,有效融合空间注意力和通道注意力机制,以增强模型的特征提取能力;再次,使用SIoU(Scylla Intersection over Union)损失函数替换原模型中的CIoU(Complete Intersection over Union)损失函数,提高模型的收敛速度与效率;最后,采用K-means++聚类算法优化先验框。实验结果表明,改进算法具有较好的检测效果,在测试集上的mAP@50%达到86.6%,相较于原始YOLOv7提升了4.8%,且检测速度并未大幅降低,能较为快速、准确地对太阳能电池片表面缺陷进行检测。