摘要

本发明公开了一种基于计算机视觉的地铁保护区工地钻机识别与预警方法,该方法步骤如下:S1、采集地铁保护区工地钻机图像数据;S2、利用图像数据增强手段扩充并构建多样化的钻机目标识别数据集;S3、构建地铁保护区工地钻机图像的检测模型;S4、实时采集地铁保护区工地钻机图像数据,输入钻机图像的检测模型中,最后得到识别结果并对危险进行预警。本发明有效解决了地铁保护区的人工巡检存在的巡检效率低、无法实时监测等不足,以及在实际检测中遇到的雨天、夜晚、曝光场景下的钻机识别困难的问题,在实际的地铁保护区场景中取得了优秀的钻机目标识别效果。