MCPcopy Create free account

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

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

Methodforward
(self, x, img)
lib/model_zoo/migan_inference.py:192
Methodforward
(self, img)
lib/model_zoo/migan_inference.py:235
Methodforward
(self, x, enc_feat)
lib/model_zoo/migan_inference.py:270
Methodforward
(self, x, enc_feat, img)
lib/model_zoo/migan_inference.py:303
Methodforward
(self, x, enc_feats)
lib/model_zoo/migan_inference.py:347
Methodforward
Args: x: 4 channel rgb+mask
lib/model_zoo/migan_inference.py:362
Methodforward
(self, x)
lib/model_zoo/migan.py:37
Methodforward
(self, x, gain=1, noise_mode='none')
lib/model_zoo/migan.py:105
Methodforward
(self, x, gain=1, noise_mode='none')
lib/model_zoo/migan.py:202
Methodforward
(self, x, img)
lib/model_zoo/migan.py:246
Methodforward
(self, img)
lib/model_zoo/migan.py:320
Methodforward
(self, x, enc_feat, noise_mode='random')
lib/model_zoo/migan.py:376
Methodforward
(self, x, enc_feat, img, noise_mode='random')
lib/model_zoo/migan.py:445
Methodforward
(self, x, enc_feats, noise_mode='random')
lib/model_zoo/migan.py:516
Methodforward
Args: x: 4 channel rgb+mask
lib/model_zoo/migan.py:546
Methodforward
(self, x, img)
lib/model_zoo/migan.py:606
Methodforward
(self, x)
lib/model_zoo/migan.py:630
Methodforward
(self, x)
lib/model_zoo/migan.py:684
Methodforward
(self, img)
lib/model_zoo/migan.py:758
Methodforward
(self, x, img)
lib/model_zoo/comodgan.py:35
Methodforward
(self, x, img=None, cmap=None)
lib/model_zoo/comodgan.py:96
Methodforward
(self, img, c=None)
lib/model_zoo/comodgan.py:190
Methodforward
(self, x, x0, ws, fused_modconv=None, noise_mode='random')
lib/model_zoo/comodgan.py:237
Methodforward
(self, x, x0, img, ws, w0, fused_modconv=None, noise_mode='random')
lib/model_zoo/comodgan.py:305
Methodforward
(self, x, feats, ws, noise_mode='random', return_intermediate_outs=False)
lib/model_zoo/comodgan.py:398
Methodforward
Args: x: 4 channel rgb+mask
lib/model_zoo/comodgan.py:438
Methodforward
(self, x)
lib/model_zoo/stylegan.py:50
Methodforward
(self, x)
lib/model_zoo/stylegan.py:85
Methodforward
(self, x, gain=1)
lib/model_zoo/stylegan.py:229
Methodforward
(self, x, w, fused_modconv=True, gain=1, noise_mode='random')
lib/model_zoo/stylegan.py:280
Methodforward
(self, x, w, fused_modconv=True)
lib/model_zoo/stylegan.py:332
Methodforward
(self, z, c, truncation_psi=1, truncation_cutoff=None, skip_w_avg_update=False)
lib/model_zoo/stylegan.py:403
Methodforward
(self, x, img, ws, fused_modconv=None, noise_mode='random')
lib/model_zoo/stylegan.py:495
Methodforward
(self, ws, noise_mode='random')
lib/model_zoo/stylegan.py:576
Methodforward
(self, z, c, truncation_psi=1, truncation_cutoff=None, **synthesis_kwargs)
lib/model_zoo/stylegan.py:615
Methodforward
(self, x, img)
lib/model_zoo/stylegan.py:672
Methodforward
(self, x)
lib/model_zoo/stylegan.py:707
Methodforward
(self, x, img=None, cmap=None)
lib/model_zoo/stylegan.py:760
Methodforward
(self, img, c, **kwargs)
lib/model_zoo/stylegan.py:846
Methodforward
Apply gaussian filter to input. Arguments: input (torch.Tensor): Input to apply gaussian filter on. Returns:
scripts/create_onnx_pipeline.py:106
Methodforward
image: (1, 3, H, W) torch.uint8 Tensor mask: (1, 1, H, W) torch.uint8 Tensor
scripts/create_onnx_pipeline.py:252
Functionfreeze
(net)
lib/model_zoo/common/utils.py:132
Methodget_all_torch
(self)
lib/evaluator/stylegan_metrics/metric_utils.py:129
Functionget_dtype_and_ctype
Given a type name string (or an object having a __name__ attribute), return matching Numpy and ctypes types that have the same size in bytes.
dnnlib/util.py:187
Functionget_module_dir_by_obj_name
Get the directory path of the module containing the given object name.
dnnlib/util.py:292
Functionget_plugin
(module_name, sources, **build_kwargs)
torch_utils/custom_ops.py:46
Functionget_top_level_function_name
Return the fully-qualified name of a top-level function.
dnnlib/util.py:303
Functiongetinstance
(*args, **kwargs)
lib/cfg_holder.py:7
Functiongetinstance
(*args, **kwargs)
lib/evaluator/eva_base.py:14
Functiongetinstance
(*args, **kwargs)
lib/model_zoo/common/get_model.py:34
Functiongetinstance
(*args, **kwargs)
lib/model_zoo/common/utils.py:16
Functiongetinstance
(*args, **kwargs)
lib/data_factory/common/ds_base.py:54
Functiongetinstance
(*args, **kwargs)
lib/data_factory/common/ds_formatter.py:4
Functiongetinstance
(*args, **kwargs)
lib/data_factory/common/ds_loader.py:10
Functiongrid_sample
(input, grid)
torch_utils/ops/grid_sample_gradfix.py:27
Functionimport_hook
r"""Register an import hook that is called whenever a persistent object is being unpickled. A typical use case is to patch the pickled source
torch_utils/persistence.py:147
Methodinit_args
(self)
torch_utils/persistence.py:111
Methodinit_kwargs
(self)
torch_utils/persistence.py:115
Methodinit_load_info
(self, cfg)
lib/data_factory/ds_ffhq.py:22
Methodinit_load_info
(self, cfg)
lib/data_factory/ds_ffhq.py:234
Methodinit_load_info
(self, cfg)
lib/data_factory/ds_ffhq.py:275
Methodinit_load_info
(self, cfg)
lib/data_factory/ds_places2.py:17
Functioninit_module
Args: module: [nn.module] list or nn.module a list of module to be initialized.
lib/model_zoo/common/utils.py:160
Functioninit_multiprocessing
r"""Initializes `torch_utils.training_stats` for collecting statistics across multiple processes. This function must be called after `tor
torch_utils/training_stats.py:34
Functioninner
(self, element)
lib/data_factory/common/ds_loader.py:61
Functionis_pickleable
(obj: Any)
dnnlib/util.py:210
Methodisbetter
(self, old, new)
lib/evaluator/eva_base.py:229
Methodisbetter
(self, old, new)
lib/evaluator/eva_base.py:267
Functionleaky_relu_jit
(inputs: List[Any], outputs: List[Any])
scripts/calculate_flops.py:28
Functionlist_dir_recursively_with_ignore
List all files recursively in a given directory while ignoring given file and directory names. Returns list of tuples containing both absolute and
dnnlib/util.py:315
Functionlist_valid_metrics
()
lib/evaluator/stylegan_metrics/metric_main.py:32
Functionmean_jit
(inputs: List[Any], outputs: List[Any])
scripts/calculate_flops.py:52
Functionmul_jit
(inputs: List[Any], outputs: List[Any])
scripts/calculate_flops.py:34
Functionno_weight_gradients
()
torch_utils/ops/conv2d_gradfix.py:26
Methodone_line_summary
(self)
lib/evaluator/eva_psnr.py:79
Methodone_line_summary
(self)
lib/evaluator/eva_base.py:232
Methodone_line_summary
(self)
lib/evaluator/eva_base.py:277
Methodone_line_summary
(self)
lib/evaluator/eva_lpips.py:76
Methodone_line_summary
(self)
lib/evaluator/eva_ssim.py:120
Methodone_line_summary
(self)
lib/evaluator/eva_null.py:18
Functionopen_url
Download the given URL and return a binary-mode file object to access the data.
dnnlib/util.py:382
Functionparams_and_buffers
(module)
torch_utils/misc.py:145
Functionpointwise_jit
(inputs: List[Any], outputs: List[Any])
scripts/calculate_flops.py:70
Functionpost_hook
(mod, _inputs, outputs)
torch_utils/misc.py:202
Functionpre_hook
(_mod, _inputs)
torch_utils/misc.py:200
Functionpre_loader_checkings
(ltype)
lib/data_factory/common/ds_loader.py:56
Functionprint_module_summary
(module, inputs, max_nesting=3, skip_redundant=True)
torch_utils/misc.py:192
Functionprofile
(module, input, nb_iters=1000)
lib/model_zoo/stylegan.py:875
Functionprofiled_function
(fn)
torch_utils/misc.py:98
Methodrecurse
(prefix, tf_net)
lib/experiments/comodgan_default.py:745
Functionregister
(name)
lib/evaluator/eva_base.py:65
Functionregister
(name, version='x')
lib/model_zoo/common/get_model.py:99
Functionregister
(name)
lib/model_zoo/common/utils.py:89
Functionregister
()
lib/data_factory/common/ds_base.py:86
Functionregister
()
lib/data_factory/common/ds_formatter.py:26
Functionregister
()
lib/data_factory/common/ds_loader.py:49
Functionregister_metric
(fn)
lib/evaluator/stylegan_metrics/metric_main.py:24
Functionreport0
r"""Broadcasts the given set of scalars by the first process (`rank = 0`), but ignores any scalars provided by the other processes. See `repor
torch_utils/training_stats.py:103
Functionreport_metric
(result_dict, run_dir=None, snapshot_pkl=None)
lib/evaluator/stylegan_metrics/metric_main.py:78
Functionrfft_flop_jit
Count flops for the rfft/rfftn operator. Source: https://github.com/raoyongming/DynamicViT/blob/master/calc_flops.py
scripts/calculate_flops.py:76
← previousnext →401–500 of 526, ranked by callers