2 Star 1 Fork 1

mayanhui / smoke_recognition

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

图片中的烟雾检测

依赖

参见 requirements.txt

训练模型

python smokeDetection.py image_train/smoke image_train/nosmoke
# 或者简写为
python smokeDetection.py
# 如上方式将使用默认的训练图片目录

调用模型分类

# 用全局模型进行分类
python smokeDetection_eval_whole.py [单张图片路径或者目录]
# 用局部模型进行分类
python smokeDetection_eval_local.py [单张图片路径或者目录]
# 综合局部模型和全局模型进行分类
python smokeDetection_eval_combine.py [单张图片路径]

配置选项

smokeDetection.ini 文件中,可以进行一些全局配置,例如

  • 特征和模型定义文件
  • 是否是全局模型(在训练局部模型时,这一项请务必置0)
  • 默认的训练图片正负样例所在目录
  • 保存的模型文件名称
  • 判为正例的阈值

logger.conf 文件中可以对日志进行配置

扩展模型

用户可以自定义特征和模型(只需定义相应的 getFeature 和 getModel 方法,然后在配置文件中注明即可(请参考文件 features/feature_RGBspace_HOG.pyalgorithms/algo_LR.py中的写法,并注意返回值的类型)。

空文件

简介

图片烟雾识别 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/TimVerion/smoke_recognition.git
git@gitee.com:TimVerion/smoke_recognition.git
TimVerion
smoke_recognition
smoke_recognition
master

搜索帮助

344bd9b3 5694891 D2dac590 5694891