MCPcopy Create free account

hub / github.com/Yipin/Dance_Dance_Generation / functions

Functions208 in github.com/Yipin/Dance_Dance_Generation

↓ 16 callersMethodsave
(self)
util/html.py:44
↓ 10 callersFunctionmake_dataset
(dir)
data/image_folder.py:22
↓ 9 callersMethod_make_seq_res_blocks
(self, num_of_feat, num_of_module)
nets_mula/hourglass_based_network.py:103
↓ 9 callersMethodforward
(self, trans_segs, ref_poses, background_image, ref_frames_foreground, ref_frames, infer=False)
models/twostage_model.py:234
↓ 8 callersMethod__init__
(self, requires_grad=False)
models/networks.py:392
↓ 8 callersMethodload_network
(self, network, network_label, epoch_label, save_dir='')
models/base_model.py:52
↓ 8 callersMethodsave_network
(self, network, network_label, epoch_label, gpu_ids)
models/base_model.py:44
↓ 7 callersFunctionpreprocess
(image)
data/preprocess_dataset.py:27
↓ 6 callersMethodforward
(self, label, inst, image, feat, infer=False)
models/pix2pixHD_model.py:152
↓ 6 callersMethodset_features
(self, idx, feat, cluster_idx)
models/ui_model.py:328
↓ 6 callersMethodsingle_forward
(self, net_input, feat_map)
models/ui_model.py:217
↓ 5 callersMethodadd_images
(self, ims, txts, links, width=512)
util/html.py:32
↓ 4 callersMethod__init__
(self, num_of_feat=256, num_of_module=1)
nets_mula/hourglass_based_network.py:84
↓ 4 callersMethod_make_multi_stage_conv1x1
(self, in_num_of_feat, out_num_of_feat, num_of_stages)
nets_mula/hourglass_based_network.py:186
↓ 4 callersMethodget_edges
(self, t)
models/ui_model.py:110
↓ 4 callersMethodget_green_mask
(self, frame)
models/twostage_model.py:226
↓ 3 callersMethodadd_header
(self, str)
util/html.py:24
↓ 3 callersMethodbackup_current_state
(self)
models/ui_model.py:288
↓ 3 callersFunctionget_transform
(opt, params, method=Image.BICUBIC, normalize=True)
data/base_dataset.py:35
↓ 3 callersMethodparse
(self, save=True)
options/base_options.py:72
↓ 3 callersMethodquery
(self, images)
util/image_pool.py:11
↓ 2 callersFunctionCreateDataLoader
(opt)
data/data_loader.py:2
↓ 2 callersMethodcomb_discriminate
(self, input_label, test_image, use_pool=False)
models/twostage_model.py:190
↓ 2 callersMethodcompute_gradient_penalty
(self, input_concat, real_data, fake_data, not_comb=True)
models/twostage_model.py:199
↓ 2 callersFunctioncreate_model
(opt)
models/models.py:5
↓ 2 callersMethoddiscriminate
(self, input_label, test_image, use_pool=False)
models/pix2pixHD_model.py:144
↓ 2 callersMethoddiscriminate
(self, input_label, test_image, use_pool=False)
models/twostage_model.py:182
↓ 2 callersMethodencode_input
(self, label_map, inst_map=None, real_image=None, feat_map=None, infer=False)
models/pix2pixHD_model.py:111
↓ 2 callersMethodencode_input
(self, trans_segs, ref_poses, background_image, ref_frames_foreground=None, ref_frames=None, infer=False)
models/twostage_model.py:161
↓ 2 callersFunctionget_norm_layer
(norm_type='instance')
models/networks.py:21
↓ 2 callersMethodget_target_tensor
(self, input, target_is_real)
models/networks.py:85
↓ 2 callersMethodinference
(self, trans_segs, ref_poses, background_image, ref_frames_foreground, ref_frames)
models/twostage_model.py:365
↓ 2 callersMethodload_data
()
data/base_data_loader.py:10
↓ 2 callersMethodloss_filter
(g_gan, g_gan_feat, g_vgg, d_real, d_fake)
models/pix2pixHD_model.py:15
↓ 2 callersMethodloss_filter
(g_gan, g_gan_feat, g_vgg, d_real, d_fake, d_gp, g_sp, comb_g_gan, comb_g_gan_feat, co
models/twostage_model.py:30
↓ 2 callersFunctionmkdir
(path)
util/util.py:48
↓ 2 callersMethodsave_image_batch
(self, outputs_dict, inds, epoch_num, total_step, out_root=None)
util/visualizer.py:241
↓ 2 callersFunctiontime_inference
(engine, batch_size, inp)
run_engine.py:104
↓ 2 callersMethodtoTensor
(self, img, normalize=False)
models/ui_model.py:35
↓ 1 callersFunctionCreateDataset
(opt)
data/custom_dataset_data_loader.py:5
↓ 1 callersFunctionMuLA_HG_MSRAInit
(stage_num_of_encoder=1, module_num_of_pose_encoder=1, num_of_joint=16, module_num_of_parsing_encoder=1, num_o
nets_mula/hourglass_based_network.py:323
↓ 1 callersFunction__crop
(img, pos, size)
data/base_dataset.py:81
↓ 1 callersFunction__flip
(img, flip)
data/base_dataset.py:89
↓ 1 callersMethod__init__
(self, gpu_ids)
models/networks_sp.py:21
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride, padding, dilation, transposed, output
nets_mula/adaptive_conv.py:10
↓ 1 callersMethod__init__
(self, cfg='VGG16', batch_norm=False)
nets_mula/vgg_based_network.py:10
↓ 1 callersMethod__init__
(self, front_end_net)
nets_mula/network_init.py:37
↓ 1 callersFunction__make_power_2
(img, base, method=Image.BICUBIC)
data/base_dataset.py:65
↓ 1 callersFunction__scale_width
(img, target_width, method=Image.BICUBIC)
data/base_dataset.py:73
↓ 1 callersMethod_initialize_weights
(self)
nets_mula/network_init.py:19
↓ 1 callersMethod_initialize_weights
(self)
nets_mula/network_init.py:46
↓ 1 callersMethod_make_layers
(self, cfg, batch_norm=False)
nets_mula/vgg_based_network.py:20
↓ 1 callersMethod_make_multi_stage_hg
(self, num_of_feat, num_of_module, num_of_stages)
nets_mula/hourglass_based_network.py:180
↓ 1 callersMethodadd_table
(self, border=1)
util/html.py:28
↓ 1 callersMethodbuild_conv_block
(self, dim, padding_type, norm_layer, activation, use_dropout)
models/networks.py:224
↓ 1 callersMethodcombine_segs
(self, seg_batch)
util/visualizer.py:216
↓ 1 callersFunctionconvert_to_datatype
(v)
run_engine.py:123
↓ 1 callersMethodcopy_features
(self, idx_src, idx_tgt)
models/ui_model.py:333
↓ 1 callersFunctioncreate_memory
(engine, name, buf, mem, batchsize, inp, inp_idx)
run_engine.py:82
↓ 1 callersMethodforward
(self, X)
models/networks.py:414
↓ 1 callersMethodget_crop_region
(self, idx)
models/ui_model.py:295
↓ 1 callersMethodget_edges
(self, t)
models/pix2pixHD_model.py:262
↓ 1 callersMethodget_image_dir
(self)
util/html.py:21
↓ 1 callersFunctionget_input_output_names
(trt_engine)
run_engine.py:62
↓ 1 callersFunctionget_params
(opt, size)
data/base_dataset.py:19
↓ 1 callersMethodinference
(self, label, inst, image=None)
models/pix2pixHD_model.py:195
↓ 1 callersMethodinit_loss_filter
(self, use_gan_feat_loss, use_vgg_loss)
models/pix2pixHD_model.py:13
↓ 1 callersMethodinit_loss_filter
(self, use_gan_feat_loss, use_vgg_loss)
models/twostage_model.py:25
↓ 1 callersMethodinitialize
(self, opt)
data/custom_dataset_data_loader.py:18
↓ 1 callersMethodinitialize
(self, opt)
data/aligned_dataset.py:9
↓ 1 callersMethodinitialize
(self)
options/base_options.py:17
↓ 1 callersMethodinitialize
(self, opt)
models/ui_model.py:17
↓ 1 callersFunctionis_image_file
(filename)
data/image_folder.py:18
↓ 1 callersFunctionlabelcolormap
(N)
util/util.py:61
↓ 1 callersMethodname
(self)
data/custom_dataset_data_loader.py:15
↓ 1 callersMethodname
(self)
data/aligned_dataset.py:77
↓ 1 callersMethodname
(self)
models/ui_model.py:14
↓ 1 callersFunctionnormalize
()
data/base_dataset.py:62
↓ 1 callersMethodplot_current_errors
(self, errors, step)
util/visualizer.py:176
↓ 1 callersMethodprint_current_errors
(self, epoch, i, errors, t)
util/visualizer.py:183
↓ 1 callersMethodprint_layer_times
(self)
run_engine.py:54
↓ 1 callersMethodreset_parameters
(self)
nets_mula/adaptive_conv.py:34
↓ 1 callersFunctionrun_onnx
(onnx_file, data_type, bs, inp)
run_engine.py:138
↓ 1 callersFunctionrun_trt_engine
(engine_file, bs, it)
run_engine.py:134
↓ 1 callersMethodsample_features
(self, inst)
models/pix2pixHD_model.py:219
↓ 1 callersMethodsave
(self, label)
models/base_model.py:40
↓ 1 callersMethodsingleD_forward
(self, model, input)
models/networks.py:315
↓ 1 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, normalize=True)
util/util.py:10
↓ 1 callersFunctionuint82bin
returns the binary of integer n, count refers to amount of bits
util/util.py:57
↓ 1 callersMethodupdate_fixed_params
(self)
models/twostage_model.py:487
↓ 1 callersMethodupdate_learning_rate
()
models/base_model.py:92
FunctionHG_with_GaussianInit
(num_of_feat=256, num_of_class=16, num_of_module=1, num_of_stages=8)
nets_mula/hourglass_based_network.py:316
FunctionHG_with_MSRAInit
(num_of_feat=256, num_of_class=16, num_of_module=1, num_of_stages=8)
nets_mula/hourglass_based_network.py:309
FunctionMuLA_HG_GaussianInit
(stage_num_of_encoder=5, module_num_of_pose_encoder=1, num_of_joint=16, module_num_of_parsing_encoder=1, num_o
nets_mula/hourglass_based_network.py:332
FunctionMuLA_VGG_GaussianInit
(pose_encoder_cfg='VGG16', num_of_joint=16, parsing_encoder_cfg='VGG16', num_of_part=20, batch_norm=False, num
nets_mula/vgg_based_network.py:140
FunctionMuLA_VGG_MSRAInit
(pose_encoder_cfg='VGG16', num_of_joint=16, parsing_encoder_cfg='VGG16', num_of_part=20, batch_norm=False, num
nets_mula/vgg_based_network.py:131
FunctionVGG_with_GaussianInit
(cfg='VGG16', batch_norm=False)
nets_mula/vgg_based_network.py:127
FunctionVGG_with_MSRAInit
(cfg='VGG16', batch_norm=False)
nets_mula/vgg_based_network.py:123
Method__call__
(self, input, _input, target_is_real)
models/networks.py:103
Method__call__
(self, gray_image)
util/util.py:90
next →1–100 of 208, ranked by callers