README.md

---
title: SegFormer (ADE20k) in TensorFlow
emoji: 🏃
colorFrom: indigo
colorTo: gray
sdk: gradio
sdk_version: 3.0.22
app_file: app.py
pinned: false
license: apache-2.0
---

This space hosts the [SegFormer model](https://arxiv.org/abs/2105.15203) in TensorFlow. This model was fine-tuned on the [ADE20k dataset](http://groups.csail.mit.edu/vision/datasets/ADE20K/). To know more about the checkpoint used in this space, refer to the model card
[here](https://huggingface.co/nvidia/segformer-b5-finetuned-ade-640-640).

Please note that since the model was fine-tuned on the ADE20k dataset, the model is expected to provide best results for images
belonging to scene categories. For an overview of the dataset, refer to its [homepage](http://groups.csail.mit.edu/vision/datasets/ADE20K/).