一种基于旋转目标检测模型的机车惯性测量单元校准方法及装置

本发明属于机车状态检测技术领域，具体涉及一种基于旋转目标检测模型的机车惯性测量单元校准方法及装置。方法包括以下步骤：获取机车驾驶室操纵台的图像；将所述图像输入到机车手柄检测模型中，识别出图像中的各个手柄，并确定手柄和对应手柄台的相对位置；其中，机车手柄检测模型是利用手柄检测数据集对旋转目标检测模型进行训练得到的，手柄检测数据集为标注出手柄和对应手柄台的机车驾驶室操作台图像集；根据手柄和对应手柄台的相对位置确定手柄的状态；根据手柄的状态确定机车的运行状态；根据确定的机车运行状态对机车惯性测量单元进行校准。保证机车惯性测量单元的检测精度，从而提升机车速度测量精度以及定位的精度，保证机车正常运行。