一种基于YOLO算法的轨道异物检测方法

本发明属于轨道异物检测领域，本发明公开了一种基于YOLO算法的轨道异物检测方法；包括：收集轨道异物图像形成数据集，对数据集中的每个图像进行样本扩充处理和标签处理，进而获取样本集；在YOLOv5中加入通道注意力机制和方向感知机制，并设置约束条件，进而获取改进后的YOLOv5算法，记为G‑YOLOv5模型；使用样本集对G‑YOLOv5模型进行预训练，获取训练好的G‑YOLOv5模型；采集轨道的实时图像并进行处理，将处理后的图像导入训练好的G‑YOLOv5模型，获取实时的检测报告；将获取实时的检测报告布置在可视化界面上，实现基于YOLO算法的轨道异物高效检测。