基于PP-YOLOv2的火灾/烟雾检测

下载PaddleX源码

本项目中已经帮大家下载好了最新版的PaddleX，无需下载，只需安装环境~

如仍需安装or安装更新，可以执行以下步骤

3 数据准备

本案例使用数据集包含MIT协议互联网下载图片和MIT协议公开数据集，共有6675张图片，分别检测烟雾和火灾，示例图片如下图所示：

数据集图片格式是VOC数据格式，VOC数据是每个图像文件对应一个同名的xml文件，xml文件内包含对应图片的基本信息，比如文件名、来源、图像尺寸以及图像中包含的物体区域信息和类别信息等。

xml文件中包含以下字段：

• filename，表示图像名称。

• size，表示图像尺寸。包括：图像宽度、图像高度、图像深度。

  <size>
      <width>500</width>
      <height>375</height>
      <depth>3</depth></size>

  登录后复制

• object字段，表示每个物体。包括:

  标签	说明
  name	物体类别名称
  pose	关于目标物体姿态描述（非必须字段）
  truncated	如果物体的遮挡超过15-20％并且位于边界框之外，请标记为truncated（非必须字段）
  difficult	难以识别的物体标记为difficult（非必须字段）
  bndbox子标签	(xmin,ymin) 左上角坐标，(xmax,ymax) 右下角坐标，

将这6675张图片按9:1比例随机切分，切分后包含6008张图片的训练集和667张图片的验证集，提供处理好的数据集，下载地址。包含以下文件夹和文件：images,annotations,labels.txt, train_list.txt和 val_list.txt，分别图片、xml标注文件、存储类别信息、训练样本列表、验证样本列表。训练样本列表和验证样本列表的每一行格式为：图片路径 对应的xml路径，例如images/fire_00348.jpg annotations/fire_00348.xml。

数据处理命令如下，执行一次即可。

最终数据集文件组织结构为：

 ├── dataset
	├── annotations
  │   ├── fire_000001.xml
  │   ├── fire_000002.xml
  │   ├── fire_000003.xml
  │   |   ...
  ├── images
  │   ├── fire_000001.jpg
  │   ├── fire_000003.jpg
  │   ├── fire_000003.jpg
  │   |   ...
  ├── label_list.txt
  ├── train.txt
  └── valid.txt

4.模型选择

PaddleX提供了5种目标检测模型：FasterRCNN、YOLOv3、PP-YOLO、PP-YOLOv2和PP-YOLO-tiny。

• Faster RCNN：Ross B. Girshick在2016年提出了新的FasterRCNN，需要先产生候选区域，再对RoI做分类和位置坐标的预测，这类算法被称为两阶段目标检测算法。在结构上，Faster RCNN已经将特征抽取(feature extraction)，proposal提取，bounding box regression(rect refine)，classification都整合在了一个网络中，使得综合性能有较大提高，在检测速度方面尤为明显。

• YOLOv3：Joseph Redmon等人在2015年提出YOLO（You Only Look Once，YOLO）算法，通常也被称为YOLOv1；2016年，他们对算法进行改进，又提出YOLOv2版本；2018年发展出YOLOv3版本。YOLOv3使用单个网络结构，在产生候选区域的同时即可预测出物体类别和位置，这类算法被称为单阶段目标检测算法。另外，YOLOv3算法产生的预测框数目比Faster R-CNN少很多。Faster R-CNN中每个真实框可能对应多个标签为正的候选区域，而YOLOv3里面每个真实框只对应一个正的候选区域。这些特性使得YOLOv3算法具有更快的速度，能到达实时响应的水平。

• PP-YOLO：PP-YOLO是PaddleDetection优化和改进的YOLOv3的模型，在COCO test-dev2017数据集上精度达到45.9%，在单卡V100上FP32推理速度为72.9 FPS，V100上开启TensorRT下FP16推理速度为155.6 FPS。PP-YOLO从如下方面优化：

  • 更优的骨干网络: ResNet50vd-DCN
  • 更大的训练batch size: 8 GPUs，每GPU batch_size=24，对应调整学习率和迭代轮数
  • Drop Block
  • Exponential Moving Average
  • IoU Loss
  • Grid Sensitive
  • Matrix NMS
  • CoordConv
  • Spatial Pyramid Pooling
  • 更优的预训练模型

• PP-YOLOv2：相较20年发布的PP-YOLO，PP-YOLOv2版本在COCO 2017 test-dev上的精度提升了3.6个百分点，由45.9%提升到了49.5%；在640*640的输入尺寸下，FPS达到68.9FPS。 主要改进点：

  • Path Aggregation Network
  • Mish Activation Function
  • Larger Input Size
  • IoU Aware Branch

