MCPcopy Create free account

hub / github.com/WeilunWang/NEGCUT / functions

Functions264 in github.com/WeilunWang/NEGCUT

↓ 1 callersMethodset_input
Unpack input data from the dataloader and perform necessary pre-processing steps. Parameters: input (dict): include the data itsel
models/negcut_model.py:157
↓ 1 callersFunctiontensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
util/util.py:44
↓ 1 callersMethodtest
Forward function used in test time. This function wraps <forward> function in no_grad() so we don't save intermediate steps for backprop
models/base_model.py:119
↓ 1 callersMethodupdate_learning_rate
Update learning rates for all the networks; called at the end of every epoch
models/base_model.py:137
↓ 1 callersMethodupdate_moving_average
(self, key, x)
models/feature_extractor.py:86
↓ 1 callersFunctionupfirdn2d_native
( input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1 )
models/stylegan_networks.py:37
Method__call__
Calculate loss given Discriminator's output and grount truth labels. Parameters: prediction (tensor) - - tpyically the prediction
models/losses.py:162
Method__getitem__
Return a data point and its metadata information. Parameters: index - - a random integer for data indexing Returns:
data/base_dataset.py:52
Method__getitem__
Return a data point and its metadata information. Parameters: index (int) -- a random integer for data indexing Ret
data/singleimage_dataset.py:67
Method__getitem__
Return a data point and its metadata information. Parameters: index (int) -- a random integer for data indexing Ret
data/unaligned_dataset.py:40
Method__getitem__
(self, index)
data/image_folder.py:55
Method__getitem__
Return a data point and its metadata information. Parameters: index -- a random integer for data indexing Returns:
data/template_dataset.py:54
Method__getitem__
Return a data point and its metadata information. Parameters: index - - a random integer for data indexing Returns a dic
data/single_dataset.py:23
Method__init__
Initialize the class; save the options in the class Parameters: opt (Option class)-- stores all the experiment flags; needs to be
data/base_dataset.py:23
Method__init__
Initialize this class Step 1: create a dataset instance given the name [dataset_mode] Step 2: create a multi-threaded data loader.
data/__init__.py:65
Method__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/singleimage_dataset.py:21
Method__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/unaligned_dataset.py:20
Method__init__
(self, root, transform=None, return_paths=False, loader=default_loader)
data/image_folder.py:42
Method__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/template_dataset.py:36
Method__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/single_dataset.py:12
Method__init__
Reset the class; indicates the class hasn't been initailized
options/base_options.py:16
Method__init__
(self, nce_layers, num_patches=256, nc=256, init_type='normal', init_gain=0.02, gpu_ids=[])
models/negative_generator.py:30
Method__init__
(self, use_conv=False, num_patches=256, nc=256, z_dim=64, init_type='normal', init_gain=0.02, gpu_ids=[])
models/negative_generator.py:54
Method__init__
(self, opt)
models/sinnegcut_model.py:50
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int
models/discriminator.py:59
Method__init__
(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNorm2d, no_antialias=False)
models/discriminator.py:158
Method__init__
(self, channels, pad_type='reflect', filt_size=3, stride=2, pad_off=0)
models/utils.py:36
Method__init__
Initialize the Resnet block A resnet block is a conv block with skip connections We construct a conv block with build_conv_block func
models/utils.py:216
Method__init__
Initialize the BaseModel class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseO
models/base_model.py:18
Method__init__
(self, opt)
models/cut_model.py:58
Method__init__
(self, opt)
models/negcut_model.py:61
Method__init__
(self, power=2)
models/feature_extractor.py:27
Method__init__
(self)
models/feature_extractor.py:49
Method__init__
(self, init_type='normal', init_gain=0.02, gpu_ids=[])
models/feature_extractor.py:62
Method__init__
(self, use_mlp=False, init_type='normal', init_gain=0.02, nc=256, gpu_ids=[])
models/feature_extractor.py:108
Method__init__
(self, opt)
models/losses.py:8
Method__init__
(self, opt)
models/losses.py:59
Method__init__
(self, opt)
models/sincut_model.py:56
Method__init__
Initialize this model class. Parameters: opt -- training/test options A few things can be done here. - (required
models/template_model.py:41
Method__init__
(self, channel, negative_slope=0.2, scale=2 ** 0.5)
models/stylegan_networks.py:22
Method__init__
(self)
models/stylegan_networks.py:79
Method__init__
(self, kernel, factor=2)
models/stylegan_networks.py:98
Method__init__
(self, kernel, factor=2)
models/stylegan_networks.py:119
Method__init__
( self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True )
models/stylegan_networks.py:159
Method__init__
( self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None )
models/stylegan_networks.py:199
Method__init__
(self, negative_slope=0.2)
models/stylegan_networks.py:236
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim, demod
models/stylegan_networks.py:248
Method__init__
(self)
models/stylegan_networks.py:351
Method__init__
(self, channel, size=4)
models/stylegan_networks.py:365
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim=None,
models/stylegan_networks.py:378
Method__init__
(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
models/stylegan_networks.py:418
Method__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
models/stylegan_networks.py:440
Method__init__
( self, in_channel, out_channel, kernel_size, downsample=False,
models/stylegan_networks.py:622
Method__init__
(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1], downsample=True, skip_gain=1.0)
models/stylegan_networks.py:671
Method__init__
(self, input_nc, ndf=64, n_layers=3, no_antialias=False, size=None, opt=None)
models/stylegan_networks.py:696
Method__init__
(self, input_nc, output_nc, ngf=64, use_dropout=False, n_blocks=6, padding_type='reflect', no_antialias=False,
models/stylegan_networks.py:798
Method__init__
(self, input_nc, output_nc, ngf=64, use_dropout=False, n_blocks=6, padding_type='reflect', no_antialias=False,
models/stylegan_networks.py:852
Method__init__
(self, input_nc, output_nc, ngf=64, use_dropout=False, n_blocks=6, padding_type='reflect', no_antialias=False,
models/stylegan_networks.py:898
Method__init__
Construct a Resnet-based generator Parameters: input_nc (int) -- the number of channels in input images outp
models/generator.py:68
Method__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/generator.py:201
Method__init__
(self, technique='cyclegan', verbose=True)
util/get_data.py:27
Method__init__
Initialize the HTML classes Parameters: web_dir (str) -- a directory that stores the webpage. HTML file will be created at <web_d
util/html.py:14
Method__init__
Initialize the Visualizer class Parameters: opt -- stores all the experiment flags; needs to be a subclass of BaseOptions
util/visualizer.py:52
Method__init__
Initialize the ImagePool class Parameters: pool_size (int) -- the size of image buffer, if pool_size=0, no buffer will be created
util/image_pool.py:12
Method__iter__
Return a batch of data
data/__init__.py:93
Method__len__
Return the total number of images in the dataset.
data/base_dataset.py:47
Method__len__
Return the number of data in the dataset
data/__init__.py:89
Method__len__
Let's pretend the single image contains 100,000 crops for convenience.
data/singleimage_dataset.py:105
Method__len__
Return the total number of images in the dataset. As we have two datasets with potentially different number of images, we take a maxi
data/unaligned_dataset.py:72
Method__len__
(self)
data/image_folder.py:65
Method__len__
Return the total number of images.
data/template_dataset.py:73
Method__len__
Return the total number of images in the dataset.
data/single_dataset.py:38
Method__patch_instance_norm_state_dict
Fix InstanceNorm checkpoints incompatibility (prior to 0.4)
models/base_model.py:182
Function__print_size_warning
Print warning information about image size(only print once)
data/base_dataset.py:223
Method__repr__
(self)
models/stylegan_networks.py:191
Method__repr__
(self)
models/stylegan_networks.py:229
Method__repr__
(self)
models/stylegan_networks.py:297
Methodcompute_D_loss
(self)
models/sinnegcut_model.py:58
Methodcompute_D_loss
(self)
models/sincut_model.py:64
Methodcompute_G_loss
(self)
models/sinnegcut_model.py:65
Methodcompute_G_loss
(self)
models/sincut_model.py:71
Functioncopyconf
(default_opt, **kwargs)
util/util.py:24
Functioncorrect_resize
(t, size, mode=Image.BICUBIC)
util/util.py:157
Functioncorrect_resize_label
(t, size)
util/util.py:143
Methoddata_dependent_initialize
(self, data)
models/base_model.py:109
Methoddata_dependent_initialize
The feature network netF is defined in terms of the shape of the intermediate, extracted features of the encoder portion of netG. Bec
models/negcut_model.py:104
Functiondefault_loader
(path)
data/image_folder.py:36
Functiondiagnose_network
Calculate and print the mean of average absolute(gradients) Parameters: net (torch network) -- Torch network name (str) -- the na
util/util.py:65
Methoddict_grad_hook_factory
(add_func=lambda x: x)
models/base_model.py:47
Functionfind_class_in_module
(target_cls_name, module)
util/util.py:31
Methodforward
(self, x)
models/negative_generator.py:23
Methodforward
(self, nce_layers, num_images)
models/negative_generator.py:43
Methodforward
(self, feats, num_patches)
models/negative_generator.py:82
Methodforward
Standard forward.
models/discriminator.py:118
Methodforward
(self, input)
models/discriminator.py:161
Methodforward
(self, inp)
models/utils.py:51
Methodforward
(self, inp)
models/utils.py:76
Methodforward
(self, x)
models/utils.py:206
Methodforward
Forward function (with skip connections)
models/utils.py:267
Methodforward
(self, x)
models/feature_extractor.py:31
← previousnext →101–200 of 264, ranked by callers