一种面向磁瓦缺陷的目标检测方法
摘要:
本发明涉及计算机视觉技术领域,具体为一种面向磁瓦缺陷的目标检测方法,包括以下步骤:对磁瓦进行拍照采集图像数据;根据数据的特点进行预处理;将步骤S2输出的结果输入添加了解耦头的预测部分;根据预测结果的损失对模型进行参数优化;选取在测试集表现最好的模型进行目标检测;有益效果为:本发明提出的面向磁瓦缺陷的目标检测方法,对磁瓦进行拍照采集图像数据,根据缺陷以及缺陷类别进行真实框以及类别的标注。之后对其进行训练集,验证集以及测试集的划分。之后将划分好的训练集输入到Yolov5检测模型的backbone与neck部分,进行特征提取与采样。
0/0