MCPcopy Create free account

hub / github.com/andyzoujm/pixmix / functions

Functions73 in github.com/andyzoujm/pixmix

↓ 11 callersFunctionsample_level
(n)
pixmix_utils.py:57
↓ 9 callersMethodupdate
(self, val, n=1)
imagenet.py:738
↓ 6 callersFunctioncalib_err
(confidence, correct, p='2', beta=100)
calibration_tools.py:4
↓ 6 callersFunctionfloat_parameter
Helper function to scale `val` between 0 and maxval. Args: level: Level of the operation that will be between [0, `PARAMETER_MAX`]. maxval:
pixmix_utils.py:43
↓ 5 callersFunctionint_parameter
Helper function to scale `val` between 0 and maxval . Args: level: Level of the operation that will be between [0, `PARAMETER_MAX`]. maxval
pixmix_utils.py:29
↓ 5 callersFunctiontest
Evaluate network on given dataset.
cifar.py:249
↓ 4 callersFunctionvalidate
(val_loader, model, criterion, args, r=False, a=False, adv=None)
imagenet.py:641
↓ 3 callersFunctioninvert
(img)
pixmix_utils.py:188
↓ 2 callersFunctionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
imagenet.py:766
↓ 2 callersFunctionaugment_input
(image)
imagenet.py:195
↓ 2 callersFunctionaugment_input
(image)
cifar.py:192
↓ 2 callersFunctionaurra
(confidence, correct)
calibration_tools.py:37
↓ 2 callersMethoddisplay
(self, batch)
imagenet.py:755
↓ 2 callersFunctionevaluate_c
(model, normalize, args)
imagenet.py:475
↓ 2 callersFunctionget_ab
(beta)
pixmix_utils.py:160
↓ 2 callersFunctionget_net_results
(dataloader, net)
imagenet.py:783
↓ 2 callersFunctionmultiply
(img1, img2, beta)
pixmix_utils.py:175
↓ 1 callersMethod__init__
(self, name, fmt=':f')
imagenet.py:727
↓ 1 callersMethod_get_batch_fmtstr
(self, num_batches)
imagenet.py:760
↓ 1 callersFunctioncosine_annealing
(step, total_steps, lr_max, lr_min)
imagenet.py:411
↓ 1 callersFunctionget_lr
Compute learning rate according to cosine annealing schedule.
cifar.py:164
↓ 1 callersFunctionmain
()
imagenet.py:220
↓ 1 callersFunctionmain
()
cifar.py:339
↓ 1 callersFunctionmain_worker
(gpu, args)
imagenet.py:239
↓ 1 callersFunctionnormalize_l2
Expects x.shape == [N, C, H, W]
cifar.py:293
↓ 1 callersFunctionpixmix
(orig, mixing_pic, preprocess)
imagenet.py:173
↓ 1 callersFunctionpixmix
(orig, mixing_pic, preprocess)
cifar.py:170
↓ 1 callersMethodreset
(self)
imagenet.py:732
↓ 1 callersFunctionsave_checkpoint
(state, is_best, filename=os.path.join(args.save, "model.pth.tar"))
imagenet.py:719
↓ 1 callersFunctionscreen
(img1, img2, beta)
pixmix_utils.py:191
↓ 1 callersFunctionsoft_f1
(confidence, correct)
calibration_tools.py:44
↓ 1 callersFunctiontest_c
Evaluate network on given corrupted dataset.
cifar.py:270
↓ 1 callersFunctiontrain
(train_loader, model, criterion, optimizer, scheduler, epoch, args)
imagenet.py:593
↓ 1 callersFunctiontrain
Train for one epoch.
cifar.py:227
Method__del__
(self)
imagenet.py:168
Method__getitem__
(self, i)
imagenet.py:209
Method__getitem__
(self, index)
cifar.py:202
Method__getitem__
(self, i)
cifar.py:217
Method__init__
(self, root, *args, **kwargs)
imagenet.py:155
Method__init__
(self, dataset, mixing_set, preprocess)
imagenet.py:204
Method__init__
(self, num_batches, meters, prefix="")
imagenet.py:750
Method__init__
(self, file, transform)
cifar.py:198
Method__init__
(self, dataset, mixing_set, preprocess)
cifar.py:212
Method__init__
(self, epsilon, num_steps, step_size, grad_sign=True)
cifar.py:302
Method__len__
(self)
imagenet.py:215
Method__len__
(self)
cifar.py:206
Method__len__
(self)
cifar.py:223
Method__str__
(self)
imagenet.py:744
Functionadd
(img1, img2, beta)
pixmix_utils.py:169
Functionautocontrast
(pil_img, _)
pixmix_utils.py:61
Functionbrightness
(pil_img, level)
pixmix_utils.py:135
Functioncolor
(pil_img, level)
pixmix_utils.py:123
Functioncontrast
(pil_img, level)
pixmix_utils.py:129
Functiondarken_or_lighten
(img1, img2, beta)
pixmix_utils.py:205
Functionequalize
(pil_img, _)
pixmix_utils.py:65
Functionevaluate_c_bar
(model, normalize, args)
imagenet.py:534
Methodforward
:param model: the classifier's forward method :param bx: batch of images :param by: true labels :return: perturbed ba
cifar.py:309
Functionget_measures
(confidence, correct)
calibration_tools.py:93
Functionoverlay
(img1, img2, beta)
pixmix_utils.py:196
Functionposterize
(pil_img, level)
pixmix_utils.py:69
Functionprint_measures
(rms, aurra_metric, mad, sf1, method_name='Baseline')
calibration_tools.py:102
Functionrotate
(pil_img, level)
pixmix_utils.py:74
Functionsharpness
(pil_img, level)
pixmix_utils.py:141
Functionshear_x
(pil_img, level)
pixmix_utils.py:86
Functionshear_y
(pil_img, level)
pixmix_utils.py:95
Functionshow_calibration_results
(confidence, correct, method_name='Baseline')
calibration_tools.py:110
Functionsolarize
(pil_img, level)
pixmix_utils.py:81
Functionswap_channel
(img1, img2, beta)
pixmix_utils.py:212
Functiontranslate_x
(pil_img, level)
pixmix_utils.py:104
Functiontranslate_y
(pil_img, level)
pixmix_utils.py:113
Functiontune_temp
(logits, labels, binary_search=True, lower=0.2, upper=5.0, eps=0.0001)
calibration_tools.py:57
Functionwif
(id)
imagenet.py:401
Functionwif
(id)
cifar.py:385