Skip to content

Dataset and models for binary classification
Compare
Choose a tag to compare
@frgfm frgfm released this 28 Oct 19:22
v0.1.0
f4ddb1f
This commit was created on GitHub.com and signed with GitHub鈥檚 verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

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

Release 0.1.0

Datasets

Wildfire detection visual datasets
New

Models

Common model architectures with pretrained parameters for wildfire detection
New

  • Add pretrained ResNet (#37)(#41)
  • Add pretrained DenseNet (#41)
  • Add pretrained MobileNetV2 (#41)

Documentation

Documentation for users, contributors and developers of pyronear

  • Add docs for pyronear.datasets (#11)
  • Add docs for pyronear.models (#37)
  • Add docs for pyronear.nn (#41)
  • Add docs for pyronear.utils (#7)
  • Add contribution guidelines (#1)(#8)(#29)

Tests

Verifications of the package well-being before release

  • Add test for pyronear.datasets (#11)(#16)(#39)
  • Add test for pyronear.models (#37)(#41)
  • Add test for pyronear.nn (#41)
  • Add test for pyronear.utils(#17)

Reference scripts

Reference training scripts used to obtain state dictionaries of pretrained models

  • Add training script with pytorch and torchvision dependencies (#23)(#30)(#32)(#41)
  • Add training script with fastai dependency (#23)(#30)(#32)(#41)

Checkpoint for the first Pyroneal Model

This first model is a rexnet1_0x with a cutom head using AdaptiveConcatPool2d

Assets 3