Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Picsart-AI-Research/MI-GAN
/ types & classes
Types & classes
121 in github.com/Picsart-AI-Research/MI-GAN
⨍
Functions
526
◇
Types & classes
121
↓ 16 callers
Class
Conv2d
lib/model_zoo/migan.py:54
↓ 15 callers
Class
get_model
lib/model_zoo/common/get_model.py:57
↓ 9 callers
Class
SeparableConv2d
lib/model_zoo/migan.py:152
↓ 9 callers
Class
dense
lib/model_zoo/stylegan.py:64
↓ 7 callers
Class
conv2d_layer
lib/model_zoo/stylegan.py:198
↓ 6 callers
Class
SeparableConv2d
lib/model_zoo/migan_inference.py:106
↓ 6 callers
Class
get_dataset
lib/data_factory/common/ds_base.py:62
↓ 5 callers
Class
DistributedSampler
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 callers
Class
collate
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 callers
Class
lrelu_agc
The lrelu layer with alpha, gain and clamp
lib/model_zoo/migan_inference.py:7
↓ 5 callers
Class
synthesis_layer
lib/model_zoo/stylegan.py:247
↓ 3 callers
Class
EncoderBlock
lib/model_zoo/migan_inference.py:173
↓ 3 callers
Class
EncoderBlock
lib/model_zoo/migan.py:208
↓ 3 callers
Class
FeatureStats
lib/evaluator/stylegan_metrics/metric_utils.py:55
↓ 3 callers
Class
torgb_layer
lib/model_zoo/stylegan.py:313
↓ 2 callers
Class
Dense
lib/model_zoo/migan.py:14
↓ 2 callers
Class
DiscriminatorBlock
lib/model_zoo/migan.py:562
↓ 2 callers
Class
Mapping
lib/model_zoo/comodgan.py:27
↓ 2 callers
Class
ProgressMonitor
lib/evaluator/stylegan_metrics/metric_utils.py:153
↓ 2 callers
Class
Upsample2d
lib/model_zoo/migan_inference.py:79
↓ 2 callers
Class
dataset_cfg_bank
lib/cfg_helper.py:157
↓ 2 callers
Class
discrim_block
lib/model_zoo/stylegan.py:638
↓ 2 callers
Class
encoder_block
lib/model_zoo/comodgan.py:31
↓ 2 callers
Class
exec_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 callers
Class
get_formatter
lib/data_factory/common/ds_formatter.py:12
↓ 2 callers
Class
get_loader
lib/data_factory/common/ds_loader.py:18
↓ 2 callers
Class
get_unit
lib/model_zoo/common/utils.py:42
↓ 2 callers
Class
model_cfg_bank
lib/cfg_helper.py:104
↓ 2 callers
Class
suppress_tracer_warnings
torch_utils/misc.py:69
↓ 1 callers
Class
DiscriminatorEpilogue
lib/model_zoo/migan.py:647
↓ 1 callers
Class
Downsample2d
lib/model_zoo/migan_inference.py:58
↓ 1 callers
Class
Encoder
lib/model_zoo/migan_inference.py:203
↓ 1 callers
Class
Encoder
lib/model_zoo/comodgan.py:115
↓ 1 callers
Class
GaussianSmoothing
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 callers
Class
Generator
lib/model_zoo/comodgan.py:425
↓ 1 callers
Class
InferenceDataset
scripts/evaluate_fid_lpips.py:124
↓ 1 callers
Class
MIGAN_Pipeline
scripts/create_onnx_pipeline.py:118
↓ 1 callers
Class
Mapping
lib/model_zoo/stylegan.py:356
↓ 1 callers
Class
Synthesis
lib/model_zoo/migan_inference.py:318
↓ 1 callers
Class
Synthesis
lib/model_zoo/comodgan.py:347
↓ 1 callers
Class
SynthesisBlock
lib/model_zoo/migan_inference.py:282
↓ 1 callers
Class
SynthesisBlock
lib/model_zoo/migan.py:393
↓ 1 callers
Class
SynthesisBlockFirst
lib/model_zoo/migan_inference.py:249
↓ 1 callers
Class
SynthesisBlockFirst
lib/model_zoo/migan.py:334
↓ 1 callers
Class
_LegacyUnpickler
lib/experiments/comodgan_default.py:736
↓ 1 callers
Class
compose
lib/evaluator/eva_base.py:245
↓ 1 callers
Class
compose
lib/data_factory/common/ds_loader.py:36
↓ 1 callers
Class
discrim_epilogue
lib/model_zoo/stylegan.py:724
↓ 1 callers
Class
encoder_epilogue
lib/model_zoo/comodgan.py:64
↓ 1 callers
Class
experiment_cfg_bank
lib/cfg_helper.py:208
↓ 1 callers
Class
get_evaluator
lib/evaluator/eva_base.py:22
↓ 1 callers
Class
minibatch_std_layer
lib/model_zoo/migan.py:624
↓ 1 callers
Class
minibatch_std_layer
lib/model_zoo/stylegan.py:701
↓ 1 callers
Class
synthesis_block
lib/model_zoo/comodgan.py:265
↓ 1 callers
Class
synthesis_block
lib/model_zoo/stylegan.py:446
↓ 1 callers
Class
synthesis_block_first
lib/model_zoo/comodgan.py:207
Class
AdvInpaintingFormatter
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
Class
BiasActCuda
torch_utils/ops/bias_act.py:145
Class
BiasActCudaGrad
torch_utils/ops/bias_act.py:178
Class
CenterMaskFormatter
This formatter that use the same center mask
lib/data_factory/ds_ffhq.py:127
Class
CenterMaskFormatter
This formatter that use the same center mask
lib/data_factory/ds_places2.py:109
Class
CoModGanLoss
lib/experiments/comodgan_loss.py:22
Class
Collector
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
Class
Conv2d
torch_utils/ops/conv2d_gradfix.py:107
Class
Conv2dGradWeight
torch_utils/ops/conv2d_gradfix.py:140
Class
Decorator
torch_utils/persistence.py:99
Class
DefaultFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_ffhq.py:103
Class
DefaultFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_places2.py:85
Class
DefaultLoader
lib/data_factory/ds_ffhq.py:71
Class
DefaultLoader
lib/data_factory/ds_places2.py:53
Class
Discriminator
lib/model_zoo/migan.py:695
Class
Discriminator
lib/model_zoo/comodgan.py:459
Class
Discriminator
lib/model_zoo/stylegan.py:776
Class
EasyDict
Convenience class that behaves like a dict but allows access with the attribute syntax.
dnnlib/util.py:40
Class
Encoder
lib/model_zoo/migan.py:261
Class
FixResolutionLoader
Loader with resolution 512
lib/data_factory/ds_places2.py:65
Class
FixedMaskFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_places2.py:132
Class
FreeFormMaskFormatter
This formatter perform the following operation. a) Rescale the image h w to the designated resolution.
lib/data_factory/ds_places2.py:187
Class
Generator
lib/model_zoo/migan_inference.py:355
Class
Generator
lib/model_zoo/migan.py:528
Class
Generator
lib/model_zoo/stylegan.py:593
Class
ImageOnlyFormatter
lib/data_factory/ds_ffhq.py:258
Class
InfiniteSampler
torch_utils/misc.py:109
Class
Logger
Redirect stderr to stdout, optionally print stdout to a file, and optionally force flushing on both stdout and the file.
dnnlib/util.py:56
Class
Loss
lib/experiments/loss.py:19
Class
Loss
lib/experiments/comodgan_loss.py:17
Class
MIGANLoss
lib/experiments/loss.py:24
Class
MetricOptions
lib/evaluator/stylegan_metrics/metric_utils.py:21
Class
R512Loader
Loader with resolution 512
lib/data_factory/ds_ffhq.py:83
Class
RandomMaskFormatter
This formatter is a direct replication of the original CoModGan TF code
lib/data_factory/ds_ffhq.py:340
Class
SSIM
lib/evaluator/eva_ssim.py:43
Class
Synthesis
lib/model_zoo/migan.py:469
Class
Synthesis
lib/model_zoo/stylegan.py:534
Class
Upfirdn2dCuda
torch_utils/ops/upfirdn2d.py:228
Class
ZipLoader
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
Class
base_evaluator
lib/evaluator/eva_base.py:71
next →
1–100 of 121, ranked by callers