参考代码:
https://github.com/gengyanlei/fire-detect-yolov4.git
参考链接:
作者开源的数据集:
- 烟火(10827张图像,无标签)-百度云盘下载链接 提取码->(hhwq)
- 烟火(2059张图像,含标签)-百度云盘下载链接 提取码->(3q4r)
以上数据集包含烟雾图像,但是标注只有1类;开源模型只有fire类,如果要同时检测烟雾,需要自己在图像上标注烟雾类别。
采用yolov5烟雾-火灾2类检测模型,为自己的数据预标注,然后修正即可!
烟火检测数据集包含的场景类型:
大火-小火,建筑起火、草原起火、森林起火、车辆(汽车、卡车、摩托车、电动车)起火、白天-黑夜起火、室内-室外起火;
烟雾同火场景一致!
数据预标注方法:
(1)下载10827烟雾-火灾检测数据集
(2)利用本人开源的yolov5火灾-烟雾检测模型,为未标注的图像预标注
(3)人工修正火灾-烟雾检测预标注,重新训练yolov4、yolov5
(4)加油!!!