电力现场作业中的安全帽检测算法

摘要

一种电力现场作业中的安全帽检测算法，包括以下步骤：S1：构建训练数据集；S2：通过dropna方法将训练数据集存在缺失值的样本删除；S3：通过数据增强的方式对训练数据集进行扩充；S4：对扩充后的训练数据集进行编码；S5：将编码后的训练数据集中的样本划分为训练集和测试集；S6：构建安全帽检测模型，包括：S61：配置caffe环境，安装faster rcnn模型；S62：将训练集中的样本导入faster rcnn模型中进行训练，获得安全帽检测模型；S7：将测试集中的样本导入安全帽检测模型，收集安全帽检测模型的识别准确度并使用Adam优化器对安全帽检测模型进行优化；S8：通过优化后的安全帽检测模型对电力现场作业中的实时图片数据进行检测。该电力现场作业中的安全帽检测算识别率高且准确。