Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/diffusion-classifier/diffusion-classifier
/ functions
Functions
88 in github.com/diffusion-classifier/diffusion-classifier
⨍
Functions
88
◇
Types & classes
15
↓ 4 callers
Function
eval_prob
(unet, latent, cond_emb, scheduler, args, all_noise=None)
run_winoground.py:19
↓ 4 callers
Function
get_target_dataset
Get the torchvision dataset that we want to use. If the dataset doesn't have a class_to_idx attribute, we add it. Also add a file-to-class map
diffusion/datasets.py:85
↓ 3 callers
Function
get_transform
(interpolation=InterpolationMode.BICUBIC, size=512)
eval_prob_adaptive.py:28
↓ 3 callers
Method
name
(self)
diffusion/dataset/imagenet.py:63
↓ 2 callers
Function
get_classnames
(source)
diffusion/dataset/imagenet_classnames.py:200
↓ 2 callers
Function
get_scheduler_config
(args)
diffusion/models.py:36
↓ 2 callers
Function
get_sd_model
(args)
diffusion/models.py:16
↓ 2 callers
Method
get_test_path
(self)
diffusion/dataset/imagenet.py:48
↓ 2 callers
Function
image_correct
(result)
run_winoground.py:237
↓ 2 callers
Method
path_to_cls
(self, path)
diffusion/datasets.py:73
↓ 2 callers
Function
project_logits
(logits, class_sublist_mask, device)
diffusion/dataset/imagenet.py:89
↓ 2 callers
Function
text_correct
(result)
run_winoground.py:232
↓ 1 callers
Method
__getitem__
(self, idx)
diffusion/dataset/common.py:123
↓ 1 callers
Method
__init__
( self, root=DATASET_ROOT, transform=lambda x: x, )
diffusion/datasets.py:58
↓ 1 callers
Method
__init__
(self, preprocess, location=os.path.expanduser('~/data'), b
diffusion/dataset/imagenet.py:9
↓ 1 callers
Method
__init__
(self, *args, **kwargs)
diffusion/dataset/objectnet.py:76
↓ 1 callers
Method
__init__
(self, indices)
diffusion/dataset/common.py:13
↓ 1 callers
Function
accuracy
(logits, targets, img_paths, args)
diffusion/dataset/objectnet.py:128
↓ 1 callers
Function
center_crop_arr
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
eval_prob_dit.py:26
↓ 1 callers
Function
cls_from_dir
(name)
scripts/print_dit_acc.py:50
↓ 1 callers
Function
conf_interval
(p, n, z=1.96)
run_winoground.py:244
↓ 1 callers
Function
eval_error
(unet, scheduler, latent, all_noise, ts, noise_idxs, text_embeds, text_embed_idxs, batch_size=3
eval_prob_adaptive.py:98
↓ 1 callers
Function
eval_prob
(unet, latent, cond_emb, diffusion, args, all_noise=None)
eval_prob_dit.py:89
↓ 1 callers
Function
eval_prob_adaptive
(unet, latent, text_embeds, scheduler, args, latent_size=64, all_noise=None)
eval_prob_adaptive.py:44
↓ 1 callers
Function
filter_imageneta
(files, labels)
scripts/print_dit_acc.py:12
↓ 1 callers
Method
get_class_sublist_and_mask
(self)
diffusion/dataset/imagenet.py:103
↓ 1 callers
Method
get_class_sublist_and_mask
(self)
diffusion/dataset/imagenet.py:113
↓ 1 callers
Function
get_classes_templates
Get a template for the text prompt. Args: dataset: dataset name Returns: template: template for the text prompt
diffusion/utils.py:47
↓ 1 callers
Function
get_features
(is_train, image_encoder, dataset, device)
diffusion/dataset/common.py:90
↓ 1 callers
Function
get_features_helper
(image_encoder, dataloader, device)
diffusion/dataset/common.py:60
↓ 1 callers
Function
get_formatstr
(n)
diffusion/utils.py:38
↓ 1 callers
Function
get_imageneta_files
(args, labels, test_subdir=True)
scripts/print_dit_acc.py:18
↓ 1 callers
Function
get_metadata
()
diffusion/dataset/objectnet.py:15
↓ 1 callers
Function
get_shared_files
(args, labels, test_subdir=True, imageneta=False)
scripts/print_dit_acc.py:33
↓ 1 callers
Method
get_test_dataset
(self)
diffusion/dataset/imagenet.py:60
↓ 1 callers
Method
get_test_dataset
(self)
diffusion/dataset/objectnet.py:99
↓ 1 callers
Method
get_test_sampler
(self)
diffusion/dataset/imagenet.py:57
↓ 1 callers
Method
get_train_sampler
(self)
diffusion/dataset/imagenet.py:54
↓ 1 callers
Function
get_transform
(image_size)
eval_prob_dit.py:51
↓ 1 callers
Function
group_correct
(result)
run_winoground.py:241
↓ 1 callers
Function
is_correct
(labels, folder, subdirs, file, test_subdir=True, idx_map=None)
scripts/print_dit_acc.py:58
↓ 1 callers
Function
main
()
eval_prob_adaptive.py:125
↓ 1 callers
Function
main
()
run_winoground.py:75
↓ 1 callers
Function
main
()
eval_prob_dit.py:173
↓ 1 callers
Function
main
()
scripts/print_acc.py:16
↓ 1 callers
Function
main
()
scripts/print_dit_acc.py:89
↓ 1 callers
Function
maybe_dictionarize
(batch)
diffusion/dataset/common.py:46
↓ 1 callers
Function
mean_per_class_acc
(correct, labels)
scripts/print_acc.py:8
↓ 1 callers
Function
normal_kl
Compute the KL divergence between two gaussians. Shapes are automatically broadcasted, so batches can be compared to scalars, among other
eval_prob_dit.py:60
↓ 1 callers
Method
populate_test
(self)
diffusion/dataset/imagenet.py:39
↓ 1 callers
Method
populate_train
(self)
diffusion/dataset/imagenet.py:24
↓ 1 callers
Function
save_noise
()
scripts/save_noise.py:5
Method
__getitem__
(self, index)
diffusion/datasets.py:77
Method
__getitem__
(self, index)
diffusion/dataset/objectnet.py:66
Method
__getitem__
(self, index)
diffusion/dataset/common.py:37
Method
__init__
( self, variant="matched-frequency", root=DATASET_ROOT, transf
diffusion/datasets.py:45
Method
__init__
(self, *args, **kwargs)
diffusion/dataset/imagenet.py:98
Method
__init__
(self, label_map, path, transform)
diffusion/dataset/objectnet.py:54
Method
__init__
(self, path, transform, flip_label_prob=0.0)
diffusion/dataset/common.py:23
Method
__init__
(self, is_train, image_encoder, dataset, device)
diffusion/dataset/common.py:117
Method
__iter__
(self)
diffusion/dataset/common.py:16
Method
__len__
(self)
diffusion/datasets.py:70
Method
__len__
(self)
diffusion/dataset/objectnet.py:63
Method
__len__
(self)
diffusion/dataset/common.py:19
Method
__len__
(self)
diffusion/dataset/common.py:120
Function
_convert_image_to_rgb
(image)
eval_prob_adaptive.py:24
Function
_convert_image_to_rgb
(image)
eval_prob_dit.py:47
Method
accuracy
(self, logits, targets, img_paths, args)
diffusion/dataset/objectnet.py:157
Function
center_crop_resize
(img, interpolation=InterpolationMode.BILINEAR)
eval_prob_adaptive.py:39
Function
crop
(img)
diffusion/dataset/objectnet.py:45
Function
get_dataloader
(dataset, is_train, args, image_encoder=None)
diffusion/dataset/common.py:129
Function
get_datetimestr
()
diffusion/utils.py:33
Method
get_test_dataset
(self)
diffusion/dataset/imagenet.py:68
Method
get_test_dataset
(self)
diffusion/dataset/objectnet.py:147
Method
get_test_sampler
(self)
diffusion/dataset/imagenet.py:119
Method
get_test_sampler
(self)
diffusion/dataset/objectnet.py:140
Method
get_train_sampler
(self)
diffusion/dataset/imagenet.py:73
Function
is_correct_wrapper
(args)
scripts/print_dit_acc.py:85
Method
populate_train
(self)
diffusion/dataset/imagenet.py:106
Method
populate_train
(self)
diffusion/dataset/imagenet.py:116
Method
populate_train
(self)
diffusion/dataset/objectnet.py:96
Method
project_labels
(self, labels, device)
diffusion/dataset/imagenet.py:127
Method
project_labels
(self, labels, device)
diffusion/dataset/objectnet.py:150
Method
project_logits
(self, logits, device)
diffusion/dataset/imagenet.py:109
Method
project_logits
(self, logits, device)
diffusion/dataset/imagenet.py:131
Method
project_logits
(self, logits, device)
diffusion/dataset/objectnet.py:105
Function
save_latent
(vae, latent, path, scaling=1 / 0.18125)
diffusion/utils.py:17
Method
scatter_weights
(self, weights)
diffusion/dataset/objectnet.py:117