Edit model card

pretrained model

  • https://huggingface.co/nvidia/mit-b0
  • SegFormer (b0-sized) encoder pre-trained-only
  • SegFormer encoder fine-tuned on Imagenet-1k. It was introduced in the paper SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers by Xie et al. and first released in this repository.

training-set

training-arguments

  • channels : rgb
  • batch : 8
  • epochs : 8
  • learning-rate : 5e-6
  • GPU : T4

results on test-set

Downloads last month
30

Space using alanoix/segformer_b0_flair_one 1