🦅

DINOv2の特徴量をPCAで可視化する

2024/06/23に公開

最近はCLIP,SAM,DINOv2のような大規模学習済みのモデルの特徴量をうまいこと使って特定のタスクを解決するのがスタンダードになりつつある印象を受けます.その中でもDINOv2の可視化についてはいまいち分かりにくいのでメモ代わりにまとめておきます.
有志が作成してくれた以下のコードをDockerですぐ動かせるようにしただけです.

https://huggingface.co/spaces/RoundtTble/dinov2-pca/tree/main
https://github.com/facebookresearch/dinov2/issues/23#issuecomment-1546543086

example


思ったよりは綺麗に出力されないですね.モデルサイズや画像の選択に工夫の余地ありかもしれません.

usage

git clone https://github.com/watanko/dinov2-pca-visualization.git
cd dinov2-pca-visualization
docker compose run --rm --service-ports --build work
python app.py

https://github.com/watanko/dinov2-pca-visualization

Discussion