一种复杂混叠场景下多类别五金件的快速识别方法

摘要

本公开提供了一种复杂混叠场景下多类别五金件的快速识别方法，涉及计算机图像识别技术领域，该方法包括获取训练样本集，所述训练样本集中的训练样本包括图像以及该图像对应的xml格式文件，所述xml格式文件用于标注对应的图像中五金件的位置和类别；将所述训练样本集输入基于yolov5改进的模型进行训练，生成五金件检测模型；获取待检测的图像，将所述待检测的图像输入五金件检测模型进行检测，对所述待检测的图像中包含的五金件进行识别；输出所述待检测的图像中包含的五金件的位置和类别；本公开能够在五金件杂乱摆放的场景下，准确地识别出其类别与位置。