一种基于机器视觉的实时检测无人机载荷装配质量的方法

本发明提供一种基于机器视觉的实时检测无人机载荷装配质量的方法，属于无人机装配技术领域；包括如下步骤；S1、使用数据采集系统获取待检测的无人机装配数据；S2、使用处理器，对进行训练、测试的视频和图像数据进行预处理；S3、使用Yolov5算法框架搭建质检平台，提取待检测装配件的图像中各个特征节点；S4、利用当前特征节点与所有特征节点的亲和度关系，将当前特征节点与所有特征节点堆叠以表示当前节点的全局结构信息。本发明通过将YOLO算法与注意力机制结合，提升了模型对于小目标和被遮挡目标的检测能力，更好地捕获这些难以检测的目标的特征，进而提高检测的召回率，减少误检和漏检的情况，提升模型的目标检测性能。