Functions — github.com/WisconsinAIVision/MixNMatch

hub / github.com/WisconsinAIVision/MixNMatch / functions

Functions130 in github.com/WisconsinAIVision/MixNMatch

⨍ Functions 130◇ Types & classes 35

↓ 20 callersMethod__init__

(self)

code/model_train.py:13

↓ 10 callersFunctionupBlock

(in_planes, out_planes)

code/model_train.py:38

↓ 10 callersFunctionupBlock

(in_planes, out_planes)

code/model_eval.py:38

↓ 9 callersMethod__init__

(self)

code/model_eval.py:13

↓ 8 callersFunctionBlock3x3_leakRelu

(in_planes, out_planes)

code/model_train.py:372

↓ 8 callersFunctionBlock3x3_leakRelu

(in_planes, out_planes)

code/model_eval.py:306

↓ 7 callersFunctionconv3x3

(in_planes, out_planes)

code/model_train.py:22

↓ 7 callersFunctionconv3x3

(in_planes, out_planes)

code/model_eval.py:22

↓ 7 callersFunctionget_images

(fire, size=[128,128])

code/eval.py:77

↓ 6 callersFunctionDown_unet

(in_c, out_c)

code/model_train.py:596

↓ 6 callersFunctionDown_unet

(in_c, out_c)

code/model_eval.py:371

↓ 6 callersFunctionUp_unet

(in_c, out_c)

code/model_train.py:590

↓ 6 callersFunctionUp_unet

(in_c, out_c)

code/model_eval.py:365

↓ 5 callersFunctionsameBlock

(in_planes, out_planes)

code/model_train.py:46

↓ 5 callersFunctionsameBlock

(in_planes, out_planes)

code/model_eval.py:46

↓ 4 callersFunctiondownBlock

(in_planes, out_planes)

code/model_train.py:364

↓ 4 callersFunctiondownBlock

(in_planes, out_planes)

code/model_eval.py:298

↓ 4 callersFunctionencode_parent_and_child_img

(ndf, in_c=3)

code/model_train.py:348

↓ 4 callersFunctionencode_parent_and_child_img

(ndf, in_c=3)

code/model_eval.py:282

↓ 4 callersFunctionsave_img_results

(imgs_tcpu, fake_imgs, count, image_dir, nrow=8)

code/utils.py:59

↓ 3 callersFunctioncopy_G_params

(model)

code/train_first_stage.py:51

↓ 3 callersFunctionload_params

(model, new_param)

code/train_first_stage.py:46

↓ 2 callersFunctionget_dataloader

(bs=None)

code/datasets.py:196

↓ 2 callersFunctionget_imgs

(img_path, imsize, bbox=None, transform=None, normalize=None)

code/datasets.py:25

↓ 2 callersFunctionload_network

load pretrained generator and encoder

code/eval.py:51

↓ 2 callersFunctionmake_output_dir

()

code/utils.py:26

↓ 2 callersMethodprepare_data

(self, data)

code/train_second_stage.py:90

↓ 2 callersMethodprepare_data

(self, data)

code/train_first_stage.py:166

↓ 2 callersFunctionsave_img

(img, file )

code/eval.py:88

↓ 2 callersMethodtrain_Dnet

(self, idx)

code/train_first_stage.py:182

↓ 1 callersMethod_make_layer

(self,num_residual,ngf)

code/model_train.py:144

↓ 1 callersMethod_make_layer

(self)

code/model_train.py:186

↓ 1 callersMethod_make_layer

(self,num_residual,ngf)

code/model_eval.py:149

↓ 1 callersMethod_make_layer

(self)

code/model_eval.py:197

↓ 1 callersFunction_merge_a_into_b

Merge config dictionary a into config dictionary b, clobbering the options in b whenever they are also specified in a.

code/config.py:66

↓ 1 callersFunctioncal_gradient_penalty

Calculate the gradient penalty loss, used in WGAN-GP paper https://arxiv.org/abs/1704.00028 Arguments: netD (network) -- dis

code/utils.py:95

↓ 1 callersFunctionchild_to_parent

(child_c_code, classes_child, classes_parent)

code/utils.py:48

↓ 1 callersFunctiondefine_optimizers

( extractor, dis_dis )

code/train_second_stage.py:25

↓ 1 callersFunctiondefine_optimizers

( netG, netsD, BD, encoder )

code/train_first_stage.py:26

↓ 1 callersFunctioneval_code

()

code/eval.py:98

↓ 1 callersFunctioneval_feature

