自动识别方法以及基于该自动识别方法的定位方法技术

本发明专利技术提供一种自动识别方法以及基于该自动识别方法的定位方法，该自动识别的方法包括：S10：获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息；S20：在螺丝帽的目标图像进行角点提取，得到初始角点集合；S30：将该初始角点集合与对应类型的螺丝帽的预设角点位置模板匹配，得到目标图像的最终角点集合；S40：对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标。本发明专利技术通过对不同类型的螺丝的螺丝帽的特征进行识别，从而确定目标螺丝的螺丝帽位置以及类型，并且进一步地对目标螺丝的螺丝帽的角点进行识别，并且与预设角点位置模板匹配，最终得到目标螺丝的中心位置，完成对目标螺丝的位置的识别。完成对目标螺丝的位置的识别。完成对目标螺丝的位置的识别。

【技术实现步骤摘要】
自动识别方法以及基于该自动识别方法的定位方法


[0001]本专利技术涉及人工智能
，具体涉及一种自动识别方法以及基于该自动识别方法的定位方法。

技术介绍

[0002]工业智能化是目前大趋势，流水线上劳动力需求巨大，并且容易产生误操作，需要用智能算法来提高工业产线的效率。光通讯的OSA生产就需要大量劳动力，有待提高效率。
[0003]螺丝帽在光模块生产使用环境中，从多个角度分析定位；螺丝帽是否使用的合适对整个产品的质量起着很重要的作用，有的位置是6角，3角内嵌，如果用人工的方式来进行判别容易出错，并且人工成本太高。

技术实现思路

