Skip to content

Segmentation Models - v0.3.1

Compare
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