Functions — github.com/Vincent-ZHQ/UniS-MMC

CodeHub MCPcopy Create free account

hub / github.com/Vincent-ZHQ/UniS-MMC / functions

Functions109 in github.com/Vincent-ZHQ/UniS-MMC

⨍ Functions 109◇ Types & classes 15

↓ 6 callersFunction_check_args_tf

src/augment_ops.py:67

↓ 6 callersFunction_randomly_negate

src/augment_ops.py:75

↓ 4 callersFunctiondict_to_str

src/functions.py:21

↓ 3 callersFunctioncollect_metrics

(dataset, y_true, y_pred)

src/metrics.py:5

↓ 3 callersMethodmmc_2

(self, f0, f1, p0, p1, l)

model/model.py:86

↓ 2 callersMethod__init__

model/model.py:29

↓ 2 callersMethod_erase

(self, img, chan, img_h, img_w, dtype)

src/augment_ops.py:695

↓ 2 callersFunctionformat_txt_file

data/dataloader.py:71

↓ 2 callersMethodget_tokenizer

model/TextEncoder.py:50

↓ 2 callersMethodinfer

(self, text=None, image=None, data_list=None)

model/model.py:82

↓ 2 callersFunctionremove_tags

data/dataloader.py:28

↓ 1 callersFunctionMMDataLoader

data/dataloader.py:174

↓ 1 callersMethodSupMMConLoss

(self, feature_a, feature_b, labels, temperature=0.07)

model/model.py:144

↓ 1 callersMethodUnSupMMConLoss

(self, feature_a, feature_b, temperature=0.07)

model/model.py:162

↓ 1 callersMethodUniSMMConLoss

(self, feature_a, feature_b, predict_a, predict_b, labels, temperature=0.07)

model/model.py:101

↓ 1 callersFunction_get_pixels

(per_pixel, rand_color, patch_size, dtype=torch.float32, device='cuda')

src/augment_ops.py:560

↓ 1 callersFunction_interpolation

src/augment_ops.py:60

↓ 1 callersFunction_select_rand_weights

(weight_idx=0, transforms=None)

src/augment_ops.py:429

↓ 1 callersFunctioncount_parameters

src/functions.py:28

↓ 1 callersMethodforward

(self, text=None, image=None, data_list=None, label=None, infer=False)

model/model.py:41

↓ 1 callersMethodget_config

src/config.py:137

↓ 1 callersFunctionget_optimizer

src/train_food101.py:31

↓ 1 callersMethodget_params

Get parameters for ``crop`` for a random sized crop. Args: img (PIL Image): Image to be cropped. scale (tuple): ran

src/augment_ops.py:602

↓ 1 callersFunctionget_scheduler

(optimizer, args)

src/train_food101.py:24

↓ 1 callersFunctionget_transforms

data/dataloader.py:45

↓ 1 callersFunctionload_checkpoint

(model, checkpoint_path, dp=True)

src/functions.py:11

↓ 1 callersFunctionparse_args

↓ 1 callersFunctionrand_augment_ops

(magnitude=10, hparams=None, transforms=None)

src/augment_ops.py:423

↓ 1 callersFunctionrun

↓ 1 callersFunctionsave_checkpoint

(model, checkpoint_path)

src/functions.py:5

↓ 1 callersFunctionset_log

↓ 1 callersFunctionsetup_seed

↓ 1 callersFunctiontest_epoch

(model, dataloader=None)

src/train_food101.py:156

↓ 1 callersFunctiontrain_food101

src/train_food101.py:174

↓ 1 callersFunctiontrain_valid

(args, model, optimizer, scheduler=None, data=None)

src/train_food101.py:104

↓ 1 callersFunctiontransform

src/augment_ops.py:131

↓ 1 callersFunctionvalid

(args, model, data=None, best_valid=None, nBetter=None, step=None)

src/train_food101.py:73

src/config.py:25

src/augment_ops.py:535

src/augment_ops.py:551

Args: img (PIL Image): Image to be cropped and resized. Returns: PIL Image: Randomly cropped and resized

src/augment_ops.py:642

src/augment_ops.py:715

