MCPcopy Create free account

hub / github.com/NVIDIA/pix2pixHD / functions

Functions139 in github.com/NVIDIA/pix2pixHD

↓ 10 callersMethodsave
(self)
util/html.py:44
↓ 8 callersMethod__init__
(self, requires_grad=False)
models/networks.py:387
↓ 6 callersMethodforward
(self, label, inst, image, feat, infer=False)
models/pix2pixHD_model.py:152
↓ 6 callersMethodset_features
(self, idx, feat, cluster_idx)
models/ui_model.py:326
↓ 6 callersMethodsingle_forward
(self, net_input, feat_map)
models/ui_model.py:215
↓ 5 callersMethodadd_images
(self, ims, txts, links, width=512)
util/html.py:32
↓ 5 callersFunctionmake_dataset
(dir)
data/image_folder.py:21
↓ 5 callersMethodparse
(self, save=True)
options/base_options.py:65
↓ 4 callersFunctionCreateDataLoader
(opt)
data/data_loader.py:2
↓ 4 callersFunctioncreate_model
(opt)
models/models.py:3
↓ 4 callersMethodget_edges
(self, t)
models/ui_model.py:108
↓ 4 callersMethodload_data
()
data/base_data_loader.py:10
↓ 4 callersMethodload_network
(self, network, network_label, epoch_label, save_dir='')
models/base_model.py:50
↓ 3 callersMethodadd_header
(self, str)
util/html.py:24
↓ 3 callersMethodbackup_current_state
(self)
models/ui_model.py:286
↓ 3 callersFunctionget_transform
(opt, params, method=Image.BICUBIC, normalize=True)
data/base_dataset.py:33
↓ 3 callersMethodsave_network
(self, network, network_label, epoch_label, gpu_ids)
models/base_model.py:42
↓ 2 callersMethoddiscriminate
(self, input_label, test_image, use_pool=False)
models/pix2pixHD_model.py:144
↓ 2 callersMethodencode_input
(self, label_map, inst_map=None, real_image=None, feat_map=None, infer=False)
models/pix2pixHD_model.py:111
↓ 2 callersMethodforward
(self, X)
models/networks.py:409
↓ 2 callersFunctionget_norm_layer
(norm_type='instance')
models/networks.py:18
↓ 2 callersMethodget_target_tensor
(self, input, target_is_real)
models/networks.py:82
↓ 2 callersMethodinference
(self, label, inst, image=None)
models/pix2pixHD_model.py:195
↓ 2 callersMethodinitialize
(self, opt)
models/ui_model.py:15
↓ 2 callersMethodloss_filter
(g_gan, g_gan_feat, g_vgg, d_real, d_fake)
models/pix2pixHD_model.py:15
↓ 2 callersFunctionmkdir
(path)
util/util.py:48
↓ 2 callersFunctiontime_inference
(engine, batch_size, inp)
run_engine.py:104
↓ 2 callersMethodtoTensor
(self, img, normalize=False)
models/ui_model.py:33
↓ 1 callersFunctionCreateDataset
(opt)
data/custom_dataset_data_loader.py:5
↓ 1 callersFunction__crop
(img, pos, size)
data/base_dataset.py:79
↓ 1 callersFunction__flip
(img, flip)
data/base_dataset.py:87
↓ 1 callersFunction__make_power_2
(img, base, method=Image.BICUBIC)
data/base_dataset.py:63
↓ 1 callersFunction__scale_width
(img, target_width, method=Image.BICUBIC)
data/base_dataset.py:71
↓ 1 callersMethodadd_table
(self, border=1)
util/html.py:28
↓ 1 callersMethodbuild_conv_block
(self, dim, padding_type, norm_layer, activation, use_dropout)
models/networks.py:219
↓ 1 callersFunctionconvert_to_datatype
(v)
run_engine.py:123
↓ 1 callersMethodcopy_features
(self, idx_src, idx_tgt)
models/ui_model.py:331
↓ 1 callersFunctioncreate_memory
(engine, name, buf, mem, batchsize, inp, inp_idx)
run_engine.py:82
↓ 1 callersMethoddisplay_current_results
(self, visuals, epoch, step)
util/visualizer.py:37
↓ 1 callersMethodencode_features
(self, image, inst)
models/pix2pixHD_model.py:240
↓ 1 callersMethodget_crop_region
(self, idx)
models/ui_model.py:293
↓ 1 callersMethodget_edges
(self, t)
models/pix2pixHD_model.py:262
↓ 1 callersMethodget_image_dir
(self)
util/html.py:21
↓ 1 callersFunctionget_input_output_names
(trt_engine)
run_engine.py:62
↓ 1 callersFunctionget_params
(opt, size)
data/base_dataset.py:17
↓ 1 callersMethodinit_loss_filter
(self, use_gan_feat_loss, use_vgg_loss)
models/pix2pixHD_model.py:13
↓ 1 callersMethodinitialize
(self, opt)
data/custom_dataset_data_loader.py:18
↓ 1 callersMethodinitialize
(self, opt)
data/aligned_dataset.py:7
↓ 1 callersMethodinitialize
(self)
options/base_options.py:11
↓ 1 callersFunctionis_image_file
(filename)
data/image_folder.py:17
↓ 1 callersFunctionlabelcolormap
(N)
util/util.py:61
↓ 1 callersFunctionlcm
(a,b)
train.py:9
↓ 1 callersMethodname
(self)
data/custom_dataset_data_loader.py:15
↓ 1 callersMethodname
(self)
data/aligned_dataset.py:75
↓ 1 callersMethodname
(self)
models/ui_model.py:12
↓ 1 callersFunctionnormalize
()
data/base_dataset.py:60
↓ 1 callersMethodplot_current_errors
(self, errors, step)
util/visualizer.py:95
↓ 1 callersMethodprint_current_errors
(self, epoch, i, errors, t)
util/visualizer.py:102
↓ 1 callersMethodprint_layer_times
(self)
run_engine.py:54
↓ 1 callersMethodquery
(self, images)
util/image_pool.py:11
↓ 1 callersFunctionrun_onnx
(onnx_file, data_type, bs, inp)
run_engine.py:138
↓ 1 callersFunctionrun_trt_engine
(engine_file, bs, it)
run_engine.py:134
↓ 1 callersMethodsample_features
(self, inst)
models/pix2pixHD_model.py:219
↓ 1 callersMethodsave
(self, label)
models/base_model.py:38
↓ 1 callersMethodsave_images
(self, webpage, visuals, image_path)
util/visualizer.py:113
↓ 1 callersMethodsingleD_forward
(self, model, input)
models/networks.py:310
↓ 1 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, normalize=True)
util/util.py:10
↓ 1 callersFunctionuint82bin
returns the binary of integer n, count refers to amount of bits
util/util.py:57
↓ 1 callersMethodupdate_fixed_params
(self)
models/pix2pixHD_model.py:279
↓ 1 callersMethodupdate_learning_rate
()
models/base_model.py:90
Method__call__
(self, input, target_is_real)
models/networks.py:100
Method__call__
(self, gray_image)
util/util.py:90
Method__getitem__
(self, index)
data/aligned_dataset.py:35
Method__getitem__
(self, index)
data/image_folder.py:54
Method__init__
(self, timing_iter)
run_engine.py:42
Method__init__
(self)
data/base_data_loader.py:3
Method__init__
(self)
data/base_dataset.py:8
Method__init__
(self, root, transform=None, return_paths=False, loader=default_loader)
data/image_folder.py:40
Method__init__
(self)
options/base_options.py:7
Method__init__
(self, use_lsgan=True, target_real_label=1.0, target_fake_label=0.0, tensor=torch.FloatTensor
models/networks.py:69
Method__init__
(self, gpu_ids)
models/networks.py:113
Method__init__
(self, input_nc, output_nc, ngf=32, n_downsample_global=3, n_blocks_global=9, n_local_enhanc
models/networks.py:130
Method__init__
(self, input_nc, output_nc, ngf=64, n_downsampling=3, n_blocks=9, norm_layer=nn.BatchNorm2d,
models/networks.py:184
Method__init__
(self, dim, padding_type, norm_layer, activation=nn.ReLU(True), use_dropout=False)
models/networks.py:215
Method__init__
(self, input_nc, output_nc, ngf=32, n_downsampling=4, norm_layer=nn.BatchNorm2d)
models/networks.py:256
Method__init__
(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNorm2d, use_sigmoid=False, num_D=3,
models/networks.py:293
Method__init__
(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNorm2d, use_sigmoid=False, getIntermFeat=False)
models/networks.py:335
Method__init__
(self, web_dir, title, refresh=0)
util/html.py:7
Method__init__
(self, opt)
util/visualizer.py:14
Method__init__
(self, n=35)
util/util.py:86
Method__init__
(self, pool_size)
util/image_pool.py:5
Method__len__
(self)
data/custom_dataset_data_loader.py:30
Method__len__
(self)
data/aligned_dataset.py:72
Method__len__
(self)
data/image_folder.py:64
Methodadd_objects
(self, click_src, label_tgt, mask, style_id=0)
models/ui_model.py:190
Methodadd_strokes
(self, click_src, label_tgt, bw, save)
models/ui_model.py:152
Methodchange_labels
(self, click_src, click_tgt)
models/ui_model.py:117
Functiondefault_loader
(path)
data/image_folder.py:34
Functiondefine_D
(input_nc, ndf, n_layers_D, norm='instance', use_sigmoid=False, num_D=1, getIntermFeat=False, gpu_ids=[])
models/networks.py:46
Functiondefine_G
(input_nc, output_nc, ngf, netG, n_downsample_global=3, n_blocks_global=9, n_local_enhancers=1,
models/networks.py:27
next →1–100 of 139, ranked by callers