Downloads · 30 days
15
6% of all-time downloads
ananthu-aniraj/pdiscoformer_cub_k_4
pdiscoformer_cub_k_4 is a image classification model from ananthu-aniraj. Use it when you need a label for an image. The card lists the license as mit.
PdiscoFormer (Vit-base-dinov2-reg4) trained on CUB with K (number of unsupervised parts to discover) set to a value of 4.
Downloads · 30 days
15
6% of all-time downloads
All-time downloads
252
Public
Parameters
86.7M
347 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors347 MB · 100%
From the Hugging Face model README
PdiscoFormer (Vit-base-dinov2-reg4) trained on CUB with K (number of unsupervised parts to discover) set to a value of 4.
PdiscoFormer is a novel method for unsupervised part discovery using self-supervised Vision Transformers which achieves state-of-the-art results for this task, both qualitatively and quantitatively. The code can be found in the following repository: https://github.com/ananthu-aniraj/pdiscoformer
@misc{aniraj2024pdiscoformerrelaxingdiscoveryconstraints,
title={PDiscoFormer: Relaxing Part Discovery Constraints with Vision Transformers},
author={Ananthu Aniraj and Cassio F. Dantas and Dino Ienco and Diego Marcos},
year={2024},
eprint={2407.04538},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2407.04538},
}