MCPcopy Create free account

hub / github.com/Janspiry/Palette-Image-to-Image-Diffusion-Models / functions

Functions188 in github.com/Janspiry/Palette-Image-to-Image-Diffusion-Models

⨍ Functions 188◇ Types & classes 39
Method__getitem__
(self, index)
core/base_dataset.py:42
Method__getitem__
(self, index)
data/dataset.py:52
Method__getitem__
(self, index)
data/dataset.py:109
Method__getitem__
(self, index)
data/dataset.py:161
Method__init__
(self, init_type='kaiming', gain=0.02)
core/base_network.py:3
Method__init__
(self, data_root, image_size=[256, 256], loader=pil_loader)
core/base_dataset.py:33
Method__init__
init model with basic input, which are from __init__(**kwargs) function in inherited class
core/base_model.py:14
Method__init__
(self, opt)
core/logger.py:14
Method__init__
(self, opt, logger)
core/logger.py:55
Method__init__
(self, *keys, phase='train')
core/logger.py:150
Method__init__
(self, data_root, mask_config={}, data_len=-1, image_size=[256, 256], loader=pil_loader)
data/dataset.py:36
Method__init__
(self, data_root, mask_config={}, data_len=-1, image_size=[256, 256], loader=pil_loader)
data/dataset.py:93
Method__init__
(self, data_root, data_flist, data_len=-1, image_size=[224, 224], loader=pil_loader)
data/dataset.py:146
Method__init__
(self)
data/util/auto_augment.py:8
Method__init__
(self)
data/util/auto_augment.py:43
Method__init__
(self, length=16)
data/util/auto_augment.py:285
Method__init__
(self, gamma=2, alpha=None, size_average=True)
models/loss.py:19
Method__init__
(self, unet, beta_schedule, module_name='sr3', **kwargs)
models/network.py:9
Method__init__
must to init BaseModel with kwargs
models/model.py:20
Method__init__
(self, channels, use_conv, out_channel=None)
models/guided_diffusion_modules/unet.py:53
Method__init__
(self, channels, use_conv, out_channel=None)
models/guided_diffusion_modules/unet.py:75
Method__init__
( self, channels, emb_channels, dropout, out_channel=None, use
models/guided_diffusion_modules/unet.py:109
Method__init__
( self, channels, num_heads=1, num_head_channels=-1, use_checkpoint=Fa
models/guided_diffusion_modules/unet.py:212
Method__init__
(self, n_heads)
models/guided_diffusion_modules/unet.py:258
Method__init__
(self, n_heads)
models/guided_diffusion_modules/unet.py:290
Method__init__
(self, dim)
models/sr3_modules/unet.py:109
Method__init__
(self, in_channels, out_channels, use_affine_level=False)
models/sr3_modules/unet.py:122
Method__init__
(self, dim)
models/sr3_modules/unet.py:145
Method__init__
(self, dim)
models/sr3_modules/unet.py:155
Method__init__
(self, dim, dim_out, groups=32, dropout=0)
models/sr3_modules/unet.py:167
Method__init__
(self, dim, dim_out, noise_level_emb_dim=None, dropout=0, use_affine_level=False, norm_groups=32)
models/sr3_modules/unet.py:181
Method__init__
(self, in_channel, n_head=1, norm_groups=32)
models/sr3_modules/unet.py:199
Method__init__
(self, dim, dim_out, *, noise_level_emb_dim=None, norm_groups=32, dropout=0, with_attn=False)
models/sr3_modules/unet.py:229
Method__len__
(self)
core/base_dataset.py:47
Method__len__
(self)
data/dataset.py:67
Method__len__
(self)
data/dataset.py:124
Method__len__
(self)
data/dataset.py:173
Method__missing__
(self, key)
core/praser.py:70
Method_forward
(self, x, emb)
models/guided_diffusion_modules/unet.py:183
Method_forward
(self, x)
models/guided_diffusion_modules/unet.py:244
Methodavg
(self, key)
core/logger.py:164
Methodcount_flops
(model, _x, y)
models/guided_diffusion_modules/unet.py:281
Methodcount_flops
(model, _x, y)
models/guided_diffusion_modules/unet.py:315
Functiondict2str
dict to string for logger
core/praser.py:85
Methodforward
(self, input, target)
models/loss.py:27
Methodforward
(self, y_0, y_cond=None, mask=None, noise=None)
models/network.py:105
Methodforward
(self, x)
models/guided_diffusion_modules/nn.py:12
Methodforward
(ctx, run_function, length, *args)
models/guided_diffusion_modules/nn.py:72
Methodforward
(self, x)
models/guided_diffusion_modules/unet.py:17
Methodforward
Apply the module to `x` given `emb` embeddings.
models/guided_diffusion_modules/unet.py:26
Methodforward
(self, x, emb)
models/guided_diffusion_modules/unet.py:37
Methodforward
(self, x)
models/guided_diffusion_modules/unet.py:61
Methodforward
(self, x)
models/guided_diffusion_modules/unet.py:89
Methodforward
Apply the block to a Tensor, conditioned on a embedding. :param x: an [N x C x ...] Tensor of features. :param emb: an [N x e
models/guided_diffusion_modules/unet.py:172
Methodforward
(self, x)
models/guided_diffusion_modules/unet.py:241
Methodforward
Apply QKV attention. :param qkv: an [N x (H * 3 * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte
models/guided_diffusion_modules/unet.py:262
Methodforward
Apply QKV attention. :param qkv: an [N x (3 * H * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte
models/guided_diffusion_modules/unet.py:294
Methodforward
Apply the model to an input batch. :param x: an [N x 2 x ...] Tensor of inputs (B&W) :param gammas: a 1-D batch of gammas.
models/guided_diffusion_modules/unet.py:525
Methodforward
(self, x, time)
models/sr3_modules/unet.py:80
Methodforward
(self, noise_level)
models/sr3_modules/unet.py:113
Methodforward
(self, x, noise_embed)
models/sr3_modules/unet.py:129
Methodforward
(self, x)
models/sr3_modules/unet.py:140
Methodforward
(self, x)
models/sr3_modules/unet.py:150
Methodforward
(self, x)
models/sr3_modules/unet.py:159
Methodforward
(self, x)
models/sr3_modules/unet.py:176
Methodforward
(self, x, time_emb)
models/sr3_modules/unet.py:190
Methodforward
(self, input)
models/sr3_modules/unet.py:208
Methodforward
(self, x, time_emb)
models/sr3_modules/unet.py:237
Methodinit_func
(m)
core/base_network.py:15
Methodload_everything
(self)
core/base_model.py:166
Functionmae
(input, target)
models/metric.py:12
Functionmean_flat
Take the mean over all non-batch dimensions.
models/guided_diffusion_modules/nn.py:34
Functionmse_loss
(output, target)
models/loss.py:14
Functionparse
(args)
core/praser.py:97
Functionpil_loader
(path)
core/base_dataset.py:29
Functionpil_loader
(path)
data/dataset.py:32
Functionpostprocess
(images)
core/util.py:33
Methodprint_network
print network structure, only work on GPU 0
core/base_model.py:83
Methodsave_everything
load pretrained model and training state.
models/model.py:212
Functionscale_module
Scale the parameters of a module and return it.
models/guided_diffusion_modules/nn.py:25
Functionset_device
set parameter to gpu or cpu
core/util.py:64
Functionset_seed
set random seed, gl_seed used in worker_init_fn function
core/util.py:37
Methodtest
(self)
core/base_model.py:69
Methodtest_step
(self)
core/base_model.py:80
Methodtrain_step
(self)
models/model.py:104
Methodval_step
(self)
models/model.py:131
Methodwrapper
(info, *args, **kwargs)
core/logger.py:25
Methodwrapper
(tag, data, *args, **kwargs)
core/logger.py:130
← previous101–188 of 188, ranked by callers