Fire Forest Detection

Figure 1 Algorithm Steps

This project is an AI model to detect and isolate the fire area in an image. The project is actually built on 2 models, a YOLOv8 model to localize the fire areas, and the result will be fed into a U-Net model to segment the fire image. Here is the input, detected fire locations, and the output image: