Edit model card

PdiscoFormer PartImageNet OOD Model (K=8)

PdiscoFormer (Vit-base-dinov2-reg4) trained on PartImageNet OOD with K (number of unsupervised parts to discover) set to a value of 8.

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

BibTex entry and citation info

@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}, 
}
Downloads last month
10
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Examples
Unable to determine this model's library. Check the docs .

Model tree for ananthu-aniraj/pdiscoformer_part_imagenet_ood_k_8

Finetuned
(17)
this model

Space using ananthu-aniraj/pdiscoformer_part_imagenet_ood_k_8 1

Collection including ananthu-aniraj/pdiscoformer_part_imagenet_ood_k_8