• PP-YOLO-tiny：在当前移动互联网、物联网、车联网等行业迅猛发展的背景下，边缘设备上直接部署目标检测的需求越来越旺盛。生产线上往往需要在极低硬件成本的硬件例如树莓派、FPGA、K210 等芯片上部署目标检测算法。而我们常用的手机 App，也很难直接在终端采用超过 6M 的深度学习算法。如何在尽量不损失精度的前提下，获得体积更小、运算速度更快的算法呢？得益于 PaddleSlim 飞桨模型压缩工具的能力，体积仅为 1.3M 的 PP-YOLO-tiny 诞生了！PP-YOLO-tiny 沿用了 PP-YOLO 系列模型的 spp，iou loss, drop block, mixup, sync bn 等优化方法，并进一步采用了针对移动端的优化策略：

  • 更适用于移动端的骨干网络：MobileNetV3
  • 更适用移动端的检测头（head）：深度可分离卷积
  • 去除对模型体积、速度有显著影响的优化策略：iou aware、matrix NMS等
  • 使用更小的输入尺寸
  • 召回率优化
  • 更大的batch size
  • 量化后压缩

根据部署场景以及烟火检测实时性、高召回低误检要求，这里我们选择了PP-YOLOv2算法进行火灾和烟雾检测。

5 模型训练

本项目采用PP-YOLOv2作为烟火检测的模型，模型训练需要经过如下环节：

具体代码请参考train.py，可修改参数：

• train_transforms：训练预处理参数，可以增加、修改预处理方法和参数；
• eval_transforms：验证预处理参数，可以增加、修改预处理方法和参数；
• train_dataset：训练使用数据集，修改图片路径、标签路径以及是否进行数据shuffle；
• eval_dataset：验证使用数据集，修改图片路径、标签路径以及是否进行数据shuffle；
• pdx.models.*：设置不同的模型，可选[FasterRCNN、YOLOv3、PP-YOLO、PP-YOLO-tiny、PP-YOLOv2]，这里选择PP-YOLOv2；
• model.train：设置训练epoch、训练和验证数据集、batch size、学习率learning rate、warmup step、lr衰减lr_decay_epoch、模型保存间隔save_interval_epoch、模型保存路径save_dir，详细介绍训练参数。

PaddleX提供了单卡/多卡训练模型，满足用户多种训练需求:

6 模型评估

因为采用多个数据集一起训练，每个数据集标注方式不同，如下图左数据集A将火标注为一个检测框，但是下图右数据集B又将火标注为多个检测框。

不同的标注方式就导致训练好的模型，无法有效的计算mAP值。我们可以通过下图从两方面分析：

• 蓝色框表示真实标注框，可以看出有很多小检测框。实际评估时，火灾只有2个绿色检测框，能正确检测火灾，但没能和真实框一一对应，如果使用mAP作为指标，就不能正确反映检测效果。
• 有些数据集没标注烟雾，但是可以检测到烟雾，同样mAP也不能正确反映检测效果。

综上，我们计算不同置信度阈值下图片级召回率和图片级误检率，找到符合要求的召回率和误检率，对应的置信度阈值用于后续模型预测阶段。使用PP-YOLOv2训练好的模型进行评估，评估数据暂未公开，大家可以按照下面步骤准备数据集：

1）新建文件夹eval_imgs

2）准备包含火灾和烟雾的图片，放在eval_imgs/fire_smoke

硅基智能

基于Web3.0的元宇宙，去中心化的互联网，高质量、沉浸式元宇宙直播平台，用数字化重新定义直播

62

查看详情

3. 准备负样本：不包含火灾和烟雾的图片，尽量选择和当前场景相似的图片，放在eval_imgs/neg_pics

4）生成图片列表TXT文件，正样本标签为1，负样本标签为0，格式如下：

fire_smoke/00001.jpg 1fire_smoke/00002.jpg 1...
neg_pics/00001.jpg 0

代码实现：

得到PP-YOLOv2烟雾和火灾检测指标，召回相对达到预期，但是召回比较高，后续我们将一起探究如何降低误检率：

【名词解释】

• 图片级别的召回率：只要在有目标的图片上检测出目标（不论框的个数），该图片被认为召回。批量有目标图片中被召回图片所占的比例，即为图片级别的召回率。
• 图片级别的误检率：只要在无目标的图片上检测出目标（不论框的个数），该图片被认为误检。批量无目标图片中被误检图片所占的比例，即为图片级别的误检率。

