Git Product home page Git Product logo

gengyanlei / fire-smoke-detect-yolov4 Goto Github PK

View Code? Open in Web Editor NEW
1.2K 17.0 291.0 32.3 MB

fire-smoke-detect-yolov4-yolov5 and fire-smoke-detection-dataset 火灾检测,烟雾检测

License: MIT License

Python 5.38% CMake 0.50% Makefile 0.10% PowerShell 0.18% Shell 0.28% C 24.30% C++ 4.96% Batchfile 0.06% Cuda 5.80% Jupyter Notebook 58.45%
fire-detection smoke-detection fire-smoke-detection-dataset yolov4 yolov5 darknet pytorch

fire-smoke-detect-yolov4's Introduction

logo

author is leilei 👋

  • 🎓 CSU(master's degree)
  • 📖 leilei's Bio
  • 📖 leilei's blog
  • 🔭 Computer vision(目标检测-语义分割-分类)
  • ❤ Program language: Python C++ (Go Rust)-toLearn

logo

fire-smoke-detect-yolov4's People

Contributors

gengyanlei avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

fire-smoke-detect-yolov4's Issues

烟雾类比

您好,下载的数据中没有提供烟雾这个类别。请问是我解析错了吗?还是就是没有提供

yolov4 模型转换成onnx模型后,inference结果差距较大

Cannot access Baidu pan from Germany

I cannot access the fire-yolov4-weights and fire-dataset from the cloud Baidu... Could you please upload them on google drive or similar ? Thanks

how to use the onnx in C++ opencv with dnn

I am using opencv 4.5.4, which supports the use of converted yolov5.onnx. So, I converted the best.pt (inside Yolov5 folder) to best.onnx.

I can now load the model successfully using: cv::dnn::readNetFromONNX("best.onnx")
BUT, I do not know how to use it to find bounding boxes around fires and smokes given a test image!

Please help me anyone does know how. thanks.

关于数据集格式问题

我解压后,为什么这个文件没有后缀名呢, 没法打开,请问能帮忙解决一下吗?

Dataset License

Hi,
thanks for sharing your work. Are there some restrictions in using the dataset that you provide?
What is the license ?

Thanks for your help

关于烟火检测结果没有框的疑义?

@ALL
【1】此权重为yolov5 v3版本的代码,未做修改,有些按照我的readme执行代码,并未产生结果,未果不清楚具体原因
【2】你们可以直接下载yolov5 v3版本代码,然后load我的烟火2类权重即可,检测效果虽然不是最好的,但还可以!

关于数据集的问题

你好呀,我通过你的下载链接下载数据集后,里面的文件只有火焰的图片,请问哪个链接是和你用yolov5训练的火焰+烟雾的数据呢?

能一起合作吗

我最近也在做yolov4的火焰检测,能加QQ好友吗,接下来准备做yolov4剪枝和tensorrt部署。QQ:406174050
我因为工作可以去工厂采集数据,后面也方便建立数据集

Improving Map

Excuse me, i already train my weight using yoloV4 configuration from alexey and some augmentation data like rotate, brightness contrast, etc and have running about 6000 iteration but only got Map around 65% when using your dataset , when i see your weight you have Map around 75% could you please tell me what data augmentation are you using or what hyperparameter and augmentation best for this case , thanks

关于数据集的问题

下载了您提供的 “烟火(2059张图像,含标签)” 后发现里面只有含有火焰的标签,并未发现烟雾的标签。

我想使用您的YOLO v5 实现一个能同时检测烟雾和火焰的检测器,请问您用来训练v5的数据集烟雾火焰数据集可以开源吗?谢谢!

Epoch and Dataset

I used weight file on yolov5: best.pt. I see it is good. Can you tell me how many epoch did you train ? And Can you share me dataset about fire-smoke?

BaiDuYunPan_Download

提取码:hhwq

此外:
感谢分享数据的小伙伴,由于涉及小伙伴的成果,因此剩余的图像标注不开源!

改进建议

【1】烟雾-进一步细分成 白烟-黑烟-灰烟 or 浓烟 淡烟
【2】火 也可以进一步细分,因为这会增加网络训练难度,但是可以提升识别效果

我使用Alexey_yolov4+这个数据集,得到的结果如下,请问这是正常的么?

(next mAP calculation at 6175 iterations)
Last accuracy [email protected] = 62.27 %, best = 66.48 %
6000: 1.022357, 1.471952 avg loss, 0.000010 rate, 4.679000 seconds, 192000 images, 0.091826 hours left
Resizing to initial size: 416 x 416 try to allocate additional workspace_size = 52.43 MB
CUDA allocate done!

calculation mAP (mean average precision)...
Detection layer: 139 - type = 28
Detection layer: 150 - type = 28
Detection layer: 161 - type = 28
260
detections_count = 1069, unique_truth_count = 413
class_id = 0, name = fire, ap = 61.61% (TP = 258, FP = 106)

for conf_thresh = 0.25, precision = 0.71, recall = 0.62, F1-score = 0.66
for conf_thresh = 0.25, TP = 258, FP = 106, FN = 155, average IoU = 54.18 %

IoU threshold = 50 %, used Area-Under-Curve for each unique Recall
mean average precision ([email protected]) = 0.616056, or 61.61 %
Total Detection Time: 9 Seconds

感觉指标是不是有点低啊=-=
chart_yolov4-fire

缺少依赖项

FileNotFoundError: Could not find module 'D:\fire_smoke_detect\fire-smoke-detect-yolov4-master\yolov4\yolo_cpp_dll.dll'. Try using the full path with constructor syntax.
我在GitHub上找了个yolo_cpp_dll.dll放到工程中去但是还是报这个错误

烟雾能检测么?

实际预训练的模型,能否检测烟雾呢?因为我看到demo图片有一张是有烟雾检测的分类的。

含标签数据集疑问

您好,请问这个数据集:烟火(2059张图像,含标签)-百度云盘下载链接 提取码->(3q4r) 我下载解压后为什么是一个整体文件呢,不是pascol类型的数据集,是还需要做其他操作吗?

Weights file?

Can you share weights file to detect smoke and fire.
Thank you!

run crash

[yolo] params: iou loss: ciou (4), iou_norm: 0.07, obj_norm: 1.00, cls_norm: 1.00, delta_norm: 1.00, scale_x_y: 1.05
nms_kind: greedynms (1), beta = 0.600000
Total BFLOPS 127.232
avg_outputs = 1046213
Allocate additional workspace_size = 52.43 MB
Try to load weights: ./backup_fire/yolov4-fire_best.weights
Loading weights from ./backup_fire/yolov4-fire_best.weights...
seen 64, trained: 128 K-images (2 Kilo-batches_64)
Done! Loaded 162 layers from weights-file
Loaded - names_list: ./cfg/fire.names, classes = 1
Segmentation fault (core dumped)
只要调用predict_image就会崩溃,而且没有任何有效信息。

yolov5s fire-smoke weight is not the best!

注释:
yolov5s 烟火检测模型是我随便训练几个epoch的结果,并不是最好的结果。目前只能开源1个看起来还说的过去的模型。最好的模型不会开源的。

Colab Demo?

Hello, Thank you for your great work
Can you provide a Colab Notebook Demo for this amazing repo, It will help everyone and it will be much appreciated,
thank you

关于Yolov5的训练数据

您好,作者。我想咨询一个问题,Yolov5的烟火检测模型训练使用的数据规模是多少呢?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.