MCPcopy Create free account

hub / github.com/AliaksandrSiarohin/monkey-net / functions

Functions140 in github.com/AliaksandrSiarohin/monkey-net

⨍ Functions 140◇ Types & classes 41
↓ 9 callersFunctioncolor_as_num
(val)
sup-mat/MGif/bg_removal_tool.py:130
↓ 6 callersMethod__init__
(self, block_expansion, in_features, num_blocks=3, max_features=256, temporal=False)
modules/util.py:134
↓ 5 callersFunction_unsqueeze_ft
add new dementions at the front and the tail
sync_batchnorm/batchnorm.py:29
↓ 5 callersMethodload_cpk
(checkpoint_path, generator=None, discriminator=None, kp_detector=None, optimizer_generator=N
logger.py:50
↓ 5 callersFunctionmake_coordinate_grid
Create a meshgrid [-1,1] x [-1,1] of given spatial_size.
modules/util.py:26
↓ 3 callersMethodget
(self)
sync_batchnorm/comm.py:32
↓ 3 callersFunctionmean_batch
(val)
modules/losses.py:4
↓ 3 callersMethodput
(self, result)
sync_batchnorm/comm.py:26
↓ 3 callersFunctionread_video
(name, image_shape)
frames_dataset.py:14
↓ 3 callersFunctionreconstruction_loss
(prediction, target, weight)
modules/losses.py:8
↓ 3 callersFunctionsplit_kp
(kp_joined, detach=False)
train.py:14
↓ 3 callersMethodvisualize_reconstruction
(self, inp, out)
logger.py:154
↓ 2 callersMethod_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:184
↓ 2 callersFunction_sum_ft
sum over the first and last dimention
sync_batchnorm/batchnorm.py:24
↓ 2 callersFunctionas_numpy
(v)
sync_batchnorm/unittest.py:17
↓ 2 callersMethodcreate_image_grid
(self, *args)
logger.py:119
↓ 2 callersMethodcreate_video_column
(self, videos)
logger.py:112
↓ 2 callersMethoddeform_input
(self, inp, deformations_absolute)
modules/generator.py:51
↓ 2 callersFunctionexecute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
sync_batchnorm/replicate.py:27
↓ 2 callersFunctiongenerate
(generator, appearance_image, kp_appearance, kp_video)
reconstruction.py:12
↓ 2 callersMethodget_params
(self, brightness, contrast, saturation, hue)
augmentation.py:236
↓ 2 callersFunctionget_resize_sizes
(im_h, im_w, size)
augmentation.py:81
↓ 2 callersFunctionkp2gaussian
Transform a keypoint into gaussian like representation
modules/keypoint_detector.py:7
↓ 2 callersFunctionmatrix_inverse
(batch_of_matrix, eps=0)
modules/util.py:206
↓ 2 callersFunctionmv_all_images
(images, in_folder, out_folder)
data/split.py:21
↓ 2 callersMethodnormalize_heatmap
(self, heatmap)
modules/movement_embedding.py:33
↓ 2 callersMethodsave_cpk
(self)
logger.py:43
↓ 2 callersFunctiontransfer_one
(generator, kp_detector, source_image, driving_video, transfer_params)
transfer.py:65
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
sync_batchnorm/batchnorm.py:80
↓ 1 callersMethod__init__
(self, kp_extractor, generator, discriminator, train_params)
train.py:29
↓ 1 callersMethod__init__
(self, in_features, out_features, norm=False, kernel_size=4)
modules/discriminator.py:12
↓ 1 callersMethod__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
sync_batchnorm/comm.py:67
↓ 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.
sync_batchnorm/batchnorm.py:113
↓ 1 callersFunctionconvert_gif_to_frames
(gif)
sup-mat/MGif/bg_removal_tool.py:104
↓ 1 callersMethodcreate_video_column_with_kp
(self, video, kp)
logger.py:108
↓ 1 callersFunctioncrop_clip
(clip, min_h, min_w, h, w)
augmentation.py:20
↓ 1 callersFunctiondiscriminator_gan_loss
(discriminator_maps_generated, discriminator_maps_real, weight)
modules/losses.py:20
↓ 1 callersFunctiondiscriminator_loss
(discriminator_maps_generated, discriminator_maps_real, loss_weights)
modules/losses.py:63
↓ 1 callersFunctiondiscriminator_loss_names
()
modules/losses.py:42
↓ 1 callersMethoddraw_video_with_kp
(self, video, kp_array)
logger.py:97
↓ 1 callersFunctiongaussian2kp
Extract the mean and the variance from a heatmap
modules/keypoint_detector.py:43
↓ 1 callersFunctiongenerator_gan_loss
(discriminator_maps_generated, weight)
modules/losses.py:14
↓ 1 callersFunctiongenerator_loss
(discriminator_maps_generated, discriminator_maps_real, video_deformed, loss_weights)
modules/losses.py:46
↓ 1 callersFunctiongenerator_loss_names
(loss_weights)
modules/losses.py:27
↓ 1 callersFunctionget_files_by_file_size
Return list of file paths sorted by file size
sup-mat/MGif/bg_removal_tool.py:68
↓ 1 callersMethodlog_epoch
(self, epoch, models)
logger.py:84
↓ 1 callersMethodlog_iter
(self, it, names, values, inp, out)
logger.py:76
↓ 1 callersMethodlog_scores
(self, loss_names)
logger.py:29
↓ 1 callersFunctionmake_symetric_matrix
(torch_matrix)
transfer.py:17
↓ 1 callersMethodnet
(self, input, h=None)
modules/prediction_module.py:19
↓ 1 callersFunctionnormalize_kp
(kp_video, kp_appearance, movement_mult=False, move_location=False, adapt_variance=False, cli
transfer.py:31
↓ 1 callersFunctionpad_clip
(clip, h, w)
augmentation.py:34
↓ 1 callersFunctionprediction
(config, generator, kp_detector, checkpoint, log_dir)
prediction.py:35
↓ 1 callersFunctionprocess
(video, filename, outdir, skipdir)
sup-mat/MGif/bg_removal_tool.py:135
↓ 1 callersFunctionreconstruction
(config, generator, kp_detector, checkpoint, log_dir, dataset)
reconstruction.py:28
↓ 1 callersMethodregister_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
sync_batchnorm/comm.py:84
↓ 1 callersFunctionresize_clip
(clip, size, interpolation='bilinear')
augmentation.py:42
↓ 1 callersMethodrun_master
Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master devi
sync_batchnorm/comm.py:102
↓ 1 callersMethodrun_slave
(self, msg)
sync_batchnorm/comm.py:49
↓ 1 callersFunctionsmallest_singular
(batch_of_matrix)
modules/util.py:244
↓ 1 callersFunctiontrain
(config, generator, discriminator, kp_detector, checkpoint, log_dir, dataset, device_ids)
train.py:78
↓ 1 callersFunctiontransfer
(config, generator, kp_detector, checkpoint, log_dir, dataset)
transfer.py:82
↓ 1 callersMethodvisualize_rec
(self, inp, out)
logger.py:39
↓ 1 callersMethodvisualize_transfer
(self, driving_video, source_image, out)
logger.py:128
Method__call__
(self, clip)
augmentation.py:96
Method__call__
(self, clip)
augmentation.py:119
Method__call__
Args: img (PIL.Image or numpy.ndarray): List of images to be cropped in format (h, w, c) in numpy.ndarray Returns:
augmentation.py:149
Method__call__
Args: img (PIL.Image or numpy.ndarray): List of images to be cropped in format (h, w, c) in numpy.ndarray Returns:
augmentation.py:197
Method__call__
Args: clip (list): list of PIL.Image Returns: list PIL.Image : list of transformed PIL.Image
augmentation.py:261
Method__call__
Args: img (PIL.Image or numpy.ndarray): List of images for selection in format (h, w, c) in numpy.ndarray Returns:
augmentation.py:328
Method__call__
(self, video)
augmentation.py:348
Method__call__
(self, driving)
augmentation.py:358
Method__call__
(self, clip)
augmentation.py:386
Method__enter__
(self)
logger.py:68
Method__exit__
(self, exc_type, exc_val, exc_tb)
logger.py:71
Method__getitem__
(self, idx)
frames_dataset.py:79
Method__getitem__
(self, idx)
frames_dataset.py:124
Method__getitem__
(self, idx)
prediction.py:28
Method__getstate__
(self)
sync_batchnorm/comm.py:78
Method__init__
(self, kp_extractor, generator, discriminator, train_params)
train.py:61
Method__init__
(self, time_flip=False, horizontal_flip=False)
augmentation.py:92
Method__init__
(self, ratio=(3. / 4., 4. / 3.), interpolation='nearest')
augmentation.py:115
Method__init__
(self, size)
augmentation.py:143
Method__init__
(self, degrees)
augmentation.py:184
Method__init__
(self, brightness=0, contrast=0, saturation=0, hue=0)
augmentation.py:230
Method__init__
(self, consequent=False, number_of_frames=2)
augmentation.py:324
Method__init__
(self, resize_param=None, rotation_param=None, flip_param=None, crop_param=None, jitter_param=None)
augmentation.py:364
Method__init__
(self, log_dir, log_file_name='log.txt', log_freq_iter=100, cpk_freq_epoch=100, zfill_num=8,
logger.py:12
Method__init__
(self, kp_size=2, draw_border=False, colormap='gist_rainbow')
logger.py:92
Method__init__
(self, root_dir, augmentation_params, image_shape=(64, 64, 3), is_train=True, random_seed=0,
frames_dataset.py:46
Method__init__
(self, initial_dataset, number_of_pairs, seed=0)
frames_dataset.py:96
Method__init__
(self, keypoints_array, num_frames)
prediction.py:21
Method__init__
(self, block_expansion, num_blocks, max_features, mask_embedding_params, num_kp, num_channels
modules/dense_motion_module.py:14
Method__init__
(self, block_expansion, num_kp, num_channels, max_features, num_blocks, temperature, kp_varia
modules/keypoint_detector.py:86
Method__init__
(self, num_channels=3, num_kp=10, kp_variance=0.01, scale_factor=1, block_expansion=64, num_b
modules/discriminator.py:39
Method__init__
(self, num_kp=10, kp_variance=0.01, num_features=1024, num_layers=1, dropout=0.5)
modules/prediction_module.py:10
Method__init__
(self, in_features, kernel_size, padding)
modules/util.py:50
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1)
modules/util.py:76
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1)
modules/util.py:96
Method__init__
(self, in_features, out_features, groups=None, kernel_size=3, padding=1)
modules/util.py:116
next →1–100 of 140, ranked by callers