7 模型预测

加载训练好的模型，置信度阈值设置为0.4，执行下行命令对验证集或测试集图片进行预测：

可视化预测结果示例如下，可以看出室内的火灾也可以有效检测出来：

注：图片来源于互联网，侵权删稿

8 模型导出

在模型训练过程中保存的模型文件是包含前向预测和反向传播的过程，在实际的工业部署则不需要反向传播，因此需要将模型进行导成部署需要的模型格式。 执行下面命令，即可导出模型

预测模型会导出到inference_model/目录下，包括model.pdmodel、model.pdiparams、model.pdiparams.info、model.yml和pipeline.yml五个文件，分别表示模型的网络结构、模型权重、模型权重名称、模型的配置文件（包括数据预处理参数等）和可用于PaddleX Manufacture SDK的流程配置文件。

• 更多关于模型导出的文档，请参考模型导出文档

9 模型推理

接下来使用PaddleX python高性能预测接口，在终端输入以下代码即可，同时可以通过以下代码进行速度预测：

关于预测速度的说明：加载模型后，前几张图片的预测速度会较慢，这是因为运行启动时涉及到内存显存初始化等步骤，通常在预测20-30张图片后模型的预测速度达到稳定。如果需要评估预测速度，可通过指定预热轮数warmup_iters完成预热。为获得更加精准的预测速度，可指定repeats重复预测后取时间平均值。

10 模型优化

本小节侧重展示在模型迭代过程中优化精度的思路，在本案例中，有些优化策略获得了精度收益，而有些没有。在其他质检场景中，可根据实际情况尝试这些优化策略，具体请参考模型优化文档

10.1 模型优化策略

• 修改图片尺寸：数据处理时，可修改target_size为480、608、640等

• 数据增强：数据处理时，使用不同的预处理、增强方法组合，包含：RandomHorizontalFlip、RandomDistort 、 RandomCrop 、RandomExpand、MixupImage等，详细解释请参考图像预处理/增强

• 不同模型：单阶段(YOLOv3、PP-YOLO、PP-YOLOv2等)

• 不同backbone：ResNet50、ResNet101、DarkNet53、MobileNet3，具体每个模型可选参数，可参考检测模型文件每个模型的backbone，如YOLOV3，就表示这些模型可以作为YOLOV3的backbone：

      if backbone not in [            'MobileNetV1', 'MobileNetV1_ssld', 'MobileNetV3',            'MobileNetV3_ssld', 'DarkNet53', 'ResNet50_vd_dcn', 'ResNet34'
      ]:

  登录后复制

• 是否加DCN(Deformable Convolution Network，可变形卷积)：FasterRCNN通过设置with_dcn=True使用可变性卷积

• 加入背景图片：当误检较高时，可以考虑使用add_negative_samplesAPI加入背景图片作为负样本进行训练，只需一行代码即可，image_dir表示负样本路径：

  train_dataset.add_negative_samples(image_dir='/home/aistudio/dataset/train_neg')

• 训练参数调整

10.2 推理时间计算方式

采用PaddleX在单卡Tesla V100上测试模型的推理时间（输入数据拷贝至GPU的时间、计算时间、数据拷贝至CPU的时间），推理时间取100次推理取平均耗时，即9 模型推理repeats参数设置为100。

10.3 不同模型结果

我们可以运行code文件夹下的训练文件，进行不同模型的训练，文件名的第一个数字和下表的序号相对应。

说明：

• 从表1的实验结论中可以发现，有些优化策略在精度优化上起到了正向结果，有些策略则相反。这些结论在不同的模型和不同的数据集上并不是相通的，还需根据具体情况验证。

• 本实验未提供"背景图"数据集(包含5116张图片)，大家自行选择不包含的烟雾和火灾的数据作为负样本即可。

模型优化思路：

• 1.通过选择更好的检测架构可以提高检测的Recall值——即Neck，Head部分的优化可以提高Recall。<YOLOv3 到 PP-YOLOv2>
• 2.添加数据增强、背景图可以降低误检率。
• 3.在数据量比较少的情况下，可以增加预训练模型。

通过以上的简单优化方式，获取了两个较好的模型结果：【前者模型速度更快、后者召回更高】

11 模型部署

在项目中为用户提供了基于Jetson NX的部署Demo方案，支持用户输入单张图片、文件夹、视频流进行预测。用户可根据实际情况自行参考。

部署方式可以参考：兼容并包的PaddleX-Inference部署方式和基于QT的Jetson Xavier部署Demo。