一种基于深度特征和视频目标检测的安全帽佩戴检测方法

摘要

本发明公开了一种基于深度特征和视频目标检测的安全帽佩戴检测方法，包括以下步骤：视频数据采集；数据标注：针对Step 1采集的数据进行人工标注；数据集准备：数据集组成包括划分后的训练集、测试集以及验证集，每一个集合中包含的是原始视频对应的图片，特别的训练集和验证集还包含每一张图片对应的标注数据；网络构建和训练：对输入视频中的关键帧提取特征，将关键帧提取的特征传递给不同的相邻帧；利用光流将关键帧特征传递并复用给当前帧的特征；目标分类和位置框预测；网络训练：每个感兴趣区域的损失函数为交叉熵损失与边界框回归损失的和。