MCPcopy Create free account

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

Functions140 in github.com/AliaksandrSiarohin/monkey-net

Method__init__
(self, block_expansion, in_features, out_features, num_blocks=3, max_features=256, temporal=False,
modules/util.py:160
Method__init__
(self, block_expansion, in_features, out_features, num_blocks=3, max_features=256, temporal=False, )
modules/util.py:197
Method__init__
(self, num_kp, kp_variance, num_channels, use_deformed_source_image=False, use_difference=False,
modules/movement_embedding.py:13
Method__init__
(self, num_channels, num_kp, kp_variance, block_expansion, max_features, num_blocks, num_refinement_blocks,
modules/generator.py:16
Method__init__
(self)
sync_batchnorm/comm.py:21
Method__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True)
sync_batchnorm/batchnorm.py:39
Method__len__
(self)
frames_dataset.py:76
Method__len__
(self)
frames_dataset.py:121
Method__len__
(self)
prediction.py:25
Method__setstate__
(self, state)
sync_batchnorm/comm.py:81
Method_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:247
Method_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:311
Method_data_parallel_master
Reduce the sum and square-sum, compute the statistics, and broadcast it.
sync_batchnorm/batchnorm.py:90
MethodassertTensorClose
(self, a, b, atol=1e-3, rtol=1e-3)
sync_batchnorm/unittest.py:24
Functioncompute_image_gradient
(image, padding=0)
modules/util.py:10
Methodforward
(self, x)
train.py:36
Methodforward
(self, x, kp_joined, generated)
train.py:68
Methodforward
(self, source_image, kp_driving, kp_source)
modules/dense_motion_module.py:42
Methodforward
(self, appearance_frame, kp_video, kp_appearance)
modules/dense_motion_module.py:80
Methodforward
(self, x)
modules/keypoint_detector.py:97
Methodforward
(self, x)
modules/discriminator.py:24
Methodforward
(self, x, kp_driving, kp_source)
modules/discriminator.py:64
Methodforward
(self, kp_batch)
modules/prediction_module.py:26
Methodforward
(self, x)
modules/util.py:59
Methodforward
(self, x)
modules/util.py:83
Methodforward
(self, x)
modules/util.py:103
Methodforward
(self, x)
modules/util.py:122
Methodforward
(self, x)
modules/util.py:147
Methodforward
(self, x)
modules/util.py:181
Methodforward
(self, x)
modules/util.py:202
Methodforward
(self, source_image, kp_driving, kp_source)
modules/movement_embedding.py:42
Methodforward
(self, source_image, kp_driving, kp_source)
modules/generator.py:60
Methodforward
(self, input)
sync_batchnorm/batchnorm.py:48
Functionmatrix_det
(batch_of_matrix)
modules/util.py:227
Functionmatrix_trace
(batch_of_matrix)
modules/util.py:237
Functionnew_replicate
(module, device_ids)
sync_batchnorm/replicate.py:89
Methodnr_slaves
(self)
sync_batchnorm/comm.py:136
Functiononmouse
(event, x, y, flags, param)
sup-mat/MGif/bg_removal_tool.py:87
Functionpatch_replication_callback
Monkey-patch an existing `DataParallel` object. Add the replication callback. Useful when you have customized `DataParallel` implementation.
sync_batchnorm/replicate.py:70
Methodreplicate
(self, module, device_ids)
sync_batchnorm/replicate.py:64
← previous101–140 of 140, ranked by callers