Skip to content

Segmentation Models - v0.0.2

Compare
Choose a tag to compare
@qubvel qubvel released this 19 Sep 11:27
f4912ce

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!