Pytorch Toolbel 0.3.0
Pytorch Toolbel 0.3.0
This release has a huge set of new features, bugfixes and breaking changes. So be careful, when upgrading.
pip install pytorch-toolbelt==0.3.0
New features
Encoders
- HRNetV2
- DenseNets
- EfficientNet
Encoder
class haschange_input_channels
method to change number of channels in input image
New losses
BCELoss
with support ofignore_index
SoftBCELoss
(Label smoothing loss for binary case with support ofignore_index
)SoftCrossEntropyLoss
(Label smoothing loss for multiclass case with support ofignore_index
)
Catalyst goodies
- Online pseudolabeling callback
- Training signal annealing callback
Other
- New activation functions support in
ABN
block: Swish, Mish, HardSigmoid - New decoders (Unet, FPN, DeeplabV3, PPM) to simplify creation of segmentation models
CREDITS.md
to include all the references to code/articles. Existing list is definitely not complete, so feel free to make PR's- Object context block from OCNet
API changes
- Focal loss now supports normalized focal loss and reduced focal loss extensions.
- Optimize computation of pyramid weight matrix #34
- Default value
align_corners=False
inF.interpolate
when doing bilinear upsampling.
Bugfixes
- Fix missing call to batch normalization block in
FPNBottleneckBN
- Fix numerical stability for
DiceLoss
andJaccardLoss
whenlog_loss=True
- Fix numerical stability when computing normalized focal loss