SegFormer vs. YOLOS: Compared and Contrasted

Models

SegFormer is a computer vision framework used in semantic segmentation tasks, implemented with transformers.

YOLOS looks at patches of an image to to form "patch tokens", which are used in place of the traditional wordpiece tokens in NLP.

Model Type

Semantic Segmentation

Object Detection

Model Features

Item 1 Info

Item 2 Info

Architecture

Transformers

Transformer, YOLO

Frameworks

PyTorch

Annotation Format

Instance Segmentation

GitHub

GitHub Stars

2.2k+

812+

License

NVIDIA Source Code

MIT

Paper

Training Notebook

Deploy Model