MCPcopy Create free account

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

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

⨍ Functions 455◇ Types & classes 95
Methodforward
(self, x)
models/networks/flownet2_pytorch/networks/FlowNetSD.py:66
Methodforward
(self, input)
models/networks/flownet2_pytorch/networks/submodules.py:44
Methodforward
(self, input)
models/networks/flownet2_pytorch/networks/submodules.py:52
Methodforward
(ctx, input1, norm_deg=2)
models/networks/flownet2_pytorch/networks/channelnorm_package/channelnorm.py:8
Methodforward
(ctx, input1, input2, pad_size, kernel_size, max_displacement, stride1, stride2, corr_multiply)
models/networks/flownet2_pytorch/networks/correlation_package/correlation.py:19
Methodforward
(ctx, input1, input2, kernel_size=1)
models/networks/flownet2_pytorch/networks/resample2d_package/resample2d.py:8
Methodforward
(self, input)
models/networks/sync_batchnorm/batchnorm.py:48
Functionget_option_setter
(dataset_name)
data/__init__.py:35
Functiongpumemusage
()
models/networks/flownet2_pytorch/utils/tools.py:117
Functionhook
(grad)
models/networks/flownet2_pytorch/networks/submodules.py:73
Functioninit_deconv_bilinear
(weight)
models/networks/flownet2_pytorch/networks/submodules.py:56
Methodinit_deconv_bilinear
(self, weight)
models/networks/flownet2_pytorch/models.py:101
Methodinit_func
(m)
models/networks/base_network.py:87
Methodinitialize
(self, opt)
data/fewshot_face_dataset.py:32
Methodinitialize
(self, opt)
data/fewshot_street_dataset.py:35
Methodinitialize
(self, opt)
data/fewshot_pose_dataset.py:37
Methodinitialize
(self, parser)
options/train_options.py:10
Methodinitialize
(self, parser)
options/test_options.py:10
Methodinitialize
(self, opt, add_face_D, refine_face)
models/face_refiner.py:16
Methodinitialize
(self, opt)
models/base_model.py:19
Methodinitialize
(self, opt)
models/loss_collector.py:19
Methodinitialize
(self, opt, epoch=0)
models/vid2vid_model.py:24
Functionkwargs_from_args
(args, argument_for_class)
models/networks/flownet2_pytorch/utils/tools.py:84
Methodload_data
(self)
data/custom_dataset_data_loader.py:34
Methodload_options
(self, opt)
options/base_options.py:190
Functionmaster_only
(func)
util/distributed.py:45
Functionmaster_only_print
master-only print
util/distributed.py:61
Functionmkdirs
(paths)
util/util.py:146
Functionmodify_commandline_options
(parser, is_train)
models/networks/__init__.py:17
Methodmodify_commandline_options
(parser, is_train)
data/fewshot_street_dataset.py:18
Methodmodify_commandline_options
(parser, is_train)
data/fewshot_pose_dataset.py:21
Methodname
(self)
data/fewshot_face_dataset.py:270
Methodname
(self)
data/fewshot_street_dataset.py:130
Methodname
(self)
data/fewshot_pose_dataset.py:269
Methodname
(self)
models/face_refiner.py:13
Methodname
(self)
models/base_model.py:16
Methodname
(self)
models/loss_collector.py:16
Methodname
(self)
models/vid2vid_model.py:21
Functionnew_replicate
(module, device_ids)
models/networks/sync_batchnorm/replicate.py:96
Methodnr_slaves
(self)
models/networks/sync_batchnorm/comm.py:130
Functionpatch_replication_callback
Monkey-patch an existing `DataParallel` object. Add the replication callback. Useful when you have customized `DataParallel` implementation.
models/networks/sync_batchnorm/replicate.py:77
Methodprint_options
(self, opt)
options/base_options.py:157
Methodquery
(self, images)
util/image_pool.py:17
FunctionreadFlow
Read .flo file in Middlebury format
models/networks/flownet2_pytorch/utils/flow_utils.py:5
Functionread_gen
(file_name)
models/networks/flownet2_pytorch/utils/frame_utils.py:6
Methodremove_dummy_from_tensor
(self, tensors, remove_size=0)
models/base_model.py:96
Methodreplicate
(self, module, device_ids)
models/networks/sync_batchnorm/replicate.py:71
Functionsave_checkpoint
(state, is_best, path, prefix, filename='checkpoint.pth.tar')
models/networks/flownet2_pytorch/utils/tools.py:138
Functionsave_grad
(grads, name)
models/networks/flownet2_pytorch/networks/submodules.py:72
Functionsave_image
(image_numpy, image_path, create_dir=False)
util/util.py:136
Functiontensor2flow
(tensor, imtype=np.uint8, tile=False)
util/util.py:82
Functionupdate_hyperparameter_schedule
(args, epoch, global_iteration, optimizer)
models/networks/flownet2_pytorch/utils/tools.py:131
Functionvisualize_label
(opt, label_tensor, model=None)
util/util.py:17
Functionwrapper
(*args, **kwargs)
util/distributed.py:47
FunctionwriteFlow
Write optical flow to file. If v is None, uv is assumed to contain both u and v channels, stacked in depth. Original code by Deqing
models/networks/flownet2_pytorch/utils/flow_utils.py:26
← previous401–455 of 455, ranked by callers