This is a GUI interface which can process forest fire detection, smoke detection and fire segmentation

This is a forest fire and smoke detection and fire segmentation system which is based on the yolov5 and unet.

If you want to train your own yolov5 model, please click here to learn more details.

If you want to train your own unet model, please click here to learn more details.

Environment

Test on the ubuntu 20.04

torch = '1.7.1+cu110'
torchvision = '0.8.2+cu110'

Installation

git clone https://github.com/bobo12391/Forest-Fire-and-Smoke-Detecton-and-Fire-Segmentation-GUI-design.git

Pre-trained model

link: https://pan.baidu.com/s/1vhItuWMZ9gC_mvUTPGHzew code: h2wf

You can put it in the root directory.

Quick Start

cd forest _fire_and_smoke_detection_and_fire_segmentation_system
python gui2.py

GUI Interface

Fire Fire and Smoke Image Detection(You can also select a video to detect)

detect

Fire Image Segmentation

segmentation

Camera Demo

camera

contact information

If you have any questions, please contact with me by email:[email protected]