Semantic-Segmentation-with-Transformer

Introduction

A repository of SegFormer model test

Architecture

Experimentals Results

Installation

Assuming a fresh Anaconda distribution with Python 3.8, you can install the dependencies with:

cd Semantic-Segmentation-with-Transformer
pip install -r requirements.txt

Training

python semantic_segmentation_main.py --api train

Evaluate

python semantic_segmentation_main.py --api eval

Inference

python semantic_segmentation_main.py --api infer

References

• https://github.com/NielsRogge/Transformers-Tutorials/blob/master/SegFormer/Fine_tune_SegFormer_on_custom_dataset.ipynb