Fire Detection using YOLOv8
The aim of this project is to implement a fire detection system using the YOLOv8 (You Only Look Once) object detection model to identify fire in images and videos.
This project leverages the YOLOv8 architecture, which is known for its speed and accuracy in object detection tasks. The model is trained to detect fire and can be applied in various domains, including safety monitoring and environmental protection.
- Real-Time Fire Detection: Quickly identifies fire in images and videos.
- Model Implementation: Easy integration of YOLOv8 for fire detection tasks.
- Flexible Usage: Can be adapted for different environments and scenarios.
To run this project, you will need the following dependencies:
- Python 3.x
- PyTorch
- OpenCV
- NumPy
- Matplotlib
-
Clone the repository:
git clone https://github.com/rugved0102/Computer-Vision-Projects.git cd Computer-Vision-Projects/Fire-Detection-using-YOLOv8 -
Install the required packages:
pip install -r requirements.txt
You can use the main.py file to implement fire detection using YOLOv8. Below is a brief explanation of how to use it:
- Prepare your input images.
- Run the following command:
python main.py --input <path_to_your_image_or_video> --output <path_to_save_output>
- The output will display the detected fire instances along with the confidence score.
python main.py --input path/to/image.jpg --output path/to/output.jpgFeel free to contribute to this project. Please adhere to the following guidelines while making contributions:
- Follow PEP 8 standards for code.
- Use meaningful commit messages.
- Ensure the code works as expected before submitting a pull request.
- Comment your code for better readability.
This project is licensed under the MIT License. See the LICENSE file for more details.
