基于计算机视觉的铁路工机具上下线清点核查方法

本发明公开了一种基于计算机视觉的铁路工机具上下线清点核查方法，包括采集高铁工务段上下线及仓库工机具图像，经区域聚焦算法裁剪目标区域以消除干扰，结合透视变换、多变换拼接等增强技术及基于SAM和扩散模型的生成方法，构建多样性训练数据集；构建文本视觉多模态旋转目标检测模型，通过增强特征提取，融合作业计划表文本信息与视觉特征，引导模型关注关键类别；采用自适应阈值优化算法调整检测参数，结合少样本检测技术，经类别感知尺度滤波、离群框抑制及跨类别非极大值抑制处理后，与作业计划比对输出结果；基于误检与漏检驱动的样本裁剪筛选机制，挖掘高价值样本补充至训练集。具有提高了铁路工机具管理的安全性和效率的优点。