Skip to content

Segmentation Models - v0.3.1

Choose a tag to compare

@qubvel qubvel released this 30 Nov 12:31

Updates

  • Added Mix Vision Transformer encoder from SegFormer [official code] [paper]. Use argument encoder_name="mit_b0" (or mit_b1..b5) to create a model.
  • Minor typo fixes and docs updates