MCPcopy Create free account

hub / github.com/Picsart-AI-Research/MI-GAN / functions

Functions526 in github.com/Picsart-AI-Research/MI-GAN

↓ 41 callersMethodmean
r"""Returns the mean of the scalars that were accumulated for the given statistic between the last two calls to `update()`, or NaN if
torch_utils/training_stats.py:188
↓ 37 callersMethodappend
(self, x)
lib/evaluator/stylegan_metrics/metric_utils.py:78
↓ 36 callersFunctionprint_log
(*console_info)
lib/log_service.py:4
↓ 21 callersMethodload
(pkl_file)
lib/evaluator/stylegan_metrics/metric_utils.py:144
↓ 20 callersMethodupdate
(self, cur_items)
lib/evaluator/stylegan_metrics/metric_utils.py:169
↓ 18 callersFunctionhighlight_print
(info)
lib/experiments/comodgan_default.py:26
↓ 13 callersMethodsave
(self, path)
lib/evaluator/eva_base.py:281
↓ 12 callersMethod__init__
( self, in_features, out_features, bias=True, bias_init=0, act
lib/model_zoo/migan.py:15
↓ 12 callersMethod__init__
(self, in_features, out_features, bias=True,
lib/model_zoo/stylegan.py:65
↓ 11 callersFunctionhighlight_print
(info)
lib/experiments/migan_default.py:27
↓ 10 callersFunctionget_source_w
(module)
scripts/export_inference_model.py:18
↓ 9 callersMethodbackward
(ctx, dy)
torch_utils/ops/bias_act.py:161
↓ 9 callersMethodregister
(self, name, unitf,)
lib/model_zoo/common/utils.py:56
↓ 8 callersMethod__init__
( self, resolution=256, ic_n=4, ch_base=32768, ch_max=512, act
lib/model_zoo/migan_inference.py:204
↓ 7 callersFunctioncfg_solve
(cfg, cfg_root)
lib/cfg_helper.py:85
↓ 7 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:93
↓ 6 callersMethod__init__
(self, resolution=256, ic_n=4, oc_n=1024,
lib/model_zoo/comodgan.py:116
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
torch_utils/ops/conv2d_resample.py:29
↓ 6 callersFunction_parse_padding
(padding)
torch_utils/ops/upfirdn2d.py:46
↓ 6 callersFunction_parse_scaling
(scaling)
torch_utils/ops/upfirdn2d.py:37
↓ 6 callersMethodsub
(self, tag=None, num_items=None, flush_interval=1000, rel_lo=0, rel_hi=1)
lib/evaluator/stylegan_metrics/metric_utils.py:184
↓ 6 callersMethodupdate
r"""Copies current values of the internal counters to the user-visible state and resets them for the next round. If `keep_previous=Tr
torch_utils/training_stats.py:147
↓ 6 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:78
↓ 6 callersMethodzipzap_arrange
Order the data so it range like this: input [[0, 2, 4, 6], [1, 3, 5, 7]] -> output [0, 1, 2, 3, 4, 5, ...]
lib/evaluator/eva_base.py:184
↓ 5 callersFunctionRandomMask
(s, hole_range=[0,1])
lib/data_factory/ds_ffhq.py:204
↓ 5 callersFunction_get_filter_size
(f)
torch_utils/ops/upfirdn2d.py:57
↓ 5 callersMethodoutput
(self, img, fname, drange, grid_size)
lib/experiments/comodgan_default.py:119
↓ 5 callersMethodoutput
(self, img, fname, drange, grid_size)
lib/experiments/migan_default.py:52
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
torch_utils/ops/conv2d_gradfix.py:68
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
torch_utils/ops/conv2d_gradfix.py:58
↓ 4 callersFunctioncount_parameters
(model)
scripts/calculate_flops.py:108
↓ 4 callersFunctionhas_same_layout
torch_utils/ops/bias_act.cpp:16
↓ 4 callersFunctiontemp
(x, xp)
lib/cfg_helper.py:443
↓ 3 callersMethod_get_delta
r"""Returns the raw moments that were accumulated for the given statistic between the last two calls to `update()`, or zero if no scal
torch_utils/training_stats.py:170
↓ 3 callersFunction_unbroadcast
(x, shape)
torch_utils/ops/fma.py:49
↓ 3 callersMethodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
torch_utils/training_stats.py:212
↓ 3 callersFunctioncheck_and_create_dir
(path)
lib/experiments/comodgan_default.py:34
↓ 3 callersFunctioncheck_and_create_dir
(path)
lib/experiments/migan_default.py:35
↓ 3 callersMethodcheck_ddp_consistency
(self, module)
lib/experiments/comodgan_default.py:182
↓ 3 callersMethodcheck_ddp_consistency
(self, module)
lib/experiments/migan_default.py:174
↓ 3 callersFunctioncreate_window
(window_size, channel)
lib/evaluator/eva_ssim.py:15
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:263
↓ 3 callersFunctionnamed_params_and_buffers
(module)
torch_utils/misc.py:149
↓ 3 callersFunctionrecurse
(obj)
torch_utils/persistence.py:236
↓ 3 callersFunctionresize
(image, max_size, interpolation=Image.BICUBIC)
scripts/demo.py:48
↓ 3 callersMethodrun_D
(self, img, sync)
lib/experiments/loss.py:139
↓ 3 callersMethodrun_D
(self, img, c, sync)
lib/experiments/comodgan_loss.py:54
↓ 3 callersMethodrun_G
(self, x, z, c, sync)
lib/experiments/comodgan_loss.py:38
↓ 3 callersFunctionstr2value
(v)
lib/model_zoo/common/utils.py:23
↓ 3 callersMethodsync
Args: data: any, the data needs to be broadcasted
lib/evaluator/eva_base.py:87
↓ 3 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
torch_utils/ops/upfirdn2d.py:120
↓ 2 callersFunctionMultiFill
(max_tries, max_size)
lib/data_factory/ds_ffhq.py:215
↓ 2 callersFunctionMultiFill
(max_tries, max_size)
scripts/generate_masks.py:83
↓ 2 callersFunctionMultiFill
(max_tries, max_size)
scripts/evaluate_fid_lpips.py:111
↓ 2 callersMethod__init__
(self)
lib/data_factory/ds_ffhq.py:317
↓ 2 callersFunction_get_weight_shape
(w)
torch_utils/ops/conv2d_resample.py:21
↓ 2 callersFunction_should_use_custom_op
(input)
torch_utils/ops/conv2d_gradfix.py:47
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average = True)
lib/evaluator/eva_ssim.py:21
↓ 2 callersFunction_upfirdn2d_cuda
Fast CUDA implementation of `upfirdn2d()` using custom ops.
torch_utils/ops/upfirdn2d.py:214
↓ 2 callersMethodaccumulate_gradients
(self, phase, real_img, real_c, gen_z, gen_c, sync, gain)
lib/experiments/loss.py:20
↓ 2 callersMethodappend_torch
(self, x, num_gpus=1, rank=0)
lib/evaluator/stylegan_metrics/metric_utils.py:112
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
torch_utils/ops/conv2d_gradfix.py:95
↓ 2 callersFunctioncfg_solvef
(cmd, root)
lib/cfg_helper.py:21
↓ 2 callersMethodcheck_ddp_consistency
(self, module)
lib/experiments/comodgan_default.py:802
↓ 2 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:100
↓ 2 callersFunctionget_activations
(model, imgs)
scripts/evaluate_fid_lpips.py:168
↓ 2 callersFunctionget_args
()
scripts/evaluate_fid_lpips.py:32
↓ 2 callersFunctionget_experiment
(stagename)
lib/experiments/get_experiment.py:4
↓ 2 callersFunctionget_feature_detector
(url, device=torch.device('cpu'), num_gpus=1, rank=0, verbose=False)
lib/evaluator/stylegan_metrics/metric_utils.py:40
↓ 2 callersMethodget_mean_cov
(self)
lib/evaluator/stylegan_metrics/metric_utils.py:132
↓ 2 callersFunctionget_module_from_obj_name
Searches for the underlying module behind the name to some python object. Returns the module and the object name (original name with module part r
dnnlib/util.py:222
↓ 2 callersMethodis_full
(self)
lib/evaluator/stylegan_metrics/metric_utils.py:75
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
torch_utils/persistence.py:134
↓ 2 callersFunctionis_valid_metric
(metric)
lib/evaluator/stylegan_metrics/metric_main.py:29
↓ 2 callersFunctionlegacy_load_G
(G, G_tfmodel)
lib/experiments/comodgan_default.py:622
↓ 2 callersFunctionmodulated_conv2d
Args: x : Input tensor of shape [batch_size, in_channels, in_height, in_width]. weight : Weight tensor of
lib/model_zoo/stylegan.py:102
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
torch_utils/training_stats.py:141
↓ 2 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
torch_utils/misc.py:49
↓ 2 callersFunctionnormalize_2nd_moment
(x, dim=1, eps=1e-8)
lib/model_zoo/stylegan.py:351
↓ 2 callersMethodregister_stage
(self, stage)
lib/utils.py:26
↓ 2 callersMethodrun_G
(self, x, sync)
lib/experiments/loss.py:123
↓ 2 callersMethodset_num_features
(self, num_features)
lib/evaluator/stylegan_metrics/metric_utils.py:66
↓ 2 callersFunctionsetup_filter
(f, device=torch.device('cpu'), normalize=True, flip_filter=False, gain=1, separable=None)
lib/model_zoo/migan_inference.py:31
↓ 2 callersMethodzipfile_close
(self)
lib/data_factory/ds_ffhq.py:332
↓ 1 callersFunctionFill
(max_size)
lib/data_factory/ds_ffhq.py:209
↓ 1 callersFunctionFill
(max_size)
scripts/generate_masks.py:77
↓ 1 callersFunctionFill
(max_size)
scripts/evaluate_fid_lpips.py:105
↓ 1 callersFunctionRandomBrush
( max_tries, s, min_num_vertex=4, max_num_vertex=18, mean_angle=2*math.pi / 5, a
lib/data_factory/ds_ffhq.py:148
↓ 1 callersFunctionRandomBrush
( max_tries, s, min_num_vertex=4, max_num_vertex=18, mean_angle=2*math.pi / 5, angle_r
scripts/generate_masks.py:16
↓ 1 callersFunctionRandomBrush
( max_tries, s, min_num_vertex=4, max_num_vertex=18, mean_angle=2*math.pi / 5, angle_r
scripts/evaluate_fid_lpips.py:44
↓ 1 callersFunctionRandomMask
(s, hole_range=[0,1])
scripts/generate_masks.py:72
↓ 1 callersFunctionRandomMask
(s, hole_range=[0,1])
scripts/evaluate_fid_lpips.py:100
↓ 1 callersMethod__init__
(self)
lib/experiments/comodgan_default.py:727
↓ 1 callersMethod__init__
(self, window_size = 11, size_average = True)
lib/evaluator/eva_ssim.py:44
↓ 1 callersMethod__init__
(self, model_path, resolution, padding = 128, device='cpu')
scripts/create_onnx_pipeline.py:120
↓ 1 callersMethod__reduce__
(self)
torch_utils/persistence.py:118
↓ 1 callersFunction_bias_act_cuda
Fast CUDA implementation of `bias_act()` using custom ops.
torch_utils/ops/bias_act.py:129
↓ 1 callersFunction_bias_act_ref
Slow reference implementation of `bias_act()` using standard TensorFlow ops.
torch_utils/ops/bias_act.py:94
↓ 1 callersFunction_check_pickleable
r"""Check that the given object is pickleable, raising an exception if it is not. This function is expected to be considerably more efficient
torch_utils/persistence.py:231
↓ 1 callersFunction_find_compiler_bindir
()
torch_utils/custom_ops.py:28
next →1–100 of 526, ranked by callers