Datasets and models for wildfire detection in PyTorch
Project description
Pyrovision: wildfire early detection
The increasing adoption of mobile phones have significantly shortened the time required for firefighting agents to be alerted of a starting wildfire. In less dense areas, limiting and minimizing this duration remains critical to preserve forest areas.
Pyrovision aims at providing the means to create a wildfire early detection system with state-of-the-art performances at minimal deployment costs.
Table of Contents
Getting started
Prerequisites
- Python 3.6 (or more recent)
- pip
Installation
You can install the latest release of the package using pypi as follows:
pip install pyrovision
or conda as follows:
conda install -c pyronear pyrovision
Usage
Python package
You can use the library like any other python package to detect wildfires as follows:
from pyrovision.datasets import OpenFire
dataset = OpenFire('./data', download=True)
Docker container
If you wish to deploy containerized environments, a Dockerfile is provided for you build a docker image:
docker build . -t <YOUR_IMAGE_TAG>
References
You are free to use any training script, but some are already provided for reference. In order to use them, install the specific requirements and check script options as follows:
pip install -r references/requirements.txt
python references/classification/train.py --help
You can then use the script to train tour model on one of our datasets:
Wildfire
Download Dataset from https://drive.google.com/file/d/1Y5IyBLA5xDMS1rBdVs-hsVNGQF3djaR1/view?usp=sharing
This dataset is protected by a password, please contact us at contact@pyronear.org
python train.py WildFireLght/ --model rexnet1_0x --lr 1e-3 -b 16 --epochs 20 --opt radam --sched onecycle --device 0
OpenFire
You can also use out opensource dataset without password
python train.py OpenFire/ --use-openfire --model rexnet1_0x --lr 1e-3 -b 16 --epochs 20 --opt radam --sched onecycle --device 0
Documentation
The full package documentation is available here for detailed specifications. The documentation was built with Sphinx using a theme provided by Read the Docs.
Contributing
Please refer to CONTRIBUTING
if you wish to contribute to this project.
Credits
This project is developed and maintained by the repo owner and volunteers from Data for Good.
License
Distributed under the AGPLv3 License. See LICENSE
for more information.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Hashes for pyrovision-0.1.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 00f68786a0eac5df9c980ebcfb16e5b3faf5c16b8e920237eba5668761474ca0 |
|
MD5 | 87d7e3b2b8cd22d48d8b81a5d2716ce5 |
|
BLAKE2b-256 | 6f7ad6cf624588ee9aaf5aa478d6977688f5e457ea2dfe15ef93e58de0a1642f |