一种轨道交通实训工具识别方法、装置及介质

本发明公开了一种轨道交通实训工具识别方法、装置及介质，通过获取记录了轨道交通实训中各类工具的操作过程的实训监控视频，将实训监控视频转换为实训图像文件，并对实训图像文件进行数据预处理，以获取训练数据集后，基于训练数据集，对YOLO模型进行训练，并将训练完成后的YOLO模型作为识别网络模型，将识别网络模型转换为适应目标格式的目标模型，获取当前实训数据，将所述当前实训数据输入至边缘计算盒子中的所述目标模型，以获取输出的指示当前轨道交通实训中使用的各类工具名称的识别结果，因此本申请通过对识别模型的适配转换和量化，将识别模型部署在边缘计算盒子上，从而降低了成本和时延，提高了数据传输的稳定性。