MCPcopy Create free account

hub / github.com/NVlabs/few-shot-vid2vid / functions

Functions455 in github.com/NVlabs/few-shot-vid2vid

Method__init__
(self, args)
models/networks/flownet2_pytorch/losses.py:41
Method__init__
(self, args, startScale = 4, numScales = 5, l_weight= 0.32, norm= 'L1')
models/networks/flownet2_pytorch/losses.py:53
Method__init__
(self, image_size, crop_size)
models/networks/flownet2_pytorch/datasets.py:14
Method__init__
(self, image_size, crop_size)
models/networks/flownet2_pytorch/datasets.py:24
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:107
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:111
Method__init__
(self, args, is_cropped, root = '/path/to/FlyingChairs_release/data', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:115
Method__init__
(self, args, is_cropped, root = '/path/to/flyingthings3d', dstype = 'frames_cleanpass', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:176
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:243
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:247
Method__init__
(self, args, is_cropped, root = '/path/to/chairssdhom/data', dstype = 'train', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:251
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:313
Method__init__
(self, args, is_cropped = False, root = '', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:317
Method__init__
(self, args, is_cropped, root = '/path/to/frames/only/folder', iext = 'png', replicates = 1)
models/networks/flownet2_pytorch/datasets.py:321
Method__init__
(self, args)
models/networks/flownet2_pytorch/main.py:162
Method__init__
(self, title)
models/networks/flownet2_pytorch/utils/tools.py:25
Method__init__
(self, iterable)
models/networks/flownet2_pytorch/utils/tools.py:99
Method__init__
(self, args, input_channels = 12, batchNorm=True)
models/networks/flownet2_pytorch/networks/FlowNetS.py:16
Method__init__
(self,args, batchNorm=True, div_flow = 20)
models/networks/flownet2_pytorch/networks/FlowNetC.py:14
Method__init__
(self,args, batchNorm=True)
models/networks/flownet2_pytorch/networks/FlowNetFusion.py:12
Method__init__
(self, args, batchNorm=True)
models/networks/flownet2_pytorch/networks/FlowNetSD.py:12
Method__init__
(self)
models/networks/flownet2_pytorch/networks/submodules.py:49
Method__init__
(self, norm_deg=2)
models/networks/flownet2_pytorch/networks/channelnorm_package/channelnorm.py:33
Method__init__
(self, pad_size=3, kernel_size=3, max_displacement=20, stride1=1, stride2=2, corr_multiply=1)
models/networks/flownet2_pytorch/networks/correlation_package/correlation.py:8
Method__init__
(self, kernel_size=1)
models/networks/flownet2_pytorch/networks/resample2d_package/resample2d.py:40
Method__init__
(self)
models/networks/sync_batchnorm/comm.py:21
Method__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
models/networks/sync_batchnorm/comm.py:67
Method__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True)
models/networks/sync_batchnorm/batchnorm.py:39
Method__init__
(self, web_dir, title, refresh=0, infer=False)
util/html.py:14
Method__init__
(self, opt)
util/visualizer.py:20
Method__init__
(self, n=35)
util/util.py:209
Method__init__
(self, pool_size)
util/image_pool.py:11
Method__len__
(self)
data/lmdb_dataset.py:50
Method__len__
(self)
data/custom_dataset_data_loader.py:37
Method__len__
(self)
data/fewshot_face_dataset.py:266
Method__len__
(self)
data/fewshot_street_dataset.py:126
Method__len__
(self)
data/image_folder.py:118
Method__len__
(self)
data/fewshot_pose_dataset.py:265
Method__len__
(self)
models/networks/flownet2_pytorch/datasets.py:103
Method__len__
(self)
models/networks/flownet2_pytorch/datasets.py:172
Method__len__
(self)
models/networks/flownet2_pytorch/datasets.py:239
Method__len__
(self)
models/networks/flownet2_pytorch/datasets.py:309
Method__len__
(self)
models/networks/flownet2_pytorch/datasets.py:364
Method__len__
(self)
models/networks/flownet2_pytorch/utils/tools.py:106
Method__next__
(self)
models/networks/flownet2_pytorch/utils/tools.py:109
Method_check_input_dim
(self, input)
models/networks/sync_batchnorm/batchnorm.py:247
Method_check_input_dim
(self, input)
models/networks/sync_batchnorm/batchnorm.py:311
Method_data_parallel_master
Reduce the sum and square-sum, compute the statistics, and broadcast it.
models/networks/sync_batchnorm/batchnorm.py:90
Functionadd_arguments_for_module
(parser, module, argument_for_class, default, skip_params=[], parameter_defaults={})
models/networks/flownet2_pytorch/utils/tools.py:55
Functionadd_norm_layer
(layer)
models/networks/normalization.py:62
MethodassertTensorClose
(self, a, b, atol=1e-3, rtol=1e-3)
models/networks/sync_batchnorm/unittest.py:24
Methodbackward
(ctx, grad_output)
models/networks/flownet2_pytorch/networks/channelnorm_package/channelnorm.py:20
Methodbackward
(ctx, grad_output)
models/networks/flownet2_pytorch/networks/correlation_package/correlation.py:39
Methodconcat
(self, tensors, dim=0)
data/base_dataset.py:58
Functioncreate_dataset
(opt)
data/__init__.py:40
Functiondatestr
()
models/networks/flownet2_pytorch/utils/tools.py:13
Functiondefault_loader
(path)
data/image_folder.py:88
Functiondefine_D
(opt, input_nc, ndf, n_layers_D, norm='spectralinstance', subarch='n_layers', num_D=1, getIntermFeat=False, st
models/networks/__init__.py:41
Functiondefine_G
(opt)
models/networks/__init__.py:29
Functiondist_all_gather_tensor
gather to all ranks
util/distributed.py:89
Functiondist_all_reduce_tensor
Reduce to all ranks
util/distributed.py:78
Functiondist_reduce_tensor
Reduce to rank 0
util/distributed.py:66
Functionformat_dictionary_of_losses
(labels, values)
models/networks/flownet2_pytorch/utils/tools.py:88
Methodforward
(self)
models/base_model.py:36
Methodforward
(self, *inputs, **kwargs)
models/models.py:93
Methodforward
(self, data_list, epoch=0, dummy_bs=0)
models/flownet.py:34
Methodforward
(self, data_list, save_images=False, mode='inference', dummy_bs=0)
models/vid2vid_model.py:47
Methodforward
(self, x, y)
models/networks/loss.py:120
Methodforward
(self, input, target, mask)
models/networks/loss.py:135
Methodforward
(self, mu, logvar)
models/networks/loss.py:141
Methodforward
(self, input, ref=None)
models/networks/discriminator.py:49
Methodforward
(self, input)
models/networks/discriminator.py:93
Methodforward
(self, x)
models/networks/discriminator.py:178
Methodforward
(self, input, ref)
models/networks/discriminator.py:197
Methodforward
(self, input, weight=None, bias=None, stride=1)
models/networks/architecture.py:23
Methodforward
(self, input, weight=None, bias=None, stride=1)
models/networks/architecture.py:29
Methodforward
(self, input, weight=None, bias=None, stride=1)
models/networks/architecture.py:34
Methodforward
(self, x, label=None)
models/networks/architecture.py:65
Methodforward
(self, x, label=None, conv_weights=[], norm_weights=[])
models/networks/architecture.py:92
Methodforward
(self, X)
models/networks/vgg.py:36
Methodforward
(self, x)
models/networks/vgg.py:53
Methodforward
(self, x, maps, weights=None)
models/networks/normalization.py:37
Methodforward
(self, label, label_refs, img_refs, prev=[None, None], t=0, img_coarse=None)
models/networks/generator.py:181
Methodforward
(self, label, label_prev, img_prev, for_ref=False)
models/networks/generator.py:497
Methodforward
(self, input, weights=None)
models/networks/generator.py:541
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:117
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:189
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:258
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:304
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:389
Methodforward
(self, inputs)
models/networks/flownet2_pytorch/models.py:466
Methodforward
(self, output, target)
models/networks/flownet2_pytorch/losses.py:17
Methodforward
(self, output, target)
models/networks/flownet2_pytorch/losses.py:24
Methodforward
(self, output, target)
models/networks/flownet2_pytorch/losses.py:35
Methodforward
(self, output, target)
models/networks/flownet2_pytorch/losses.py:47
Methodforward
(self, output, target)
models/networks/flownet2_pytorch/losses.py:72
Methodforward
(self, data, target, inference=False )
models/networks/flownet2_pytorch/main.py:169
Methodforward
(self, x)
models/networks/flownet2_pytorch/networks/FlowNetS.py:60
Methodforward
(self, x)
models/networks/flownet2_pytorch/networks/FlowNetC.py:71
Methodforward
(self, x)
models/networks/flownet2_pytorch/networks/FlowNetFusion.py:47
← previousnext →301–400 of 455, ranked by callers