检测模型的生成方法、线轴的检测方法及其装置

本发明公开了一种检测模型的生成方法、线轴的检测方法及其装置，该生成方法包括以下步骤：获取若干训练图片，创建基于YOLO的检测模型；基于所述训练图片对所述检测模型进行训练，在训练时，采用余弦退火学习率调度策略；在前Num1个epoch中，将学习率逐步提高到预设的初始学习率值，在训练Num2个epoch后，将初始学习率衰减至初始学习率值*0.01。该检测模型能够检测线轴是否发生故障。