树莓派系列五:火焰检测二(基于yolov3)

前言

上一篇文章中利用RGB判据和HIS判据,设定合适的阈值条件,检测出火焰对应像素的区域,将原图二值化,经过中值滤波以及数学形态学的膨胀运算等图像处理,消除一些噪声及离散点,连通一些遗漏的区域,从而来检测火焰,这个方法在背景比较单调且与火焰差别较大时,效果良好,几乎没有任何噪声对其造成干扰。但当背景复杂或与火焰颜色比较相似时,会不时出现噪声和误判。今天和大家分享利用目标检测算法中的yolov3来检测火焰。

数据数据并标注

我在网络上找到了一批关于火焰的数据
树莓派系列五:火焰检测二(基于yolov3)
将这些数据标注
树莓派系列五:火焰检测二(基于yolov3)
树莓派系列五:火焰检测二(基于yolov3)

将标注好的数据进行训练,得到检测模型

树莓派系列五:火焰检测二(基于yolov3)

检测

树莓派系列五:火焰检测二(基于yolov3)
由于实验数据比较少,只有200多张图片,效果还有待提升,后续会继续收集图片数据,并利用树莓派上的摄像头做一个监控系统。

THE END

树莓派系列五:火焰检测二(基于yolov3)
扫描上方二维码关注“嵌入式案例Show”公众号,看更多嵌入式案例