GitHub - bobo12391/Forest-Fire-and-Smoke-Detecton-and-Fire-Segmentation-GUI-design: This is a GUI interface which can process forest fire detection, smoke detection and fire segmentation. Yolov5 is used to detect fire and smoke and unet is used to segment fire.

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.

Test on the ubuntu 20.04

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

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

You can put it in the root directory.

cd forest _fire_and_smoke_detection_and_fire_segmentation_system
python gui2.py

If you have any questions, please contact with me by email:15108513361@163.com

Name		Name	Last commit message	Last commit date
Latest commit History 81 Commits
U-Net		U-Net
__pycache__		__pycache__
data		data
inference		inference
models		models
seg_result_img		seg_result_img
utils		utils
_config.yml		_config.yml
best.pt		best.pt
camera.png		camera.png
dataset.py		dataset.py
detect.png		detect.png
detect.py		detect.py
gui.png		gui.png
gui2.py		gui2.py
hubconf.py		hubconf.py
readme.md		readme.md
segmentation.png		segmentation.png
stackimage.py		stackimage.py
test.py		test.py
train.py		train.py
unet.py		unet.py
unet_test.py		unet_test.py