改进YOLOv5的油田作业现场安全着装小目标检测

摘要

针对油田作业现场监控视频中的工人安全着装小目标检测效果较差的问题,提出了改进YOLOv5的油田场景规范化着装检测方法Cascade-YOLOv5(C-YOLOv5).首先搭建YOLO-people与YOLO-dress级联的小目标检测网络,定位行人目标,然后裁剪出行人区域并进行尺度变换,最后对行人进行安全着装检测;为了充分融合浅层与深层特征信息,在各级网络中使用4个不同尺度的卷积特征层来预测待检测目标.最后在原始图像中用不同颜色的框标出行人以及行人的着装部件类别,从而判定行人是否着装规范.实验证明,相比原始YOLOv5算法,C-YOLOv5方法不仅满足实时性的要求,而且检测的mAP提升了2.3%.同时,融合了深浅层信息的改进方法有效地增强了特征的表征能力,提高了小目标的检测精度.