MCPcopy Create free account

hub / github.com/NVIDIA/vid2vid / functions

Functions345 in github.com/NVIDIA/vid2vid

⨍ Functions 345◇ Types & classes 78
Functiongpumemusage
()
models/flownet2_pytorch/utils/tools.py:117
Functionhook
(grad)
models/flownet2_pytorch/networks/submodules.py:73
Functioninit_deconv_bilinear
(weight)
models/flownet2_pytorch/networks/submodules.py:56
Methodinit_deconv_bilinear
(self, weight)
models/flownet2_pytorch/models.py:80
Methodinitialize
(self, opt)
data/pose_dataset.py:12
Methodinitialize
(self, opt)
data/base_data_loader.py:6
Methodinitialize
(self, opt)
data/base_dataset.py:16
Methodinitialize
(self, opt)
data/temporal_dataset.py:12
Methodinitialize
(self, opt)
data/face_dataset.py:14
Methodinitialize
(self)
options/train_options.py:5
Methodinitialize
(self)
options/test_options.py:5
Methodinitialize
(self, opt)
models/base_model.py:10
Functionkwargs_from_args
(args, argument_for_class)
models/flownet2_pytorch/utils/tools.py:84
Methodload_data
(self)
data/custom_dataset_data_loader.py:40
Functionmkdirs
(paths)
util/util.py:141
Methodn_of_seqs
(self)
data/test_dataset.py:73
Methodname
(self)
data/pose_dataset.py:137
Methodname
(self)
data/base_dataset.py:13
Methodname
(self)
data/temporal_dataset.py:76
Methodname
(self)
data/face_dataset.py:212
Methodname
(self)
models/vid2vid_model_D.py:14
Methodname
(self)
models/base_model.py:7
Methodname
(self)
models/flownet.py:7
Methodname
(self)
models/vid2vid_model_G.py:16
Methodnormalize_pose
(self, A_img, target_yc, target_len, first=False)
data/pose_dataset.py:100
Methodoptimize_parameters
(self)
models/base_model.py:30
Functionprint_network
(net)
models/networks.py:70
Functionprint_numpy
(x, val=True, shp=False)
util/util.py:132
Methodquery
(self, images)
util/image_pool.py:12
FunctionreadFlow
Read .flo file in Middlebury format
models/flownet2_pytorch/utils/flow_utils.py:5
Functionread_gen
(file_name)
models/flownet2_pytorch/utils/frame_utils.py:6
Functionremove_dummy_from_tensor
(tensors, remove_size=0)
util/util.py:119
Methodsave
(self, label)
models/vid2vid_model_D.py:266
Methodsave
(self, label)
models/vid2vid_model_G.py:338
Functionsave_all_tensors
(opt, real_A, fake_B, fake_B_first, fake_B_raw, real_B, flow_ref, conf_ref, flow, weight, modelD)
util/util.py:13
Functionsave_checkpoint
(state, is_best, path, prefix, filename='checkpoint.pth.tar')
models/flownet2_pytorch/utils/tools.py:138
Functionsave_grad
(grads, name)
models/flownet2_pytorch/networks/submodules.py:72
Functionsave_image
(image_numpy, image_path)
util/util.py:128
Methodset_input
(self, input)
models/base_model.py:17
Methodtest
(self)
models/base_model.py:24
FunctiontoTensor_normalize
()
data/base_dataset.py:154
Functionupdate_hyperparameter_schedule
(args, epoch, global_iteration, optimizer)
models/flownet2_pytorch/utils/tools.py:131
Methodupdate_training_batch
(self, ratio)
data/base_dataset.py:19
Functionweights_init
(m)
models/networks.py:15
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/flownet2_pytorch/utils/flow_utils.py:26
← previous301–345 of 345, ranked by callers