MCPcopy Create free account

hub / github.com/NVIDIA/vid2vid / functions

Functions345 in github.com/NVIDIA/vid2vid

↓ 39 callersFunctionconv
(batchNorm, in_planes, out_planes, kernel_size=3, stride=1)
models/flownet2_pytorch/networks/submodules.py:7
↓ 25 callersMethodlog
(self, string)
models/flownet2_pytorch/utils/tools.py:42
↓ 18 callersFunctionpredict_flow
(in_planes)
models/flownet2_pytorch/networks/submodules.py:31
↓ 16 callersMethod__init__
(self, requires_grad=False)
models/networks.py:841
↓ 15 callersMethodsave
(self, label)
models/base_model.py:39
↓ 14 callersFunctiondeconv
(in_planes, out_planes)
models/flownet2_pytorch/networks/submodules.py:34
↓ 11 callersFunctionmake_grouped_dataset
(dir)
data/image_folder.py:36
↓ 10 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, normalize=True)
util/util.py:48
↓ 8 callersFunctionconcat_frame
(A, Ai, nF)
data/base_dataset.py:204
↓ 8 callersFunctionget_transform
(opt, params, method=Image.BICUBIC, normalize=True, toTensor=True)
data/base_dataset.py:130
↓ 8 callersMethodresample
(self, image, flow)
models/base_model.py:189
↓ 7 callersFunctioncheck_path_valid
(A_paths, B_paths)
data/image_folder.py:50
↓ 7 callersFunctionparse_flownets
(modules, weights, biases, param_prefix='net2_')
models/flownet2_pytorch/utils/param_utils.py:51
↓ 6 callersMethod__init__
(self, args=None, batchNorm=False, div_flow = 20., fp16=False)
models/flownet2_pytorch/models.py:32
↓ 6 callersMethod__init__
(self, args, is_cropped = False, root = '', dstype = 'clean', replicates = 1)
models/flownet2_pytorch/datasets.py:31
↓ 6 callersMethodcrop
(self, img)
data/face_dataset.py:171
↓ 6 callersMethodforward
(self, scale_T, tensors_list, dummy_bs=0)
models/vid2vid_model_D.py:93
↓ 6 callersFunctioni_conv
(batchNorm, in_planes, out_planes, kernel_size=3, stride=1, bias = True)
models/flownet2_pytorch/networks/submodules.py:20
↓ 5 callersMethodadd_images
(self, ims, txts, links, width=400, height=0)
util/html.py:33
↓ 5 callersMethodconcat
(self, tensors, dim=0)
models/base_model.py:109
↓ 5 callersFunctiondrawEdge
(im, x, y, bw=1, color=(255,255,255), draw_end_points=False)
data/keypoint2img.py:26
↓ 5 callersFunctionparse_flownetc
(modules, weights, biases)
models/flownet2_pytorch/utils/param_utils.py:5
↓ 4 callersFunctionEPE
(input_flow, target_flow)
models/flownet2_pytorch/losses.py:11
↓ 4 callersMethod__init__
(self)
models/flownet2_pytorch/losses.py:15
↓ 4 callersMethodbackward
(ctx, grad_output)
models/flownet2_pytorch/networks/resample2d_package/resample2d.py:24
↓ 4 callersMethodbuild_pyr
(self, tensor, nearest=False)
models/base_model.py:122
↓ 4 callersMethodforward
(self, input_A, input_B, inst_A, fake_B_prev, dummy_bs=0)
models/vid2vid_model_G.py:114
↓ 4 callersFunctionget_img_params
(opt, size)
data/base_dataset.py:85
↓ 4 callersFunctionget_skipped_frames
(B_all, B, t_scales, tD)
models/vid2vid_model_D.py:275
↓ 4 callersFunctionget_skipped_frames_sparse
(B_all, B, t_scales, tD, n_frames_load, i, is_flow=False)
models/vid2vid_model_D.py:304
↓ 4 callersFunctioninterpPoints
(x, y)
data/keypoint2img.py:45
↓ 4 callersMethodload_network
(self, network, network_label, epoch_label, save_dir='')
models/base_model.py:63
↓ 4 callersFunctionmake_power_2
(n, base=32.0)
data/base_dataset.py:82
↓ 4 callersMethodsave
(self)
util/html.py:48
↓ 4 callersMethodsave_network
(self, network, network_label, epoch_label, gpu_ids)
models/base_model.py:43
↓ 4 callersMethodupdate_learning_rate
(self, epoch, model)
models/base_model.py:154
↓ 4 callersMethodupdate_training_batch
(self, ratio)
models/base_model.py:169
↓ 3 callersMethodadd_header
(self, str)
util/html.py:25
↓ 3 callersMethodcompute_loss_D
(self, netD, real_A, real_B, fake_B)
models/vid2vid_model_D.py:168
↓ 3 callersMethodcrop
(self, Ai)
data/pose_dataset.py:93
↓ 3 callersMethodget_image
(self, A_path, size, params, input_type)
data/pose_dataset.py:69
↓ 3 callersMethodget_image
(self, A_path, transform_scaleA, is_label=False)
data/test_dataset.py:63
↓ 3 callersMethodget_image
(self, A_path, transform_scaleA, is_label=False)
data/temporal_dataset.py:66
↓ 3 callersFunctionget_video_params
(opt, n_frames_total, cur_seq_len, index)
data/base_dataset.py:177
↓ 3 callersMethodinit_frame_idx
(self, A_paths)
data/base_dataset.py:26
↓ 3 callersMethodinitialize
(self, opt)
models/flownet.py:10
↓ 3 callersFunctionloss_backward
(opt, loss, optimizer)
train.py:130
↓ 3 callersMethodnorm
(self, t)
models/flownet.py:61
↓ 3 callersMethodremove_dummy_from_tensor
(self, tensors, remove_size=0)
models/models.py:52
↓ 3 callersMethodscale_points
(self, keypoints, part, index, sym=False)
data/face_dataset.py:177
↓ 3 callersMethodupdate_frame_idx
(self, A_paths, index)
data/base_dataset.py:40
↓ 3 callersMethodvis_print
(self, message)
util/visualizer.py:138
↓ 2 callersFunctionCreateDataLoader
(opt)
data/data_loader.py:2
↓ 2 callersMethodGAN_and_FM_loss
(self, pred_real, pred_fake)
models/vid2vid_model_D.py:199
↓ 2 callersMethodcompute_flow_and_conf
(self, im1, im2)
models/flownet.py:43
↓ 2 callersMethodcompute_mask
(self, real_As, ts, te=None)
models/vid2vid_model_G.py:322
↓ 2 callersFunctioncreate_model
(opt)
models/models.py:61
↓ 2 callersFunctiondownload_file_from_google_drive
(id, destination)
scripts/download_gdrive.py:3
↓ 2 callersFunctiondownload_file_from_google_drive
(id, destination)
scripts/street/download_gdrive.py:3
↓ 2 callersMethodencode_input
(self, input_map, real_image, inst_map=None)
models/vid2vid_model_G.py:86
↓ 2 callersMethodgenerate_first_frame
(self, real_A, real_B, pool_map=None)
models/vid2vid_model_G.py:231
↓ 2 callersMethodget_face_region
(self, real_A)
models/vid2vid_model_D.py:215
↓ 2 callersFunctionget_grid
(batchsize, rows, cols, gpu_id=0, dtype=torch.float32)
models/networks.py:79
↓ 2 callersFunctionget_norm_layer
(norm_type='instance')
models/networks.py:23
↓ 2 callersMethodget_target_tensor
(self, input, target_is_real)
models/networks.py:745
↓ 2 callersFunctionis_image_file
(filename)
data/image_folder.py:21
↓ 2 callersMethodload_data
()
data/base_data_loader.py:10
↓ 2 callersFunctionmkdir
(path)
util/util.py:148
↓ 2 callersMethodparse
(self, save=True)
options/base_options.py:96
↓ 2 callersFunctionparse_flownetsd
(modules, weights, biases, param_prefix='netsd_')
models/flownet2_pytorch/utils/param_utils.py:156
↓ 2 callersMethodparse_str
(self, ids)
options/base_options.py:87
↓ 2 callersMethodresample
(self, image, flow)
models/networks.py:108
↓ 2 callersFunctionsave_models
(opt, epoch, epoch_iter, total_steps, visualizer, iter_path, modelG, modelD, end_of_epoch=False)
models/models.py:135
↓ 2 callersFunctionsetColor
(im, yy, xx, color)
data/keypoint2img.py:15
↓ 2 callersFunctiontensor2flow
(output, imtype=np.uint8)
util/util.py:89
↓ 2 callersFunctiontrain
(args, epoch, start_iteration, data_loader, model, optimizer, logger, is_validate=False, offset=0)
models/flownet2_pytorch/main.py:244
↓ 2 callersFunctionunzip_file
(file_name, unzip_path)
scripts/street/download_gdrive.py:24
↓ 2 callersMethodupdate_fixed_params
(self)
models/base_model.py:161
↓ 2 callersFunctionwrap_model
(opt, modelG, modelD, flowNet)
models/models.py:10
↓ 1 callersFunctionCreateDataset
(opt)
data/custom_dataset_data_loader.py:5
↓ 1 callersFunction__crop
(img, size, pos)
data/base_dataset.py:164
↓ 1 callersFunction__flip
(img, flip)
data/base_dataset.py:172
↓ 1 callersMethod__init__
(self)
models/flownet2_pytorch/networks/submodules.py:41
↓ 1 callersMethod__init__
(self, pad_size=0, kernel_size=0, max_displacement=0, stride1=1, stride2=2, corr_multiply=1)
models/flownet2_pytorch/networks/correlation_package/correlation.py:48
↓ 1 callersMethod__iter__
(self)
models/flownet2_pytorch/utils/tools.py:103
↓ 1 callersFunction__scale_image
(img, size, method=Image.BICUBIC)
data/base_dataset.py:160
↓ 1 callersMethodadd_dummy_to_tensor
(self, tensors, add_size=0)
models/models.py:42
↓ 1 callersMethodadd_table
(self, border=1)
util/html.py:29
↓ 1 callersMethodbuild_conv_block
(self, dim, padding_type, norm_layer, activation, use_dropout)
models/networks.py:559
↓ 1 callersMethodcompute_fake_B_prev
(self, real_B_prev, fake_B_last, fake_B)
models/vid2vid_model_G.py:332
↓ 1 callersMethodcompute_loss_D_T
(self, real_B, fake_B, flow_ref, conf_ref, scale_T)
models/vid2vid_model_D.py:181
↓ 1 callersFunctionconnect_keypoints
(pts, edge_lists, size, random_drop_prob, remove_face_labels, basic_point_only)
data/keypoint2img.py:106
↓ 1 callersFunctioncreate_optimizer
(opt, models)
models/models.py:86
↓ 1 callersFunctiondefine_edge_lists
(basic_point_only)
data/keypoint2img.py:151
↓ 1 callersMethoddisplay_current_results
(self, visuals, epoch, step)
util/visualizer.py:38
↓ 1 callersMethoddists_min
(self, a, b, num=1)
models/base_model.py:136
↓ 1 callersFunctiondownload_file_from_google_drive
(id, destination)
scripts/face/download_gdrive.py:3
↓ 1 callersMethoddraw_face_edges
(self, keypoints, part_list, transform_A, size, add_dist_map)
data/face_dataset.py:129
↓ 1 callersFunctionextract_valid_keypoints
(pts, edge_lists)
data/keypoint2img.py:85
↓ 1 callersMethodforward
(self, input1)
models/flownet2_pytorch/networks/channelnorm_package/channelnorm.py:37
next →1–100 of 345, ranked by callers