Giter Site home page Giter Site logo

pedbrgs / fire-detection Goto Github PK

View Code? Open in Web Editor NEW
30.0 2.0 8.0 7.13 MB

Fire and smoke detection using spatial and temporal patterns.

License: MIT License

Python 97.78% Shell 1.81% Dockerfile 0.41%
fire-detection convolutional-neural-networks smoke-detection yolo dfire

fire-detection's Introduction

About me

I received my bachelor's degree1 in Systems Engineering from Universidade Federal de Minas Gerais (UFMG) in 2019, followed by my master's degree2 in Computational Intelligence from UFMG in 2021. Currently, I am pursuing a PhD in Optimization at UFMG while also holding the position of Senior AI Engineer at BTG Pactual. My experiences include problem solving in computer vision, data science, machine learning and optimization. In my free time, I like training Brazilian jiu jitsu, skateboarding, traveling and cooking. Feel free to get in touch!

1 Bachelor's thesis (in Portuguese)
2 Master's thesis (in Portuguese), Video presentation (in Portuguese)


Selected publications:

For additional publications, please visit my profiles on Google Scholar or ResearchGate.


Projects:


fire-detection's People

Contributors

pedbrgs 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

Watchers

 avatar  avatar

fire-detection's Issues

Is baseline models for Fire only?

I was trying your baseline models (Firenet and Mobilenet). Are they for Fire only and not Smoke?
python baseline.py --video <video_file> --model firenet

How can convert yolov5 to yolov8

Hi, I want to ask how can I convert this code into yolov8 format to use with temporal features after detection? Please tell me how can I do this?

Help with some configuration values

Thank you for your work "A hybrid method for fire detection based on spatial and temporal patterns (2023)".

My question is:

  • About test video: Is a video start with "FP" is the positive video (having fire/smoke) and those starting with "VP" is the negative sample?
    The video:
    image

  • About the result table 3: are you use these following values as provided in the code (or the different ones) to produce this result table?

 + conf_thres=0.25,  # confidence threshold
 + iou_thres=0.45,  # NMS IOU threshold"

Result table: image

Thank you for your clarification!

Show fire/smoke detections?

I am running the model with python3 baseline.py --video ex1.mp4 --model mobilenet, but it only shows some bash outputs and no output videos. Does it save the output file somewhere?

1/1 [==============================] - 0s 25ms/step
Time taken =  0.04462027549743652
Prediction: non_fire
1/1 [==============================] - 0s 25ms/step
Time taken =  0.043842315673828125
Prediction: non_fire
1/1 [==============================] - 0s 26ms/step
Time taken =  0.04700660705566406
Prediction: non_fire
1/1 [==============================] - 0s 27ms/step
Time taken =  0.050565481185913086
Prediction: non_fire
Results
     video    network  detected  first_frame  time_avg
0  ex1.mp4  mobilenet      True           10  0.049584

Runtime error on CUDA

I have a hard time running it. I guess because the code is for a few years ago and I get issues with new API & A100 GPU etc.
Is there any plans to test or address issues with newer library versions, new GPUs, etc?

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

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.