基于改进YOLO架构的齿轮表面缺陷检测方法

本发明公开了基于改进YOLO架构的齿轮表面缺陷检测方法，属于齿轮表面缺陷检测的技术领域，包括：获取齿轮表面图像；将所述齿轮表面图像输入至预先训练好的YOLO目标检测网络，输出齿轮表面缺陷的类别和位置；所述YOLO目标检测网络包括骨干网络、颈部网络和检测头；其中，采用CSP‑EDLAN模块替代所述骨干网络和所述颈部网络中的C3k2模块；采用ADown模块替代所述骨干网络和所述颈部网络中的下采样卷积模块；采用C2TSSA模块替代所述骨干网络中的C2PSA模块；所述检测头采用LQEHead检测头；通过轻量化下采样、增强型特征表征、全局上下文建模与质量感知检测头的联合优化，在兼顾模型复杂度与推理效率的同时，提高了齿轮表面缺陷检测的精度、鲁棒性和泛化能力。