基于YoloV5的烟火监控方位提取方法及系统技术方案

本发明专利技术公开了一种基于YoloV5的烟火监控方位提取方法及系统，涉及文字识别技术领域。具体步骤为：获取监控图像，并对所述监控图像中的方位信息进行标注，构建方位信息侦测数据集；对yolov5网络进行改进，得到第一网络；训练所述第一网络，得到识别网络；利用所述识别网络对所述方位信息侦测数据集进行识别，提取每张图中方位信息区域，并保存为图像，构建文字识别数据集，对所述文字识别数据集进行标注；将paddlepaddleocr作为识别模型进行训练，得到第一模型；把所述识别网络提取的图像区域输入到所述第一模型中，获取监控方位信息。使用本发明专利技术方法，可以通过烟火监控图像获取摄像头的方位信息，提高效率。提高效率。提高效率。

【技术实现步骤摘要】
基于Yolo V5的烟火监控方位提取方法及系统


[0001]本专利技术涉及文字识别
，更具体的说是涉及一种基于Yolo V5的烟火监控方位提取方法及系统。

技术介绍

[0002]在森林防火监控领域，通过摄像头进行火灾识别和预警是一种常用的技术手段。但是有很多用户，只能拿到视频通道，而摄像头方位信息等各种参数都显示在画面上，不能被用户直接获取。因此，对本领域技术人员来说，如何直接获取到烟火监控方位信息是亟待解决的问题。

技术实现思路

