MCPcopy Create free account

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

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

Method__init__
(self)
lib/data_factory/ds_places2.py:113
Method__init__
(self)
lib/data_factory/ds_places2.py:136
Method__init__
(self, resolution=512, hole_range=[0, 1])
lib/data_factory/ds_places2.py:163
Method__init__
(self, random_flip=True, resolution=512, hole_range=[0, 1])
lib/data_factory/ds_places2.py:192
Method__init__
(self, cfg, loader=None, formatter=None)
lib/data_factory/common/ds_base.py:12
Method__init__
(self)
lib/data_factory/common/ds_base.py:63
Method__init__
(self)
lib/data_factory/common/ds_base.py:101
Method__init__
(self)
lib/data_factory/common/ds_formatter.py:13
Method__init__
Args: extend: bool, True, even the number of samples between processes by extending it.
lib/data_factory/common/ds_sampler.py:20
Method__init__
(self)
lib/data_factory/common/ds_loader.py:19
Method__init__
(self, loaders)
lib/data_factory/common/ds_loader.py:37
Method__init__
(self, channels, kernel_size, sigma, dim=2)
scripts/create_onnx_pipeline.py:62
Method__init__
( self, real_dir: Path, mask_dir: Optional[Path] = None, resolution: int = Non
scripts/evaluate_fid_lpips.py:125
Method__init__
(self, file_name: str = None, file_mode: str = "w", should_flush: bool = True)
dnnlib/util.py:59
Method__iter__
(self)
torch_utils/misc.py:123
Method__iter__
(self)
lib/data_factory/common/ds_sampler.py:58
Method__len__
(self)
lib/data_factory/common/ds_base.py:35
Method__len__
(self)
lib/data_factory/common/ds_sampler.py:70
Method__len__
(self)
scripts/evaluate_fid_lpips.py:164
Method__repr__
(self)
lib/model_zoo/migan.py:50
Method__repr__
(self)
lib/model_zoo/migan.py:148
Method__repr__
(self)
lib/model_zoo/stylegan.py:98
Method__repr__
(self)
lib/model_zoo/stylegan.py:243
Method__repr__
(self,)
lib/model_zoo/common/utils.py:124
Method__setattr__
(self, name: str, value: Any)
dnnlib/util.py:49
Function_reconstruct_persistent_obj
r"""Hook that is called internally by the `pickle` module to unpickle a persistent object.
torch_utils/persistence.py:179
Methodaccumulate_gradients
( self, phase, real_img, mask, real_img_erased, sync,
lib/experiments/loss.py:144
Methodaccumulate_gradients
(self, phase, real_img, real_c, gen_z, gen_c, sync, gain)
lib/experiments/comodgan_loss.py:18
Methodaccumulate_gradients
(self, phase, real_img, mask, real_img_erased, real_c, gen_z, gen_c, sync, gain)
lib/experiments/comodgan_loss.py:61
Methodadd_batch
(self, pred, gt, fn=None, **dummy)
lib/evaluator/eva_psnr.py:23
Methodadd_batch
(self, **args)
lib/evaluator/eva_base.py:218
Methodadd_batch
(self, *args, **kwargs)
lib/evaluator/eva_base.py:251
Methodadd_batch
(self, pred, gt, fn=None, **dummy)
lib/evaluator/eva_lpips.py:25
Methodadd_batch
(self, pred, gt, fn=None, **dummy)
lib/evaluator/eva_ssim.py:89
Methodadd_batch
(self, **dummy)
lib/evaluator/eva_null.py:11
Methodadd_code
A new main code is reached and its name is added.
lib/cfg_holder.py:28
Functionadd_jit
(inputs: List[Any], outputs: List[Any])
scripts/calculate_flops.py:40
Functionask_yes_no
Ask the user the question until the user inputs a valid answer.
dnnlib/util.py:153
Functionassert_shape
(tensor, ref_shape)
torch_utils/misc.py:80
Methodbackward
(ctx, grad_output)
torch_utils/ops/grid_sample_gradfix.py:54
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
torch_utils/ops/grid_sample_gradfix.py:70
Methodbackward
(ctx, d_dx)
torch_utils/ops/bias_act.py:189
Methodbackward
(ctx, dy)
torch_utils/ops/upfirdn2d.py:246
Methodbackward
(ctx, dout)
torch_utils/ops/fma.py:29
Methodbackward
(ctx, grad_output)
torch_utils/ops/conv2d_gradfix.py:119
Methodbackward
(ctx, grad2_grad_weight)
torch_utils/ops/conv2d_gradfix.py:151
Functionbias_act
r"""Fused bias and activation function. Adds bias `b` to activation tensor `x`, evaluates activation function `act`, and scales the result by
torch_utils/ops/bias_act.py:55
Functionbias_act
torch_utils/ops/bias_act.cpp:32
Functioncalc_metric
(metric, **kwargs)
lib/evaluator/stylegan_metrics/metric_main.py:37
Functioncfg_to_debug
(cfg)
lib/cfg_helper.py:241
Functioncheck_ddp_consistency
(module, ignore_regex=None)
torch_utils/misc.py:178
Methodclear_data
(self)
lib/evaluator/eva_psnr.py:83
Methodclear_data
(self)
lib/evaluator/eva_base.py:242
Methodclear_data
(self)
lib/evaluator/eva_base.py:288
Methodclear_data
(self)
lib/evaluator/eva_lpips.py:80
Methodclear_data
(self)
lib/evaluator/eva_ssim.py:124
Methodclear_data
(self)
lib/evaluator/eva_null.py:21
Methodcollect_tf_params
(tf_net)
lib/experiments/comodgan_default.py:742
Methodcompute
(self)
lib/evaluator/eva_psnr.py:73
Methodcompute
(self)
lib/evaluator/eva_base.py:224
Methodcompute
(self)
lib/evaluator/eva_base.py:260
Methodcompute
(self)
lib/evaluator/eva_lpips.py:70
Methodcompute
(self)
lib/evaluator/eva_ssim.py:114
Methodcompute
(self)
lib/evaluator/eva_null.py:15
Functioncompute_feature_stats_for_dataset
(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, batch_size=64, max_items=None, **stats_kwargs)
lib/evaluator/stylegan_metrics/metric_utils.py:197
Functioncompute_feature_stats_for_inpainting
(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, **stats_kwargs)
lib/evaluator/stylegan_metrics/metric_utils.py:256
Functioncompute_r1_penalty_from_outputs
Computes R1 grad penalty based on the existing d_outputs to save memory This is get from INR GAN
lib/model_zoo/stylegan.py:625
Functionconstant
(value, shape=None, dtype=None, device=None, memory_format=None)
torch_utils/misc.py:22
Functionconstruct_class_by_name
Finds the python class with the given name and constructs it with the given arguments.
dnnlib/util.py:287
Functionconv2d
(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
torch_utils/ops/conv2d_gradfix.py:35
Functionconv2d_resample
r"""2D convolution with optional up/downsampling. Padding is performed only once at the beginning, not between the operations. Args:
torch_utils/ops/conv2d_resample.py:59
Functionconv_transpose2d
(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dilation=1)
torch_utils/ops/conv2d_gradfix.py:40
Methodconvert_tf
(tf_G)
lib/experiments/comodgan_default.py:753
Functioncopy_files_and_create_dirs
Takes in a list of tuples of (src, dst) paths and copies files. Will create all necessary directories.
dnnlib/util.py:348
Functioncopy_params_and_buffers
(src_module, dst_module, require_all=False)
torch_utils/misc.py:153
Functionddp_sync
(module, sync)
torch_utils/misc.py:167
Functiondecorator
(*args, **kwargs)
torch_utils/misc.py:99
Functiondownsample2d
r"""Downsample a batch of 2D images using the given 2D FIR filter. By default, the result is padded so that its shape is a fraction of the input.
torch_utils/ops/upfirdn2d.py:347
Functiondp
(v)
lib/model_zoo/stylegan.py:863
Functionfid10k_full_inpainting
(opts)
lib/evaluator/stylegan_metrics/metric_main.py:99
Functionfid36500_full_inpainting
(opts)
lib/evaluator/stylegan_metrics/metric_main.py:104
Functionfid50k_full_inpainting
(opts)
lib/evaluator/stylegan_metrics/metric_main.py:94
Functionfid5k_full_inpainting
(opts)
lib/evaluator/stylegan_metrics/metric_main.py:115
Functionfilter2d
r"""Filter a batch of 2D images using the given 2D FIR filter. By default, the result is padded so that its shape matches the input. User-spe
torch_utils/ops/upfirdn2d.py:272
Methodfind_class
(self, module, name)
lib/experiments/comodgan_default.py:737
Functionfixseed
(seed)
lib/model_zoo/stylegan.py:869
Functionfma
(a, b, c)
torch_utils/ops/fma.py:15
Functionformat_time
Convert the seconds to human readable string with days, hours, minutes and seconds.
dnnlib/util.py:139
Methodforward
(ctx, input, grid)
torch_utils/ops/grid_sample_gradfix.py:46
Methodforward
(ctx, grad_output, input, grid)
torch_utils/ops/grid_sample_gradfix.py:63
Methodforward
(ctx, x, b)
torch_utils/ops/bias_act.py:147
Methodforward
(ctx, dy, x, b, y)
torch_utils/ops/bias_act.py:180
Methodforward
(ctx, x, f)
torch_utils/ops/upfirdn2d.py:230
Methodforward
(ctx, a, b, c)
torch_utils/ops/fma.py:22
Methodforward
(ctx, input, weight, bias)
torch_utils/ops/conv2d_gradfix.py:109
Methodforward
(ctx, grad_output, input)
torch_utils/ops/conv2d_gradfix.py:142
Methodforward
(self, img1, img2)
lib/evaluator/eva_ssim.py:51
Methodforward
(self, x)
lib/model_zoo/migan_inference.py:74
Methodforward
(self, x)
lib/model_zoo/migan_inference.py:98
Methodforward
(self, x)
lib/model_zoo/migan_inference.py:154
← previousnext →301–400 of 526, ranked by callers