Is baseline models for Fire only?
tjasmin111 opened this issue · 7 comments
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
Hi @tjasmin111! Thank you very much for your interest in our work!
The FireNet network was trained on a dataset with fire and smoke, but the MobileNet network was trained on a dataset with only fire images. However, the FireNet network is not very accurate for smoke detection. In that case, I suggest you use one of our approaches (YOLOv5 + AVT or YOLOv5 + TPT).
You can see more details in our paper:
-
Pedro Vinícius Almeida Borges de Venâncio, Roger Júnio Campos, Tamires Martins Rezende, Adriano Chaves Lisboa, Adriano Vilela Barbosa: A hybrid method for fire detection based on spatial and temporal patterns. In: Neural Computing and Applications, 2023.
Ok thank you for explaining this. But I don't knownwhy I have difficulty running your models. Baseline.py runs nicely but can't run detect.py at all
Did you install the dependencies in a Python environment? See the Dependencies section.
Yes. It would be best if you could have it as a docker image.
@tjasmin111 I just added a Docker image to the repository. Follow the steps in Tutorial to run the models.
If you like the work and want to help us spread it to the community, please star the repository. Thank you very much once again for your interest.
I tried it and the detections are pretty good. Was the model trained only on the D-fire dataset?
@tjasmin111 I'm glad the results were pretty good. Yes, the model was trained only with images from the D-Fire dataset.