轨面区域识别方法、装置、设备、存储介质

本申请提供一种轨面区域识别方法、装置、设备、存储介质，该方法包括：获取轨面图像；通过预先训练的轨面区域识别模型识别轨面图像中的轨面区域；其中，轨面区域识别模型由识别模块和对齐模块构成；识别模块基于YOLO模型构建；识别模块的卷积层采用深度可分离卷积实现；识别模块的卷积层中的原型数量根据轨面图像确定；对齐模块，用于识别模块得到的对轨面区域mask的质量进行显式监督。其中，识别模块可以减少轨面识别过程的特征参数量和计算复杂度，进而保证轨面识别的实时性；对齐模块不仅显著提升了轨面区域识别的准确性，还使得轨面区域识别模型在不同线路下都能对轨面区域进行准确的识别。