基于改进YOLOV5的电子粉涂敷均匀检测方法及设备
摘要:
本发明公开一种基于改进YOLOV5的电子粉涂敷均匀检测方法及设备,包括如下步骤:获取相机拍摄的电子粉涂敷照片,对电子粉涂敷不均匀样本数据集数据进行增强和扩充;建立基于YOLOV5网络的检测算法模型,使用STDC模块改进网络;使用K‑means++算法随机选择初始聚类中心作为自适应锚框;对检测算法模型进行训练和测试,使用SDG优化算法完成权值更新,本发明的技术方案中,在目标检测算法YOLOV5上进行改进,使用STDC模块改进特征提取网络;保留两个适应数据集的尺寸的特征图;使用Kmeans++聚类获取较优初始聚类中心,将图像算法应用于检测电子粉涂敷均匀能够大幅降低成本,节省人力、物力和财力。
0/0