AI engineering glossary · Computer vision

What is vision transformer?

Vision transformer (also called ViT): A model that splits an image into fixed-size patches, treats each patch as a token and processes them with a transformer encoder using self-attention. With enough data it matches or beats CNNs on classification.

Level: advancedTopic: Computer vision

Vision transformer: an example

A 224 x 224 image cut into 16 x 16 patches becomes a sequence of 196 tokens.

Related terms

More Computer vision terms

Where you learn it

Vision transformer is part of the AI Engineering Program, which teaches it with notes, a hands-on lab and an exam. The first lesson is free.

Before you go

Take the cheat sheet with you

One free PDF: the key AI-engineering terms, formulas and Python idioms. No spam; unsubscribe any time.