/SegFormer_Segmentation

Semantic Segmentation with SegFormer on Drone Dataset.

Primary LanguageJupyter Notebook

SegFormer_Segmentation

Semantic Segmentation with SegFormer on Drone Dataset.

You can check out the blog on Medium

You can also try out the model with your own custom data on HuggingFace Spaces here

The model is also available on HuggingFace Model Hub here

Citations and References

SegFormer
@article{xie2021segformer,
  title={SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers},
  author={Xie, Enze and Wang, Wenhai and Yu, Zhiding and Anandkumar, Anima and Alvarez, Jose M and Luo, Ping},
  journal={arXiv preprint arXiv:2105.15203},
  year={2021}
}

Drone Dataset
http://dronedataset.icg.tugraz.at/