search

kaiko-ai / eva

Evaluation framework for oncology foundation models (FMs)
https://kaiko-ai.github.io/eva/
Apache License 2.0
50 stars 2 forks source link

Add visiontransformer-wrapper for segmentation tasks #550

Open roman807 opened 1 week ago

roman807 commented 1 week ago

forward pass of torch/hub/facebookresearch_dino_main/vision_transformer.py returns only class token (see here), however, for segmentation tasks we need a list of embeddings from all patches.