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
↓ 2 callersFunctionfunc
(x, a, b, c)
data/keypoint2img.py:260
↓ 2 callersFunctiongeneralNorm
(norm)
models/networks/architecture.py:40
↓ 2 callersMethodgenerate_images
(self, tgt_labels, tgt_images, ref_labels, ref_images, prevs=[None]*3)
models/vid2vid_model.py:130
↓ 2 callersMethodget_face_image
(self, keypoints, transform_L, size)
data/fewshot_face_dataset.py:155
↓ 2 callersMethodget_images
(self, img_paths, op_paths, dp_paths, ppl_indices, i, size, params, crop_coords, ref_face_pts=None)
data/fewshot_pose_dataset.py:143
↓ 2 callersFunctionget_part_mask
(pose)
models/input_process.py:64
↓ 2 callersMethodgetitem_by_path
(self, path, is_img=True)
data/lmdb_dataset.py:36
↓ 2 callersFunctionhas_overlap
(pose_pts_1, pose_pts_2)
data/preprocess/util/check_valid.py:130
↓ 2 callersMethodinit_temporal_model
(self)
models/base_model.py:259
↓ 2 callersFunctionis_full_body
(keypoint_dicts)
data/preprocess/util/check_valid.py:116
↓ 2 callersFunctionis_valid_frame
(args, img_path)
data/preprocess/util/check_valid.py:105
↓ 2 callersMethodload_data
()
data/base_data_loader.py:15
↓ 2 callersFunctionmkdir
(path)
util/util.py:153
↓ 2 callersMethodoption_file_path
(self, opt, makedir=False)
options/base_options.py:169
↓ 2 callersMethodparse
(self, save=False)
options/base_options.py:196
↓ 2 callersFunctionparse_flownetsd
(modules, weights, biases, param_prefix='netsd_')
models/networks/flownet2_pytorch/utils/param_utils.py:156
↓ 2 callersMethodprint_network
(self)
models/networks/base_network.py:77
↓ 2 callersMethodread_all_keypoints
(self, L_paths, crop_coords, is_ref)
data/fewshot_face_dataset.py:149
↓ 2 callersMethodrefine_face_region
(self, netGf, label_valid, fake_image, label, ref_label_valid, ref_image, ref_label)
models/face_refiner.py:24
↓ 2 callersMethodremove_dummy_from_tensor
(self, tensors, remove_size=0)
models/models.py:110
↓ 2 callersFunctionremove_invalid_frames
(args, video_idx)
data/preprocess/util/check_valid.py:17
↓ 2 callersFunctionremove_isolated_frames
(args, video_idx)
data/preprocess/util/check_valid.py:52
↓ 2 callersFunctionremove_static_frames
(args, video_idx)
data/preprocess/util/check_valid.py:27
↓ 2 callersMethodreshape_embed_input
(self, x)
models/networks/base_network.py:169
↓ 2 callersFunctionrun_openpose_cmd
(args, video_idx)
data/preprocess/util/get_poses.py:127
↓ 2 callersFunctionsave_models
(opt, epoch, epoch_iter, total_steps, visualizer, iter_path, model, end_of_epoch=False)
models/models.py:49
↓ 2 callersFunctionset_color
(im, yy, xx, color)
data/keypoint2img.py:267
↓ 2 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, normalize=True, tile=False)
util/util.py:43
↓ 2 callersFunctiontrain
(args, epoch, start_iteration, data_loader, model, optimizer, logger, is_validate=False, offset=0)
models/networks/flownet2_pytorch/main.py:244
↓ 2 callersMethodweight_generation
(self, img_ref, label_ref, label, t=0, img_coarse=None)
models/networks/generator.py:396
↓ 1 callersMethodGAN_matching_loss
(self, pred_real, pred_fake, for_discriminator=False)
models/loss_collector.py:206
↓ 1 callersMethodSPADE_combine
(self, encoded_label, ds_ref)
models/networks/generator.py:448
↓ 1 callersFunction__color_aug
(img, params)
data/base_dataset.py:164
↓ 1 callersFunction__crop
(img, pos, size)
data/base_dataset.py:153
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
models/networks/sync_batchnorm/batchnorm.py:80
↓ 1 callersFunction__flip
(img, flip)
data/base_dataset.py:159
↓ 1 callersMethod__init__
(self, requires_grad=False)
models/networks/vgg.py:14
↓ 1 callersMethod__init__
(self)
models/networks/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/networks/flownet2_pytorch/networks/correlation_package/correlation.py:56
↓ 1 callersMethod__iter__
(self)
models/networks/flownet2_pytorch/utils/tools.py:103
↓ 1 callersFunction__scale_image
(img, size, method=Image.BICUBIC)
data/base_dataset.py:149
↓ 1 callersMethod_compute_mean_std
Compute the mean and standard-deviation with sum and square-sum. This method also maintains the moving average on the master device.
models/networks/sync_batchnorm/batchnorm.py:113
↓ 1 callersMethod_shortcut
(self, x, label, conv_weights, norm_weights)
models/networks/architecture.py:101
↓ 1 callersMethodadd_dummy_to_tensor
(self, tensors, add_size=0)
models/models.py:100
↓ 1 callersMethodadd_table
(self, border=1)
util/html.py:40
↓ 1 callersMethodbatch_conv
(self, conv, x)
models/networks/discriminator.py:190
↓ 1 callersFunctioncheck_densepose_exists
(args, video_idx)
data/preprocess/util/check_valid.py:94
↓ 1 callersMethodcompute_VGG_losses
(self, fake_image, fake_raw_image, tgt_image, fg_mask_union)
models/loss_collector.py:122
↓ 1 callersMethodcompute_flow_losses
(self, flow, warped_image, tgt_image, flow_gt, flow_conf_gt, fg_mask, tgt_label, ref_label)
models/loss_collector.py:132
↓ 1 callersMethodcompute_kld
(self, x, label, img_coarse)
models/networks/generator.py:319
↓ 1 callersMethodcompute_loss
(self, x_vgg, y_vgg)
models/networks/loss.py:114
↓ 1 callersMethodcompute_mask_losses
(self, flow_mask, fake_image, warped_image, tgt_label, tgt_image, fake_raw_image, fg_mask, ref_fg_
models/loss_collector.py:164
↓ 1 callersFunctionconcat
(a, b, dim=0)
models/networks/base_network.py:49
↓ 1 callersFunctionconnect_keypoints
(opt, pts, edge_lists, size, basic_point_only, remove_face_labels)
data/keypoint2img.py:78
↓ 1 callersFunctioncontains_non_overlapping_people
(keypoint_dicts)
data/preprocess/util/check_valid.py:142
↓ 1 callersMethodcreate_singleD
(self, opt, subarch, input_nc, ndf, n_layers, norm_layer, getIntermFeat, stride)
models/networks/discriminator.py:30
↓ 1 callersFunctiondefine_edge_lists
(basic_point_only)
data/keypoint2img.py:205
↓ 1 callersMethoddefine_losses
(self)
models/loss_collector.py:27
↓ 1 callersMethoddefine_networks
(self, start_epoch)
models/base_model.py:167
↓ 1 callersFunctiondetect_motion
(keypoint_dicts_1, keypoint_dicts_2)
data/preprocess/util/check_valid.py:73
↓ 1 callersMethoddiscriminate_face
(self, netDf, fake_image, tgt_label, tgt_image, ref_label, ref_image, faceRefiner, for_discriminator)
models/loss_collector.py:70
↓ 1 callersMethoddisplay_current_results
(self, visuals, epoch, step)
util/visualizer.py:90
↓ 1 callersMethoddisplay_visdom_results
(self, visuals, epoch, step)
util/visualizer.py:52
↓ 1 callersMethoddivide_pred
(self, pred)
models/base_model.py:141
↓ 1 callersFunctiondivide_sequences
(args, video_idx)
data/preprocess/util/track.py:122
↓ 1 callersMethoddownload_all
( self, download_path=None, prefix_number=True, reverse_numbering=False,
data/preprocess/download_youTube_playlist.py:7
↓ 1 callersFunctionencode_input
(opt, data_list, dummy_bs)
models/input_process.py:10
↓ 1 callersMethodend_of_epoch
(self, model)
models/trainer.py:86
↓ 1 callersMethodend_of_iter
(self, loss_dicts, output_list, model)
models/trainer.py:53
↓ 1 callersFunctionextract_all_frames
(args, video_path)
data/preprocess/util/get_poses.py:110
↓ 1 callersFunctionextract_key_frames
(args, video_path, img_dir)
data/preprocess/util/get_poses.py:72
↓ 1 callersFunctionextract_valid_frames
(args, video_path, img_dir)
data/preprocess/util/get_poses.py:86
↓ 1 callersFunctionextract_valid_keypoints
(pts, edge_lists)
data/keypoint2img.py:56
↓ 1 callersMethodfinetune
(self, ref_labels, ref_images)
models/vid2vid_model.py:207
↓ 1 callersMethodflow_generation
(self, label, label_refs, img_refs, prev, atn, ref_idx)
models/networks/generator.py:424
↓ 1 callersMethodforward
(self, input1)
models/networks/flownet2_pytorch/networks/channelnorm_package/channelnorm.py:37
↓ 1 callersMethodforward
(self, input1, input2)
models/networks/flownet2_pytorch/networks/correlation_package/correlation.py:65
↓ 1 callersMethodforward
(self, input1, input2)
models/networks/flownet2_pytorch/networks/resample2d_package/resample2d.py:44
↓ 1 callersMethodforward_face
(self, label, label_refs, img_refs, img_coarse)
models/networks/generator.py:232
↓ 1 callersMethodgather_options
(self)
options/base_options.py:134
↓ 1 callersMethodgen_conv_weights
(self, encoded_ref)
models/networks/discriminator.py:142
↓ 1 callersFunctiongeneralConv
(adaptive=False, transpose=False)
models/networks/architecture.py:19
↓ 1 callersMethodget_SPADE_weights
(self, x, i)
models/networks/generator.py:245
↓ 1 callersFunctionget_confirm_token
(response)
scripts/download_gdrive.py:19
↓ 1 callersMethodget_conv_weights
(self, x, i)
models/networks/generator.py:276
↓ 1 callersMethodget_crop_coords
(self, pose_pts, size, offset=None)
data/fewshot_pose_dataset.py:210
↓ 1 callersFunctionget_data_t
(data, n_frames_load, t)
train.py:70
↓ 1 callersFunctionget_grid
(batchsize, rows, cols, gpu_id=0)
models/networks/base_network.py:13
↓ 1 callersMethodget_image_dir
(self)
util/html.py:33
↓ 1 callersMethodget_input_t
(self, tgt_labels, tgt_images, prevs, t)
models/vid2vid_model.py:160
↓ 1 callersMethodget_label_tensor
(self, label_path, transform_label)
data/fewshot_street_dataset.py:111
↓ 1 callersFunctionget_norm
(norm)
models/networks/architecture.py:42
↓ 1 callersMethodget_target_tensor
(self, input, target_is_real)
models/networks/loss.py:39
↓ 1 callersMethodget_train_params
(self, netG, train_names)
models/base_model.py:149
↓ 1 callersFunctioninference
(args, epoch, data_loader, model, offset=0)
models/networks/flownet2_pytorch/main.py:341
↓ 1 callersMethodinference
(self, tgt_label, ref_labels, ref_images)
models/vid2vid_model.py:179
↓ 1 callersFunctioninit_dist
(launcher='pytorch', backend='nccl', **kwargs)
util/distributed.py:15
↓ 1 callersMethodinit_temporal_network
(self)
models/networks/generator.py:155
↓ 1 callersMethodinitialize
(self, opt)
data/base_data_loader.py:11
↓ 1 callersMethodinitialize
(self, opt)
data/custom_dataset_data_loader.py:16
← previousnext →101–200 of 455, ranked by callers