├── .gitignore ├── LICENSE ├── README.md ├── argument_parser_train.py ├── data_sets ├── __init__.py ├── celeba.py ├── classes_mapping_imagenet.py ├── fg_bird_dataset.py ├── flowers102seg.py ├── imagenet_with_ood_eval.py ├── part_imagenet.py ├── plantnet.py └── train_test_split_pimagenet_ood.txt ├── engine ├── __init__.py ├── distributed_trainer_pdisco.py ├── eval_fg_bg.py ├── eval_interpretability_nmi_ari_keypoint.py └── losses │ ├── __init__.py │ ├── concentration_loss.py │ ├── enforced_presence_loss.py │ ├── equivarance_loss.py │ ├── orthogonality_loss.py │ ├── pixel_wise_entropy_loss.py │ ├── presence_loss.py │ └── total_variation.py ├── environment.yml ├── evaluate_parts.py ├── evaluation_instructions.md ├── hubconf.py ├── inference_benchmark_models.py ├── layers ├── __init__.py ├── independent_mlp.py └── transformer_layers.py ├── load_dataset.py ├── load_losses.py ├── load_model.py ├── model_zoo.md ├── models ├── __init__.py ├── individual_landmark_convnext.py ├── individual_landmark_resnet.py ├── individual_landmark_vit.py └── vit_baseline.py ├── prepare_partimagenet_ood.py ├── torch_hub ├── __init__.py └── load_pretrained_models.py ├── train_net.py ├── training_instructions.md └── utils ├── __init__.py ├── data_utils ├── __init__.py ├── class_balanced_distributed_sampler.py ├── class_balanced_sampler.py ├── dataset_utils.py ├── reversible_affine_transform.py └── transform_utils.py ├── get_landmark_coordinates.py ├── misc_utils.py ├── training_utils ├── __init__.py ├── ddp_utils.py ├── engine_utils.py ├── linear_lr_scheduler.py ├── optimizer_params.py ├── scheduler_params.py └── snapshot_class.py ├── visualize_att_maps.py └── wandb_params.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/README.md -------------------------------------------------------------------------------- /argument_parser_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/argument_parser_train.py -------------------------------------------------------------------------------- /data_sets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/__init__.py -------------------------------------------------------------------------------- /data_sets/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/celeba.py -------------------------------------------------------------------------------- /data_sets/classes_mapping_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/classes_mapping_imagenet.py -------------------------------------------------------------------------------- /data_sets/fg_bird_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/fg_bird_dataset.py -------------------------------------------------------------------------------- /data_sets/flowers102seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/flowers102seg.py -------------------------------------------------------------------------------- /data_sets/imagenet_with_ood_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/imagenet_with_ood_eval.py -------------------------------------------------------------------------------- /data_sets/part_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/part_imagenet.py -------------------------------------------------------------------------------- /data_sets/plantnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/plantnet.py -------------------------------------------------------------------------------- /data_sets/train_test_split_pimagenet_ood.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/data_sets/train_test_split_pimagenet_ood.txt -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/__init__.py -------------------------------------------------------------------------------- /engine/distributed_trainer_pdisco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/distributed_trainer_pdisco.py -------------------------------------------------------------------------------- /engine/eval_fg_bg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/eval_fg_bg.py -------------------------------------------------------------------------------- /engine/eval_interpretability_nmi_ari_keypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/eval_interpretability_nmi_ari_keypoint.py -------------------------------------------------------------------------------- /engine/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/__init__.py -------------------------------------------------------------------------------- /engine/losses/concentration_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/concentration_loss.py -------------------------------------------------------------------------------- /engine/losses/enforced_presence_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/enforced_presence_loss.py -------------------------------------------------------------------------------- /engine/losses/equivarance_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/equivarance_loss.py -------------------------------------------------------------------------------- /engine/losses/orthogonality_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/orthogonality_loss.py -------------------------------------------------------------------------------- /engine/losses/pixel_wise_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/pixel_wise_entropy_loss.py -------------------------------------------------------------------------------- /engine/losses/presence_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/presence_loss.py -------------------------------------------------------------------------------- /engine/losses/total_variation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/engine/losses/total_variation.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluate_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/evaluate_parts.py -------------------------------------------------------------------------------- /evaluation_instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/evaluation_instructions.md -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/hubconf.py -------------------------------------------------------------------------------- /inference_benchmark_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/inference_benchmark_models.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/independent_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/layers/independent_mlp.py -------------------------------------------------------------------------------- /layers/transformer_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/layers/transformer_layers.py -------------------------------------------------------------------------------- /load_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/load_dataset.py -------------------------------------------------------------------------------- /load_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/load_losses.py -------------------------------------------------------------------------------- /load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/load_model.py -------------------------------------------------------------------------------- /model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/model_zoo.md -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/individual_landmark_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/models/individual_landmark_convnext.py -------------------------------------------------------------------------------- /models/individual_landmark_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/models/individual_landmark_resnet.py -------------------------------------------------------------------------------- /models/individual_landmark_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/models/individual_landmark_vit.py -------------------------------------------------------------------------------- /models/vit_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/models/vit_baseline.py -------------------------------------------------------------------------------- /prepare_partimagenet_ood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/prepare_partimagenet_ood.py -------------------------------------------------------------------------------- /torch_hub/__init__.py: -------------------------------------------------------------------------------- 1 | from .load_pretrained_models import * -------------------------------------------------------------------------------- /torch_hub/load_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/torch_hub/load_pretrained_models.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/train_net.py -------------------------------------------------------------------------------- /training_instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/training_instructions.md -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/data_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/__init__.py -------------------------------------------------------------------------------- /utils/data_utils/class_balanced_distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/class_balanced_distributed_sampler.py -------------------------------------------------------------------------------- /utils/data_utils/class_balanced_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/class_balanced_sampler.py -------------------------------------------------------------------------------- /utils/data_utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/dataset_utils.py -------------------------------------------------------------------------------- /utils/data_utils/reversible_affine_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/reversible_affine_transform.py -------------------------------------------------------------------------------- /utils/data_utils/transform_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/data_utils/transform_utils.py -------------------------------------------------------------------------------- /utils/get_landmark_coordinates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/get_landmark_coordinates.py -------------------------------------------------------------------------------- /utils/misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/misc_utils.py -------------------------------------------------------------------------------- /utils/training_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/__init__.py -------------------------------------------------------------------------------- /utils/training_utils/ddp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/ddp_utils.py -------------------------------------------------------------------------------- /utils/training_utils/engine_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/engine_utils.py -------------------------------------------------------------------------------- /utils/training_utils/linear_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/linear_lr_scheduler.py -------------------------------------------------------------------------------- /utils/training_utils/optimizer_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/optimizer_params.py -------------------------------------------------------------------------------- /utils/training_utils/scheduler_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/scheduler_params.py -------------------------------------------------------------------------------- /utils/training_utils/snapshot_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/training_utils/snapshot_class.py -------------------------------------------------------------------------------- /utils/visualize_att_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/visualize_att_maps.py -------------------------------------------------------------------------------- /utils/wandb_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ananthu-aniraj/pdiscoformer/HEAD/utils/wandb_params.py --------------------------------------------------------------------------------