OK-AI/ViT-Patch-PCA-Visualisation

Space

SSL ViT Patch PCA Visualiser

1

14 commits

1 linked in READMEs

updated Jun 12, 2026

See the code

README

SSL ViT Patch PCA Visualiser

Visualise the patch-level feature space of ViT-S/16 models from the OK-AI ImageNet-1K SSL collection.

Upload any image and the app runs it through all three models, applies PCA to the patch token representations, and renders the first 3 principal components as an RGB image.

gradio