Skip to content

Releases: qubvel-org/segmentation_models.pytorch

Segmentation Models - v0.1.0

09 Dec 13:59
Compare
Choose a tag to compare

Updates

  1. New backbones (mobilenet, efficientnet, inception)
  2. depth and in_channels options for all models
  3. Auxiliary classification output

Note!

Model architectures have been changed, use previous versions for weights compatibility!

Segmentation Models - v0.0.3

28 Sep 18:43
127f930
Compare
Choose a tag to compare

Updates

  • Conv2D Initialization
    • kaiming_normal -> kaiming_uniform;
    • fan_out -> fan_in;
    • bias -> 0
  • package dependencies

Segmentation Models - v0.0.2

19 Sep 11:27
f4912ce
Compare
Choose a tag to compare

Updates

  • New backbones:
    • resnext50_32x4d
    • resnext101_32x8d
    • resnext101_32x16d
    • resnext101_32x32d
    • resnext101_32x48d
  • Unet scSE attention block (optional)
  • torchvision version update
  • get_preprocessing_params function

Thanks @laol777 for contribution!