Downloads · 30 days
12
17% of all-time downloads
ananthu-aniraj/pdiscoformer_flowers_k_2
pdiscoformer_flowers_k_2 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 Flowers102 with K (number of unsupervised parts to discover) set to a value of 2.
Downloads · 30 days
12
17% of all-time downloads
All-time downloads
69
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
PdiscoFormer (Vit-base-dinov2-reg4) trained on Flowers102 with K (number of unsupervised parts to discover) set to a value of 2.
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},
}