MCPcopy Create free account

hub / github.com/Picsart-AI-Research/MI-GAN / types & classes

Types & classes121 in github.com/Picsart-AI-Research/MI-GAN

↓ 16 callersClassConv2d
lib/model_zoo/migan.py:54
↓ 15 callersClassget_model
lib/model_zoo/common/get_model.py:57
↓ 9 callersClassSeparableConv2d
lib/model_zoo/migan.py:152
↓ 9 callersClassdense
lib/model_zoo/stylegan.py:64
↓ 7 callersClassconv2d_layer
lib/model_zoo/stylegan.py:198
↓ 6 callersClassSeparableConv2d
lib/model_zoo/migan_inference.py:106
↓ 6 callersClassget_dataset
lib/data_factory/common/ds_base.py:62
↓ 5 callersClassDistributedSampler
This is a new DistributedSampler that use more sophisticate way for random sync. Arguments: dataset: Dataset used for sampli
lib/data_factory/common/ds_sampler.py:7
↓ 5 callersClasscollate
Modified from torch.utils.data._utils.collate It handle list different from the default. List collate just by append e
lib/data_factory/common/ds_base.py:95
↓ 5 callersClasslrelu_agc
The lrelu layer with alpha, gain and clamp
lib/model_zoo/migan_inference.py:7
↓ 5 callersClasssynthesis_layer
lib/model_zoo/stylegan.py:247
↓ 3 callersClassEncoderBlock
lib/model_zoo/migan_inference.py:173
↓ 3 callersClassEncoderBlock
lib/model_zoo/migan.py:208
↓ 3 callersClassFeatureStats
lib/evaluator/stylegan_metrics/metric_utils.py:55
↓ 3 callersClasstorgb_layer
lib/model_zoo/stylegan.py:313
↓ 2 callersClassDense
lib/model_zoo/migan.py:14
↓ 2 callersClassDiscriminatorBlock
lib/model_zoo/migan.py:562
↓ 2 callersClassMapping
lib/model_zoo/comodgan.py:27
↓ 2 callersClassProgressMonitor
lib/evaluator/stylegan_metrics/metric_utils.py:153
↓ 2 callersClassUpsample2d
lib/model_zoo/migan_inference.py:79
↓ 2 callersClassdataset_cfg_bank
lib/cfg_helper.py:157
↓ 2 callersClassdiscrim_block
lib/model_zoo/stylegan.py:638
↓ 2 callersClassencoder_block
lib/model_zoo/comodgan.py:31
↓ 2 callersClassexec_container
This is the base functor for all types of executions. One execution can have multiple stages, but are only allowed to use the sa
lib/utils.py:11
↓ 2 callersClassget_formatter
lib/data_factory/common/ds_formatter.py:12
↓ 2 callersClassget_loader
lib/data_factory/common/ds_loader.py:18
↓ 2 callersClassget_unit
lib/model_zoo/common/utils.py:42
↓ 2 callersClassmodel_cfg_bank
lib/cfg_helper.py:104
↓ 2 callersClasssuppress_tracer_warnings
torch_utils/misc.py:69
↓ 1 callersClassDiscriminatorEpilogue
lib/model_zoo/migan.py:647
↓ 1 callersClassDownsample2d
lib/model_zoo/migan_inference.py:58
↓ 1 callersClassEncoder
lib/model_zoo/migan_inference.py:203
↓ 1 callersClassEncoder
lib/model_zoo/comodgan.py:115
↓ 1 callersClassGaussianSmoothing
Apply gaussian smoothing on a 1d, 2d or 3d tensor. Filtering is performed seperately for each channel in the input using a depthwise conv
scripts/create_onnx_pipeline.py:49
↓ 1 callersClassGenerator
lib/model_zoo/comodgan.py:425
↓ 1 callersClassInferenceDataset
scripts/evaluate_fid_lpips.py:124
↓ 1 callersClassMIGAN_Pipeline
scripts/create_onnx_pipeline.py:118
↓ 1 callersClassMapping
lib/model_zoo/stylegan.py:356
↓ 1 callersClassSynthesis
lib/model_zoo/migan_inference.py:318
↓ 1 callersClassSynthesis
lib/model_zoo/comodgan.py:347
↓ 1 callersClassSynthesisBlock
lib/model_zoo/migan_inference.py:282
↓ 1 callersClassSynthesisBlock
lib/model_zoo/migan.py:393
↓ 1 callersClassSynthesisBlockFirst
lib/model_zoo/migan_inference.py:249
↓ 1 callersClassSynthesisBlockFirst
lib/model_zoo/migan.py:334
↓ 1 callersClass_LegacyUnpickler
lib/experiments/comodgan_default.py:736
↓ 1 callersClasscompose
lib/evaluator/eva_base.py:245
↓ 1 callersClasscompose
lib/data_factory/common/ds_loader.py:36
↓ 1 callersClassdiscrim_epilogue
lib/model_zoo/stylegan.py:724
↓ 1 callersClassencoder_epilogue
lib/model_zoo/comodgan.py:64
↓ 1 callersClassexperiment_cfg_bank
lib/cfg_helper.py:208
↓ 1 callersClassget_evaluator
lib/evaluator/eva_base.py:22
↓ 1 callersClassminibatch_std_layer
lib/model_zoo/migan.py:624
↓ 1 callersClassminibatch_std_layer
lib/model_zoo/stylegan.py:701
↓ 1 callersClasssynthesis_block
lib/model_zoo/comodgan.py:265
↓ 1 callersClasssynthesis_block
lib/model_zoo/stylegan.py:446
↓ 1 callersClasssynthesis_block_first
lib/model_zoo/comodgan.py:207
ClassAdvInpaintingFormatter
This formatter perform the following operation. a) Random scale the image h w so it is in a designated range = [1, 1.2] or larger. i.e
lib/data_factory/ds_places2.py:156
ClassBiasActCuda
torch_utils/ops/bias_act.py:145
ClassBiasActCudaGrad
torch_utils/ops/bias_act.py:178
ClassCenterMaskFormatter
This formatter that use the same center mask
lib/data_factory/ds_ffhq.py:127
ClassCenterMaskFormatter
This formatter that use the same center mask
lib/data_factory/ds_places2.py:109
ClassCoModGanLoss
lib/experiments/comodgan_loss.py:22
ClassCollector
r"""Collects the scalars broadcasted by `report()` and `report0()` and computes their long-term averages (mean and standard deviation) over us
torch_utils/training_stats.py:113
ClassConv2d
torch_utils/ops/conv2d_gradfix.py:107
ClassConv2dGradWeight
torch_utils/ops/conv2d_gradfix.py:140
ClassDecorator
torch_utils/persistence.py:99
ClassDefaultFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_ffhq.py:103
ClassDefaultFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_places2.py:85
ClassDefaultLoader
lib/data_factory/ds_ffhq.py:71
ClassDefaultLoader
lib/data_factory/ds_places2.py:53
ClassDiscriminator
lib/model_zoo/migan.py:695
ClassDiscriminator
lib/model_zoo/comodgan.py:459
ClassDiscriminator
lib/model_zoo/stylegan.py:776
ClassEasyDict
Convenience class that behaves like a dict but allows access with the attribute syntax.
dnnlib/util.py:40
ClassEncoder
lib/model_zoo/migan.py:261
ClassFixResolutionLoader
Loader with resolution 512
lib/data_factory/ds_places2.py:65
ClassFixedMaskFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_places2.py:132
ClassFreeFormMaskFormatter
This formatter perform the following operation. a) Rescale the image h w to the designated resolution.
lib/data_factory/ds_places2.py:187
ClassGenerator
lib/model_zoo/migan_inference.py:355
ClassGenerator
lib/model_zoo/migan.py:528
ClassGenerator
lib/model_zoo/stylegan.py:593
ClassImageOnlyFormatter
lib/data_factory/ds_ffhq.py:258
ClassInfiniteSampler
torch_utils/misc.py:109
ClassLogger
Redirect stderr to stdout, optionally print stdout to a file, and optionally force flushing on both stdout and the file.
dnnlib/util.py:56
ClassLoss
lib/experiments/loss.py:19
ClassLoss
lib/experiments/comodgan_loss.py:17
ClassMIGANLoss
lib/experiments/loss.py:24
ClassMetricOptions
lib/evaluator/stylegan_metrics/metric_utils.py:21
ClassR512Loader
Loader with resolution 512
lib/data_factory/ds_ffhq.py:83
ClassRandomMaskFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_ffhq.py:340
ClassSSIM
lib/evaluator/eva_ssim.py:43
ClassSynthesis
lib/model_zoo/migan.py:469
ClassSynthesis
lib/model_zoo/stylegan.py:534
ClassUpfirdn2dCuda
torch_utils/ops/upfirdn2d.py:228
ClassZipLoader
lib/data_factory/ds_ffhq.py:316
Class_FusedMultiplyAdd
torch_utils/ops/fma.py:20
Class_GridSample2dBackward
torch_utils/ops/grid_sample_gradfix.py:61
Class_GridSample2dForward
torch_utils/ops/grid_sample_gradfix.py:44
Class_TFNetworkStub
lib/experiments/comodgan_default.py:733
Classbase_evaluator
lib/evaluator/eva_base.py:71
next →1–100 of 121, ranked by callers