[0003]有鉴于此，本专利技术提供了一种基于Yolo V5的烟火监控方位提取方法及系统，以解决
技术介绍
中存在的问题。
[0004]为了实现上述目的，本专利技术采用如下技术方案：一种基于Yolo V5的烟火监控方位提取方法，具体步骤包括如下：
[0005]获取监控图像，并对所述监控图像中的方位信息进行标注，构建方位信息侦测数据集；
[0006]对yolov5网络进行改进，得到第一网络；
[0007]训练所述第一网络，得到识别网络；
[0008]利用所述识别网络对所述方位信息侦测数据集进行识别，提取每张图中方位信息区域，并保存为图像，构建文字识别数据集，对所述文字识别数据集进行标注；
[0009]将paddlepaddle ocr作为识别模型进行训练，得到第一模型；
[0010]把所述识别网络提取的图像区域输入到所述第一模型中，获取监控方位信息。
[0011]可选的，所述对yolov5网络进行改进的方式为：删减C3结构数量，减小backbone和head各卷积层的通道参数，具体实现为：banckbone第4层C3，基础个数由6改为3，第6层C3基础个数由9改为3，第7层Conv输出channel由1024改为256，第8层C3输出channel由1024改为256，第9层SPPF，输出channel由1024改为256，其余所有层输出channel均改为原来的一半。
[0012]可选的，对所述监控图像中的方位信息根据不同字体打上分类标签，并将所述方位信息侦测数据集按9：1划分为训练集和测试集。
[0013]可选的，对所述第一网络进行训练时，imgsize设为640，multi
‑
scale设为True。
[0014]可选的，加载预训练权重对paddlepaddle ocr模型进行训练，利用ch_ppocr_mobile_v2.0_rec作为预训练权重。
[0015]可选的，所述监控图像为不同监测地点在不同时间段的图像，均为可见光图像。
[0016]另一方面，提供一种基于Yolo V5的烟火监控方位提取系统，包括数据集获取模块、识别网络构建模块、文字识别数据集构建模块、第一模型构建模块、方位信息获取模块；其中，
[0017]所述数据集获取模块，用于获取监控图像，并对所述监控图像中的方位信息进行标注，构建方位信息侦测数据集；
[0018]所述识别网络构建模块，用于对yolov5网络进行改进，得到第一网络；训练所述第一网络，得到识别网络；
[0019]所述文字识别数据集构建模块，用于利用所述识别网络对所述方位信息侦测数据集进行识别，提取每张图中方位信息区域，并保存为图像，构建文字识别数据集，对所述文字识别数据集进行标注；
[0020]所述第一模型构建模块，用于将paddlepaddle ocr作为识别模型进行训练，得到第一模型；
[0021]所述方位信息获取模块，用于把所述识别网络提取的图像区域输入到所述第一模型中，获取监控方位信息。
[0022]可选的，还包括训练模块，所述训练模块包括第一训练模块与第二训练模块；其中，所述第一训练模块，用于对所述第一网络进行训练，所述第二训练模块，用于对paddlepaddle ocr模型进行训练。
[0023]经由上述的技术方案可知，与现有技术相比，本专利技术公开提供了一种基于Yolo V5的烟火监控方位提取方法及系统，具有以下有益的技术效果：
[0024](1)可以通过烟火监控图像获取摄像头的方位信息，提高了方位信息的获取效率；
[0025](2)通过对yolov5网络的改进，模型权重文件大小从40.8MB(yolov5m)下降到6.34MB，模型参数量从21.2M(yolov5m)下降到3.15M，大大减小了模型大小，提升了运行速度，也可适用于边缘设备，使其适配于各种小型系统，兼容性高。
附图说明
[0026]为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据提供的附图获得其他的附图。
[0027]图1为本专利技术的方法流程图；
[0028]图2为本专利技术的系统结构图。
具体实施方式
[0029]下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0030]本专利技术实施例1公开了一种基于Yolo V5的烟火监控方位提取方法，如图1所示，具体步骤包括如下：
[0031]S1、获取监控图像，并对监控图像中的方位信息进行标注，构建方位信息侦测数据集；
[0032]具体的，获取不同监测地点在不同时间段的图像，获取的监控图像均为可见光图
像，对图像进行标注，框出位置区域，并根据不同字体打上分类标签，数据集按9：1划分训练集和测试集。在本实施例中，分出type1和type2两个种类的字体，并最终整理出2006张标注的图片作为训练数据集，236张作为测试数据集。
[0033]S2、对yolov5网络进行改进，得到第一网络；
[0034]yolov5网络中，各层设置很高的输出channel值，是为了分出更多的特征信息，应对几百乃至上千类的目标识别，在本实施例中没有这么多种类目标，只有2类，所以改进方式如下：在yolov5m.yaml文件中修改网络参数，banckbone第4层C3，基础个数由6改为3，第6层C3基础个数由9改为3，第7层Conv输出channel由1024改为256，第8层C3输出channel由1024改为256，第9层SPPF，输出channel由1024改为256，其余所有层输出channel均改为原来的一半，Head除最后Detect层，其余所有输出channel均设为256。C3是Yolov5网络中的一种结构模块。
[0035]S3、训练第一网络，得到识别网络；
[0036]训练时，imgsize设为640，multi
‑
scale设为True，以增强数据的丰富程度，提升网络适应本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于Yolo V5的烟火监控方位提取方法，其特征在于，具体步骤包括如下：获取监控图像，并对所述监控图像中的方位信息进行标注，构建方位信息侦测数据集；对yolov5网络进行改进，得到第一网络；训练所述第一网络，得到识别网络；利用所述识别网络对所述方位信息侦测数据集进行识别，提取每张图中方位信息区域，并保存为图像，构建文字识别数据集，对所述文字识别数据集进行标注；将paddlepaddle ocr作为识别模型进行训练，得到第一模型；把所述识别网络提取的图像区域输入到所述第一模型中，获取监控方位信息。2.根据权利要求1所述的一种基于Yolo V5的烟火监控方位提取方法，其特征在于，所述对yolov5网络进行改进的方式为：删减C3结构数量，减小backbone和head各卷积层的通道参数，具体实现为：banckbone第4层C3，基础个数由6改为3，第6层C3基础个数由9改为3，第7层Conv输出channel由1024改为256，第8层C3输出channel由1024改为256，第9层SPPF，输出channel由1024改为256，其余所有层输出channel均改为原来的一半。3.根据权利要求1所述的一种基于Yolo V5的烟火监控方位提取方法，其特征在于，对所述监控图像中的方位信息根据不同字体打上分类标签，并将所述方位信息侦测数据集按9：1划分为训练集和测试集。4.根据权利要求1所述的一种基于Yolo V5的烟火监控方位提取方法，其特征在于，对所述第一网络进行训练时，imgsize设为640，multi
‑
scale设为Tru...

【专利技术属性】
技术研发人员：廖艳红，
申请(专利权)人：成都鸿钰网络科技有限公司，
类型：发明
国别省市：