摘要
本发明涉及目标检测技术领域,具体公开了一种基于分层加权特征融合网络的玻璃瓶缺陷检测方法,获取不同缺陷的玻璃瓶口图像数据;搭建原始的YOLOv7网络模型,对YOLOv7的网络结构进行改进,在主干特征提取网络中引入CA注意力机制,构建高效层聚合网络S‑ELAN模块替换原有的ELAN;将Neck部分的特征融合网络替换为分层加权的特征融合网络HWFF,对主干网络获得的浅层特征图进行特征融合,选用EIoU作为损失函数进行模型优化;对改进的YOLOv7模型进行训练,利用训练后的模型进行缺陷检测。与现有技术相比,本发明能够提高深度学习技术对玻璃瓶口缺陷检测的准确率,加快检测速度,实验验证相较于原算法提高了4.6个百分点,并且在小缺陷识别上,相较于原算法更准确。