SegFormer

SegFormer

Description

SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers. website

Model

Model	Download
SegFormer	14.4 MB

• Convert to onnx in mmsegmentation

python tools/pytorch2onnx.py 
          local_configs/segformer/B0/segformer.b0.512x512.ade.160k.py
          --checkpoint onnx/segformer.b0.512x512.ade.160k.pth
          --show
          --verify
          --output-file onnx/segformer.onnx
          --opset-version 12
          --shape 512 512

Dataset

ADE20K

References

• SegFormer

License

Apache-2.0