Smoke_Fire_Detection

烟雾火焰检测

介绍

用的yolov5框架

weights中smoke.pt是用torch1.4以上版本用的，smoke2.pt是torch1.4及其以下版本用的。

已封装成类，更改图片路径后直接运行smoke_fire_obj.py即可，返回一个列表

如需准对特定场景数据集进行Fine-Tune,请参考yolov5项目: https://github.com/ultralytics/yolov5

Name		Name	Last commit message	Last commit date
Latest commit History 20 Commits
README_IMG		README_IMG
models		models
utils		utils
weights		weights
README.md		README.md
smoke_file_obj.py		smoke_file_obj.py