[发明专利]一种基于YOLO算法的mini-LED芯片精密定位方法

权利要求书

1.一种基于YOLO算法的mini-LED芯片精密定位方法，其特征在于，包括步骤：

设定待跟踪的mini-LED芯片中的mark图片，设置在图像中的预定像素位置，令预定像素位置作为芯片最终要停留的位置；相机放置到指定拍摄位置，启动传送带后开启相机实时逐帧采集图像并将图像实时发送给计算机；

将作为特定目标的mark图片作为模板存入模板文件中，作为训练的样本；

将出现mark图片的第一帧图像作为第一帧图像，使用YOLO检测算法从第一帧图像起的每一帧图像都进行目标检测，得到每一帧图像中mark图片的置信度和像素位置；

根据图像的置信度与设定的置信度阈值的大小关系，确定图像匹配情况，具体为：

如果第i帧图像的置信度大于等于置信度阈值，则表示跟踪成功，第i帧图像的目标跟踪完成，将匹配到的mark图片从第i帧图像中截取出来作为新的模板存入模板文件中，令i=i+1，进行下一帧图像的目标跟踪；

如果置信度小于置信度阈值，则第i帧图像中没有特定目标，令i=i+1，进行下一帧图像的目标跟踪；

将mark图片在所检测帧图像中的像素位置与预定像素位置进行比较，调节传送带的速度，其中，实时像素位置与预定像素位置越近，传送带的速度需要控制得越小，传送带的速度调节公式为：

v=k/S

其中v是传送带速度，k是系数，S是mark图片的像素位置{x,y}的x值；

当实时的mark图片的像素位置与像素预定位置距离相差小于等于允许误差时，停止传送带，完成定位。

2.根据权利要求1所述的方法，其特征在于，所述设置在图像中的预定像素位置，方法为：将mini-LED芯片放置在理想终点位置，查看图片取得mark点的中心像素的坐标，令该坐标作为预定像素位置。