[0004]有鉴于此，本专利技术提供一种自动识别方法以及基于该自动识别方法的定位方法，以实现通过人工智能代替人工。
[0005]为解决上述技术问题，本专利技术采用的技术方案是：一种自动识别方法，该方法包括：S10：获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息；S20：在螺丝帽的目标图像进行角点提取，得到初始角点集合；S30：将该初始角点集合与对应类型的螺丝帽的预设角点位置模板匹配，得到目标图像的最终角点集合；S40：对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标。
[0006]优选地，步骤S10包括：S101：获取拍摄的目标视频；S102：通过预训练的检测模型对目标视频进行检测，获取螺丝帽的目标图像、目标图像的位置信息以及螺丝帽类型。
[0007]优选地，步骤S10还包括S103：对目标图像按照预定比例外扩，以使目标图像包含整个螺丝帽。
[0008]优选地，角点提取的方法为：选取一个固定窗口在目标图像上进行任意方向上的滑动，比较滑动前与滑动后，窗口中的像素灰度变化程度，根据像素灰度变化程度判断该窗口内是否存在角点。
[0009]优选地，角点提取公式为：其中，是对应的进行角点检测的图像，是窗口内所对应的像素坐标位置；是窗口函数，u是窗口在横坐标方向的偏移量，v是窗口在纵坐标方向的偏移量，是窗口在横坐标方向偏移量为u，在纵坐标方向偏移量为v时的灰度值。
[0010]优选地，初始角点集合与对应类型的螺丝帽的预设角点位置模板匹配过程为：S301：从初始角点集合中找到与对应类型的螺丝帽的预设角点位置模板匹配的最终角点集合；S302：计算出目标图像的最终角点集合与预设角点位置模板中角点之间的旋转角度。
[0011]优选地，最终角点集合中角点位置的数量小于等于预设角点位置模板中角点位置的数量。
[0012]本专利技术提供一种基于自动识别的定位方法，包括：通过上述的自动识别方法对螺丝帽进行识别，得到目标螺丝帽的中心以及最终角点集合与预设角点位置模板中角点之间的旋转角度，并根据目标螺丝帽的中心以及最终角点集合与预设角点位置模板中角点之间的旋转角度驱动机械臂。
[0013]优选地，驱动机械臂的过程为计算机械臂控制的螺丝刀的中心与目标螺丝帽的中心之间的位置偏移量，根据该位置偏移量移动机械臂。
[0014]本专利技术具有的优点和积极效果是：本专利技术通过对不同类型的螺丝的螺丝帽的特征进行识别，从而确定目标螺丝的螺丝帽位置以及类型，并且进一步地对目标螺丝的螺丝帽的角点进行识别，并且与预设角点位置模板匹配，得到目标图像的最终角点集合，进一步地对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标，所述外接圆的中心坐标即为目标螺丝的螺丝帽的中心位置。
附图说明
[0015]图1是本专利技术一个具体的实施例中目标图像外扩之后的示意图；图2是本专利技术一个具体的实施例中矩形框移动的过程中获得的角点的示意图。
具体实施方式
[0016]为了更好的理解本专利技术，下面结合具体实施例和附图对本专利技术进行进一步的描述。
[0017]本专利技术提供一种自动识别方法，该方法包括：S10：获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息；S20：在螺丝帽的目标图像进行角点提取，得到初始角点集合；S30：将该初始角点集合与对应类型的螺丝帽的预设角点位置模板匹配，得到目标图像的最终角点集合；S40：对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标。
[0018]本专利技术通过对不同类型的螺丝的螺丝帽的特征进行识别，从而确定目标螺丝的螺丝帽位置以及类型，并且进一步地对目标螺丝的螺丝帽的角点进行识别，并且与预设角点位置模板匹配，得到目标图像的最终角点集合，进一步地对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标，所述外接圆的中心坐标即为目标螺丝的螺丝帽的中心位置，经过上述步骤完成对目标螺丝的螺丝帽的中心位置的识别。
[0019]进一步地，S10：获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息的具体过程包括：
S101：获取拍摄的目标视频；S102：通过预训练的检测模型对目标视频进行检测，获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息。其中，目标图像包括待识别的螺丝帽的完整图像，通常为矩形图像。
[0020]螺丝的种类包括内六角，内三角，外六角等，本专利技术实施例通过对标准图片以及其位置信息、大小以及螺丝帽类型信息进行深度网络模型训练，得到检测模型，每个标准图片中的螺丝帽通过矩形框进行标注，标准图片中标注螺丝的矩形框的位置、大小表示为：（x，y，w，h），其中，x，y是目标螺丝矩形框的左上角的横坐标以及纵坐标，w，h是目标螺丝矩形框的宽和高。
[0021]进一步地，在本专利技术的一个具体的实施例中，设置有检测模型分类器，使得该检测模型支持多种类型的螺丝的检测。
[0022]在本专利技术的一个具体的实施例中，深度网络模型为YOLOV4。
[0023]本专利技术的训练结果表示为：（x，y，w，h，label），其中，x，y是目标螺丝矩形框的左上角的横坐标以及纵坐标，w，h是目标螺丝矩形框的宽和高，label是对应的螺丝的标签值。
[0024]例如，在一个具体的实施例中，训练结果为（Xsix，Ysix，w，h，0），其中，Xsix 是检测到的螺丝矩形框的左上角横坐标，Ysix是检测到的螺丝矩形框的左上角纵坐标，w是检测到的螺丝矩形框的宽度，h是检测到的螺丝矩形框的高度，0 是标签值，在该实施例中，0例如代表是内六角。
[0025]经过检测模型的检测之后得到的目标图像中，有可能会存在螺丝帽的图像覆盖不全的问题，本专利技术在经过检测模型的检测之后，进行精细的定位。
[0026]经过检测模型的检测之后，得到检测结果：（x，y，w，h，label），其为一个粗略的结果，本专利技术的步骤S10还包括S103：对目标图像按照预定比例外扩，以使目标图像包含整个螺丝帽。
[0027]在本专利技术的一个具体的实施例中，训练结果例如为（Xsix，Ysix，w，h，0），将其按照一个预设比例scale进行外扩，得到螺丝的矩形框的外扩矩形（xorigin，yorigin，Wimg，Himg），扩展的方法为：Himg），扩展的方法为：Himg），本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种自动识别方法，其特征在于：该自动识别方法包括：S10：获取螺丝帽的目标图像以及目标图像的位置信息、大小和螺丝帽类型信息；S20：在螺丝帽的目标图像进行角点提取，得到初始角点集合；S30：将该初始角点集合与对应类型的螺丝帽的预设角点位置模板匹配，得到目标图像的最终角点集合；S40：对最终角点集合进行最小外接圆拟合，得到所述外接圆的中心坐标。2.根据权利要求1所述的自动识别方法，其特征在于：步骤S10包括：S101：获取拍摄的目标视频；S102：通过预训练的检测模型对目标视频进行检测，获取螺丝帽的目标图像、目标图像的位置信息、大小以及螺丝帽类型。3.根据权利要求2所述的自动识别方法，其特征在于：步骤S10还包括S103：对目标图像按照预定比例外扩，以使目标图像包含整个螺丝帽。4.根据权利要求1所述的自动识别方法，其特征在于：角点提取的方法为：选取一个固定窗口在目标图像上进行任意方向上的滑动，比较滑动前与滑动后，窗口中的像素灰度变化程度，根据像素灰度变化程度判断该窗口内是否存在角点。5.根据权利要求1所述的自动识别方法，其特征在于：角点提取公式为：其中，是对应的进行角点检测的图像，是窗口内所对应的像素坐...

【专利技术属性】
技术研发人员：陈敏杰，盛长永，高超，
申请(专利权)人：天津聚芯光禾科技有限公司，
类型：发明
国别省市：