助力工业园区作业违规行为检测预警，基于YOLOv3开发构建工业园区场景下作业人员违规行为检测识别系统

在很多工业园区生产作业场景下保障合规合法进行作业生产操作，对于保护工人生命安全降低安全隐患有着非常重要的作用，但是往往在实际的作业生产中，因为一个安全观念的淡薄或者是粗心大意，对于纪律约束等意思薄弱，导致在进行正常的作业生产中并没有按照安全规范要求去进行操作，比如：工地内没有佩戴或者是没有正确佩戴安全帽就进行施工生产，电力设备园区场景下未穿戴绝缘手套或者是反光标识就进行作业施工，等等，对于这类问题的早发现早预警能够在源头端极大程度降低此类问题可能带来的安全隐患，本文正是在这样的背景基础上设想从技术的角度来开发构建工业园区场景下作业人员违规行为检测识别系统，助力建设智慧安全园区。

首先看下实例效果：

在前文我们已经进行了相关的实践，感兴趣的话可以自行移步阅读即可：

《助力工业园区作业违规行为检测预警，基于YOLOv8【n/s/m/l/x】全系列参数模型开发构建工业园区场景下作业人员违规行为检测识别系统》

《助力工业园区作业违规行为检测预警，基于YOLOv7【tiny/l/x】不同系列参数模型开发构建工业园区场景下作业人员违规行为检测识别系统》

本文是选择的比较经典的也是比较古老的YOLOv3来进行模型的开发，YOLOv3（You Only Look Once v3）是一种目标检测算法模型，它是YOLO系列算法的第三个版本。该算法通过将目标检测任务转化为单个神经网络的回归问题，实现了实时目标检测的能力。

YOLOv3的主要优点如下：

实时性能：YOLOv3采用了一种单阶段的检测方法，将目标检测任务转化为一个端到端的回归问题，因此具有较快的检测速度。相比于传统的两阶段方法（如Faster R-CNN），YOLOv3能够在保持较高准确率的情况下实现实时检测。

多尺度特征融合：YOLOv3引入了多尺度特征融合的机制，通过在不同层级的特征图上进行检测，能够有效地检测不同尺度的目标。这使得YOLOv3在处理尺度变化较大的场景时表现出较好的性能。

全局上下文信息：YOLOv3在网络结构中引入了全局上下文信息，通过使用较大感受野的卷积核，能够更好地理解整张图像的语义信息，提高了模型对目标的识别能力。

简洁的网络结构：YOLOv3的网络结构相对简洁，只有75个卷积层和5个池化层，使得模型较易于训练和部署，并且具有较小的模型体积。

YOLOv3也存在一些缺点：

较低的小目标检测能力：由于YOLOv3采用了较大的感受野和下采样操作，对于小目标的检测能力相对较弱。当场景中存在大量小目标时，YOLOv3可能会出现漏检或误检的情况。

较高的定位误差：由于YOLOv3将目标检测任务转化为回归问题，较粗糙的特征图和较大的感受野可能导致较高的定位误差。这意味着YOLOv3在需要较高精度的目标定位时可能会受到一定的限制。

YOLOv3是YOLO系列里程碑性质的模型，随着不断地演变和发展，目前虽然已经在性能上难以与YOLOv5之类的模型对比但是不可否认其做出的突出贡献。

接下来简单看下数据集情况：

本文选择的是yolov3-tiny模型，训练数据配置文件如下：

# path
train: ./dataset/images/train/
val: ./dataset/images/test/


# number of classes
nc: 3

 
# class names
names: ['nosafehat', 'nowear', 'smoke']

模型配置文件如下：

# parameters
nc: 3   # number of classes
depth_multiple: 1.0  # model depth multiple
width_multiple: 1.0  # layer channel multiple
 
# anchors
anchors:
  - [10,14, 23,27, 37,58]  # P4/16
  - [81,82, 135,169, 344,319]  # P5/32
 
# YOLOv3-tiny backbone
backbone:
  # [from, number, module, args]
  [[-1, 1, Conv, [16, 3, 1]],  # 0
   [-1, 1, nn.MaxPool2d, [2, 2, 0]],  # 1-P1/2
   [-1, 1, Conv, [32, 3, 1]],
   [-1, 1, nn.MaxPool2d, [2, 2, 0]],  # 3-P2/4
   [-1, 1, Conv, [64, 3, 1]],
   [-1, 1, nn.MaxPool2d, [2, 2, 0]],  # 5-P3/8
   [-1, 1, Conv, [128, 3, 1]],
   [-1, 1, nn.MaxPool2d, [2, 2, 0]],  # 7-P4/16
   [-1, 1, Conv, [256, 3, 1]],
   [-1, 1, nn.MaxPool2d, [2, 2, 0]],  # 9-P5/32
   [-1, 1, Conv, [512, 3, 1]],
   [-1, 1, nn.ZeroPad2d, [0, 1, 0, 1]],  # 11
   [-1, 1, nn.MaxPool2d, [2, 1, 0]],  # 12
  ]
 
# YOLOv3-tiny head
head:
  [[-1, 1, Conv, [1024, 3, 1]],
   [-1, 1, Conv, [256, 1, 1]],
   [-1, 1, Conv, [512, 3, 1]],  # 15 (P5/32-large)
 
   [-2, 1, Conv, [128, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 8], 1, Concat, [1]],  # cat backbone P4
   [-1, 1, Conv, [256, 3, 1]],  # 19 (P4/16-medium)
 
   [[19, 15], 1, Detect, [nc, anchors]],  # Detect(P4, P5)
  ]

主要包括：未戴安全帽、未穿工服和违规抽烟三个类别的目标。

默认100次epoch的迭代计算，终端日志输出如下所示：

等待训练完成后来整体看下结果详情：

【数据分布可视化】

【PR曲线】
精确率-召回率曲线（Precision-Recall Curve）是一种用于评估二分类模型性能的可视化工具。它通过绘制不同阈值下的精确率（Precision）和召回率（Recall）之间的关系图来帮助我们了解模型在不同阈值下的表现。
精确率是指被正确预测为正例的样本数占所有预测为正例的样本数的比例。召回率是指被正确预测为正例的样本数占所有实际为正例的样本数的比例。
绘制精确率-召回率曲线的步骤如下：
使用不同的阈值将预测概率转换为二进制类别标签。通常，当预测概率大于阈值时，样本被分类为正例，否则分类为负例。
对于每个阈值，计算相应的精确率和召回率。
将每个阈值下的精确率和召回率绘制在同一个图表上，形成精确率-召回率曲线。
根据曲线的形状和变化趋势，可以选择适当的阈值以达到所需的性能要求。
精确率-召回率曲线提供了更全面的模型性能分析，特别适用于处理不平衡数据集和关注正例预测的场景。曲线下面积（Area Under the Curve, AUC）可以作为评估模型性能的指标，AUC值越高表示模型的性能越好。
通过观察精确率-召回率曲线，我们可以根据需求选择合适的阈值来权衡精确率和召回率之间的平衡点。根据具体的业务需求和成本权衡，可以在曲线上选择合适的操作点或阈值。

【训练可视化】

【Batch实例】

感兴趣的话也都可以自行动手尝试下！

如果自己不具备开发训练的资源条件或者是没有时间自己去训练的话这里我提供出来对应的训练结果可供自行按需索取。

单个模型的训练结果默认YOLOv3-tiny