Args: img (Tensor): Tensor image of size (N, C, H, W). Returns: Tensor: Image with n_holes of dimension l

src/augment_ops.py:738

src/augment_ops.py:786

(self, pil_img)

src/augment_ops.py:795

Method__delattr__

src/functions.py:51

Method__getattr__

src/functions.py:42

Method__getitem__

data/dataloader.py:140

txt_encoder: base / large

model/TextEncoder.py:19

image_encoder: base / large

model/ImageEncoder.py:17

(self, in_dim, out_dim)

model/model.py:19

(self, dropout, in_dim, post_dim, out_dim)

model/model.py:182

(self, args, labels)

data/dataloader.py:117

src/config.py:8

(self, name, prob=0.5, magnitude=10, hparams=None)

src/augment_ops.py:522

(self, ops, num_layers=2, choice_weights=None)

src/augment_ops.py:546

(self, size, scale=(0.08, 1.0), ratio=(3. / 4., 4. / 3.), interpolation='bilinear')

src/augment_ops.py:585

( self, probability=0.5, min_area=0.02, max_area=1/3, min_aspect=0.3, max_aspect=Non

src/augment_ops.py:672

(self, n_holes=1, length=16)

src/augment_ops.py:734

(self, mean, std, inplace=True, use_cuda=True)

src/augment_ops.py:778

data/dataloader.py:137

Method__setattr__

(self, key, value)

src/functions.py:48

src/functions.py:57

Function_blur_level_to_arg

(level, _hparams)

src/augment_ops.py:265

Function_distort_level_to_arg

(level, _hparams)

src/augment_ops.py:257

Function_enhance_level_to_arg

(level, _hparams)

src/augment_ops.py:203

Function_posterize_original_level_to_arg

(level, _hparams)

src/augment_ops.py:229

Function_posterize_research_level_to_arg

(level, _hparams)

src/augment_ops.py:235

Function_posterize_tpu_level_to_arg

(level, _hparams)

src/augment_ops.py:241

Function_rotate_level_to_arg

(level, _hparams)

src/augment_ops.py:196

Function_shear_level_to_arg

(level, _hparams)

src/augment_ops.py:208

Function_skew_level_to_arg

(level, _hparams)

src/augment_ops.py:271

Function_solarize_add_level_to_arg

(level, _hparams)

src/augment_ops.py:252

Function_solarize_level_to_arg

(level, _hparams)

src/augment_ops.py:247

Function_translate_abs_level_to_arg

(level, hparams)

src/augment_ops.py:215

Function_translate_rel_level_to_arg

(level, _hparams)

src/augment_ops.py:222

Function_zoom_level_to_arg

(level, _hparams)

src/augment_ops.py:261

Functionauto_contrast

src/augment_ops.py:144

src/augment_ops.py:418

Functionbrightness

(img, factor, **__)

src/augment_ops.py:188

(img, factor, **__)

src/augment_ops.py:184

Functioncontrast

(img, factor, **__)

src/augment_ops.py:180

Functiondistort

src/augment_ops.py:277

Functionequalize

src/augment_ops.py:152

src/augment_ops.py:373

text: (batch_size, 3, seq_len) 3: input_ids, input_mask, segment_ids input_ids: input_ids, input_mask: attention_

model/TextEncoder.py:56

model/ImageEncoder.py:38

model/model.py:24

model/model.py:189

Methodget_tokenize

model/TextEncoder.py:53

Methodget_tokenizer

model/ImageEncoder.py:35

src/augment_ops.py:148

Functionposterize

(img, bits_to_keep, **__)

src/augment_ops.py:174

Functionpreprocess_text

data/dataloader.py:32

Functionrand_augment_transform

(config_str, hparams)

src/augment_ops.py:438

(img, degrees, **kwargs)

src/augment_ops.py:111

Functionsharpness

(img, factor, **__)

src/augment_ops.py:192

Functionshear_x

(img, factor, **kwargs)

src/augment_ops.py:79

Functionshear_y

(img, factor, **kwargs)

src/augment_ops.py:84

src/augment_ops.py:389

next →1–100 of 109, ranked by callers