基于深度学习网络的烟雾检测算法matlab仿真

举报
简简单单做算法 发表于 2023/11/27 19:56:45 2023/11/27
【摘要】 1.算法运行效果图预览 2.算法运行软件版本matlab2022a 3.算法理论概述       基于深度学习网络的烟雾检测算法是一种端到端的检测方法,主要分为基于候选区域的二阶段目标检测器和基于回归的单阶段目标检测器两类。       基于候选区域的二阶段目标检测器的原理是,先通过训练区域候选网络(RPN)生成候选区域,然后将图片划分为背景和目标两种类别,同时对目标位置进行初步预测;再通过...

1.算法运行效果图预览

1.jpeg

2.jpeg

 

2.算法运行软件版本

matlab2022a

 

3.算法理论概述

       基于深度学习网络的烟雾检测算法是一种端到端的检测方法,主要分为基于候选区域的二阶段目标检测器和基于回归的单阶段目标检测器两类。

       基于候选区域的二阶段目标检测器的原理是,先通过训练区域候选网络(RPN)生成候选区域,然后将图片划分为背景和目标两种类别,同时对目标位置进行初步预测;再通过训练目标区域检测网络,实现对候选区域内目标类别的判定和目标位置的确定

具体实现步骤:

1、用遗传算法与Kmeans迭代算出最大可能召回率的anchor组合

2、初始化锚框,并输出预测框

3、将预测框与真实框进行对比,计算两个框之间的差值,再根据差值进行反向更新

4、迭代网络参数,直到收敛,求出最佳的锚框值 该方法具有很好的实用性,但网络结构复杂度较高,导致训练和检测的效率较低。

        另一种是基于回归的单阶段目标检测器。这类检测器的核心思想是将物体的检测视作一个回归任务,直接对每个像素进行分类并回归其边界框。

         其训练方法是采用锚框对每一个像素进行回归,训练出边框的位置和大小。由于所有像素都要进行回归,因此其计算量巨大,训练时间较长。

 

 

 

 

4.部分核心程序

% 预处理训练数据
data           = read(trainingData);
In_layer_Size  = [224 224 3];
 
% 估计锚框
pre_train_data = transform(trainingData, @(data)preprocessData(data,In_layer_Size));
NAnchor        = 3;
NBoxes         = estimateAnchorBoxes(pre_train_data,NAnchor);
numClasses     = width(vehicleDataset)-1;
% 创建Faster R-CNN网络
lgraph         = fasterRCNNLayers(In_layer_Size,numClasses,NBoxes,Initial_nn,featureLayer);
% 数据增强
aug_train_data = transform(trainingData,@augmentData);
augmentedData  = cell(4,1);
 
% 预处理数据并显示标注
trainingData   = transform(aug_train_data,@(data)preprocessData(data,In_layer_Size));
validationData = transform(validationData,@(data)preprocessData(data,In_layer_Size));
data           = read(trainingData);
I              = data{1};
bbox           = data{2};
% 设置训练参数
options = trainingOptions('sgdm',...
    'MaxEpochs',240,...
    'MiniBatchSize',2,...
    'InitialLearnRate',2e-5,...
    'CheckpointPath',tempdir,...
    'ValidationData',validationData);
% 训练Faster R-CNN目标检测器
[detector, info] = trainFasterRCNNObjectDetector(trainingData,lgraph,options,'NegativeOverlapRange',[0 0.15],'PositiveOverlapRange',[0.15 1]);
【版权声明】本文为华为云社区用户原创内容,未经允许不得转载,如需转载请自行联系原作者进行授权。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。