基于机器视觉的全自动BLDC电机绕线机产品缺陷检测方法

摘要

本发明涉及绕线机产品的缺陷检测技术领域，具体公开基于机器视觉的全自动BLDC电机绕线机产品缺陷检测方法，包括步骤：S1：选取全自动BLDC电机绕线机缺陷产品图像数据，构建全自动BLDC电机绕线机缺陷产品数据集；S2：对缺陷产品数据集进行预处理，得到预处理数据集，并将其按VOC格式划分为训练集、测试集和验证集；S3：将ShuffleNet网络作为主干网络，构建轻量化改进的YOLOv4网络；S4：使用预处理数据集对改进的YOLOv4网络进行训练，得到缺陷检测网络；S5：将全自动BLDC电机绕线机产品数据输入缺陷检测网络，得到产品的缺陷检测结果。本发明中YOLOv4主干网络为ShuffleNetv2网络，并对其相关模块进行轻量化改进，可以大幅度降低网络参数，减少计算量，提升检测网络的速度与精度。