Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/andyzoujm/pixmix
/ functions
Functions
73 in github.com/andyzoujm/pixmix
⨍
Functions
73
◇
Types & classes
7
↓ 11 callers
Function
sample_level
(n)
pixmix_utils.py:57
↓ 9 callers
Method
update
(self, val, n=1)
imagenet.py:738
↓ 6 callers
Function
calib_err
(confidence, correct, p='2', beta=100)
calibration_tools.py:4
↓ 6 callers
Function
float_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 callers
Function
int_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 callers
Function
test
Evaluate network on given dataset.
cifar.py:249
↓ 4 callers
Function
validate
(val_loader, model, criterion, args, r=False, a=False, adv=None)
imagenet.py:641
↓ 3 callers
Function
invert
(img)
pixmix_utils.py:188
↓ 2 callers
Function
accuracy
Computes the accuracy over the k top predictions for the specified values of k
imagenet.py:766
↓ 2 callers
Function
augment_input
(image)
imagenet.py:195
↓ 2 callers
Function
augment_input
(image)
cifar.py:192
↓ 2 callers
Function
aurra
(confidence, correct)
calibration_tools.py:37
↓ 2 callers
Method
display
(self, batch)
imagenet.py:755
↓ 2 callers
Function
evaluate_c
(model, normalize, args)
imagenet.py:475
↓ 2 callers
Function
get_ab
(beta)
pixmix_utils.py:160
↓ 2 callers
Function
get_net_results
(dataloader, net)
imagenet.py:783
↓ 2 callers
Function
multiply
(img1, img2, beta)
pixmix_utils.py:175
↓ 1 callers
Method
__init__
(self, name, fmt=':f')
imagenet.py:727
↓ 1 callers
Method
_get_batch_fmtstr
(self, num_batches)
imagenet.py:760
↓ 1 callers
Function
cosine_annealing
(step, total_steps, lr_max, lr_min)
imagenet.py:411
↓ 1 callers
Function
get_lr
Compute learning rate according to cosine annealing schedule.
cifar.py:164
↓ 1 callers
Function
main
()
imagenet.py:220
↓ 1 callers
Function
main
()
cifar.py:339
↓ 1 callers
Function
main_worker
(gpu, args)
imagenet.py:239
↓ 1 callers
Function
normalize_l2
Expects x.shape == [N, C, H, W]
cifar.py:293
↓ 1 callers
Function
pixmix
(orig, mixing_pic, preprocess)
imagenet.py:173
↓ 1 callers
Function
pixmix
(orig, mixing_pic, preprocess)
cifar.py:170
↓ 1 callers
Method
reset
(self)
imagenet.py:732
↓ 1 callers
Function
save_checkpoint
(state, is_best, filename=os.path.join(args.save, "model.pth.tar"))
imagenet.py:719
↓ 1 callers
Function
screen
(img1, img2, beta)
pixmix_utils.py:191
↓ 1 callers
Function
soft_f1
(confidence, correct)
calibration_tools.py:44
↓ 1 callers
Function
test_c
Evaluate network on given corrupted dataset.
cifar.py:270
↓ 1 callers
Function
train
(train_loader, model, criterion, optimizer, scheduler, epoch, args)
imagenet.py:593
↓ 1 callers
Function
train
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
Function
add
(img1, img2, beta)
pixmix_utils.py:169
Function
autocontrast
(pil_img, _)
pixmix_utils.py:61
Function
brightness
(pil_img, level)
pixmix_utils.py:135
Function
color
(pil_img, level)
pixmix_utils.py:123
Function
contrast
(pil_img, level)
pixmix_utils.py:129
Function
darken_or_lighten
(img1, img2, beta)
pixmix_utils.py:205
Function
equalize
(pil_img, _)
pixmix_utils.py:65
Function
evaluate_c_bar
(model, normalize, args)
imagenet.py:534
Method
forward
:param model: the classifier's forward method :param bx: batch of images :param by: true labels :return: perturbed ba
cifar.py:309
Function
get_measures
(confidence, correct)
calibration_tools.py:93
Function
overlay
(img1, img2, beta)
pixmix_utils.py:196
Function
posterize
(pil_img, level)
pixmix_utils.py:69
Function
print_measures
(rms, aurra_metric, mad, sf1, method_name='Baseline')
calibration_tools.py:102
Function
rotate
(pil_img, level)
pixmix_utils.py:74
Function
sharpness
(pil_img, level)
pixmix_utils.py:141
Function
shear_x
(pil_img, level)
pixmix_utils.py:86
Function
shear_y
(pil_img, level)
pixmix_utils.py:95
Function
show_calibration_results
(confidence, correct, method_name='Baseline')
calibration_tools.py:110
Function
solarize
(pil_img, level)
pixmix_utils.py:81
Function
swap_channel
(img1, img2, beta)
pixmix_utils.py:212
Function
translate_x
(pil_img, level)
pixmix_utils.py:104
Function
translate_y
(pil_img, level)
pixmix_utils.py:113
Function
tune_temp
(logits, labels, binary_search=True, lower=0.2, upper=5.0, eps=0.0001)
calibration_tools.py:57
Function
wif
(id)
imagenet.py:401
Function
wif
(id)
cifar.py:385