Computer vision library for wildfire detection π² Deep learning models in PyTorch & ONNX for inference on edge devices (e.g. Raspberry Pi)
APACHE-2.0 License
Published by frgfm over 2 years ago
This release brings a big performance leap to classification models while providing easier contribution and deployment tools.
Note: pyrovision 0.2.0 requires PyTorch 1.11 and torchvision 0.12 or newer.
In order to showcase pyrovision added value, a short Gradio demo app was added to the repo!
You can try it out live on HF Spaces over π here
Thanks to the great FastAPI by SebastiΓ‘n RamΓrez (@tiangolo
), we have added a small template to deploy your own vision API π
Check out in the folder ./api
how to get yours running!
Contributions are important in open source projects and we're happy to announce that our model checkpoints are now available on HF Hub. See it as a github for data files (checkpoints for instance) π
You can load a model from the hub with two lines:
from pyrovision.models.utils import model_from_hf_hub
model = model_from_hf_hub("pyronear/rexnet1_0x")
If you upload your model as well, you can load it from the hub by changing "pyronear/rexnet1_0x" into "hf_user/model_repo"
You might have noticed that our first version of OpenFire started to have a lot of URLs failing. We've scaled up our queries to retrieve public images and checked them manually to produce a new updated version of ~7000 train images and ~800 validation images. The dataset was used to train the new model checkpoints!
The dataset is also available on HF datasets: https://huggingface.co/datasets/pyronear/openfire
The following modules and features were deprecated:
pyrovision.nn
pyrovision.datasets.wildfire
& pyrovision.datasets.video_utils
pyrovision.models.densenet
& pyrovision.models.ssresnet
Full Changelog: https://github.com/pyronear/pyro-vision/compare/v0.1.2...v0.2.0
Published by frgfm over 3 years ago
This release expands compatibility to latest releases of pytorch and torchvision.
Note: pyrovision 0.1.2 requires PyTorch 1.8 and torchvision 0.9 or newer.
Published by frgfm over 3 years ago
This release adds a video dataset and multiple pretrained models.
Note: pyrovision 0.1.1 requires PyTorch 1.2 and torchvision 0.4.0 or newer.
Wildfire detection visual datasets
New
Improvements
Common model architectures with pretrained parameters for wildfire detection
New
Improvements
Scripts for training models
New
Improvements
Verification of the package well-being before release
Improvements
Online resources for potential users
New
Improvements
Fixes
Other tools and implementations
New
Improvements
Fixes
Published by frgfm almost 5 years ago
This release adds a binary classification dataset and several model architectures for wildfire detection.
Note: pyronear 0.1.0 requires PyTorch 1.2 and torchvision 0.4.0 or newer
Wildfire detection visual datasets
New
OpenFire
dataset for image classification (#11)(#15)(#18)(#21)(#32)(#39)Common model architectures with pretrained parameters for wildfire detection
New
ResNet
(#37)(#41)DenseNet
(#41)MobileNetV2
(#41)Documentation for users, contributors and developers of pyronear
pyronear.datasets
(#11)pyronear.models
(#37)pyronear.nn
(#41)pyronear.utils
(#7)Verifications of the package well-being before release
pyronear.datasets
(#11)(#16)(#39)pyronear.models
(#37)(#41)pyronear.nn
(#41)pyronear.utils
(#17)Reference training scripts used to obtain state dictionaries of pretrained models
pytorch
and torchvision
dependencies (#23)(#30)(#32)(#41)fastai
dependency (#23)(#30)(#32)(#41)This first model is a rexnet1_0x with a cutom head using AdaptiveConcatPool2d