Transformer-based semantic segmentation model by Nvidia
Note https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/examples/segformer_inference_pretrained.ipynb