()

code/eval.py:125

↓ 1 callersMethodload_bbox

(self)

code/datasets.py:122

↓ 1 callersMethodload_filenames

(self, data_dir)

code/datasets.py:143

↓ 1 callersFunctionload_network

()

code/train_second_stage.py:32

↓ 1 callersFunctionload_network

()

code/train_first_stage.py:69

↓ 1 callersMethodprepare_code

(self)

code/train_first_stage.py:145

↓ 1 callersFunctionsave_model

( dis_dis, extractor, epoch, model_dir)

code/train_second_stage.py:62

↓ 1 callersFunctionsave_model

( encoder, myG, D0, D1, D2, BD, epoch, model_dir)

code/train_first_stage.py:99

↓ 1 callersFunctionsave_opt

( optimizerGE, optimizerD0, optimizerD2, optimizerBD, epoch, opt_dir)

code/train_first_stage.py:109

↓ 1 callersMethodtrain

(self)

code/train_second_stage.py:116

↓ 1 callersMethodtrain

(self)

code/train_first_stage.py:327

↓ 1 callersMethodtrain_BD

(self)

code/train_first_stage.py:266

↓ 1 callersMethodtrain_EG

(self)

code/train_first_stage.py:284

FunctionBottleNeck

(in_c, out_c)

code/model_train.py:593

FunctionBottleNeck

(in_c, out_c)

code/model_eval.py:368

Method__call__

(self, prediction, label)

code/train_first_stage.py:59

code/model_train.py:55

Method__init__

(self, ngf)

code/model_train.py:72

Method__init__

(self, ngf)

code/model_train.py:80

Method__init__

(self, ngf)

code/model_train.py:89

Method__init__

(self, ngf)

code/model_train.py:120

Method__init__

(self, ngf, num_residual=2)

code/model_train.py:175

Method__init__

(self)

code/model_train.py:205

Method__init__

(self, ndf=64)

code/model_train.py:269

Method__init__

(self, ndf=32)

code/model_train.py:285

Method__init__

(self, ndf=64)

code/model_train.py:316

Method__init__

(self, )

code/model_train.py:381

Method__init__

(self, std)

code/model_train.py:431

Method__init__

(self, in_c, out_c, k, s, p, bn=True, activation=None, noise=False, std=None )

code/model_train.py:439

Method__init__

(self, in_c, out_c, k, s, p=0, bn=True, activation=None, noise=False, std=None )

code/model_train.py:467

Method__init__

(self, in_c, out_c, bn=True, activation=None, noise=False, std=None )

code/model_train.py:495

Method__init__

(self, shape)

code/model_train.py:523

Method__init__

(self, code_len, ngf=16)

code/model_train.py:534

Method__init__

(self)

code/model_train.py:563

Method__init__

(self,in_c, out_c)

code/model_train.py:599

Method__init__

(self, in_c )

code/model_train.py:660

Method__init__

(self, channel_num)

code/model_eval.py:55

Method__init__

(self, ngf)

code/model_eval.py:72

Method__init__

(self, ngf)

code/model_eval.py:80

Method__init__

(self, ngf)

code/model_eval.py:89

Method__init__

(self, ngf)

code/model_eval.py:125

Method__init__

(self, ngf, num_residual=2)

code/model_eval.py:186

Method__init__

(self)

code/model_eval.py:216

Method__init__

(self, )

code/model_eval.py:315

Method__init__

(self,in_c, out_c)

code/model_eval.py:374

Method__init__

(self, data_dir, base_size=64, transform=None)

code/datasets.py:99

Method__init__

(self, output_dir)

code/train_second_stage.py:68

Method__init__

(self)

code/train_first_stage.py:57

Method__init__

(self, output_dir)

code/train_first_stage.py:121

Method__len__

(self)

code/datasets.py:190

Functioncfg_from_file

Load a config file and merge it into the default options.

code/config.py:99

Functionchild_to_parent

(child_c_code, classes_child, classes_parent)

code/model_train.py:27

Functionchild_to_parent

(child_c_code, classes_child, classes_parent)

code/model_eval.py:27

Methodforward

(self, x)

code/model_train.py:15

Methodforward

(self, x)

code/model_train.py:63

Methodforward

(self, h_code)

code/model_train.py:75

Methodforward

(self, h_code)

code/model_train.py:83

Methodforward

(self, z_input, input)

code/model_train.py:108

Methodforward

(self, z_input, input, which)

code/model_train.py:150

next →1–100 of 130, ranked by callers