Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/annegnx/PnP-Flow
/ functions
Functions
418 in github.com/annegnx/PnP-Flow
⨍
Functions
418
◇
Types & classes
92
Method
forward
(self, input)
pnpflow/image_generation/op/fused_act.py:82
Method
forward
( ctx, grad_output, kernel, grad_kernel, up, down, pad, g_pad, in_size, out_size )
pnpflow/image_generation/op/upfirdn2d.py:21
Method
forward
(ctx, input, kernel, up, down, pad)
pnpflow/image_generation/op/upfirdn2d.py:90
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:147
Method
forward
(self, x, y)
pnpflow/image_generation/models/layers.py:171
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:196
Method
forward
(self, x, y)
pnpflow/image_generation/models/layers.py:222
Method
forward
(self, xs, shape)
pnpflow/image_generation/models/layers.py:244
Method
forward
(self, xs, y, shape)
pnpflow/image_generation/models/layers.py:267
Method
forward
(self, xs, output_shape)
pnpflow/image_generation/models/layers.py:295
Method
forward
(self, xs, y, output_shape)
pnpflow/image_generation/models/layers.py:333
Method
forward
(self, inputs)
pnpflow/image_generation/models/layers.py:365
Method
forward
(self, inputs)
pnpflow/image_generation/models/layers.py:377
Method
forward
(self, inputs)
pnpflow/image_generation/models/layers.py:390
Method
forward
(self, x, y)
pnpflow/image_generation/models/layers.py:437
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:494
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:552
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:568
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:591
Method
forward
(self, x)
pnpflow/image_generation/models/layers.py:606
Method
forward
(self, x, temb=None)
pnpflow/image_generation/models/layers.py:645
Method
forward
(self, x, labels)
pnpflow/image_generation/models/ddpm.py:110
Method
forward
(self, x, time_cond)
pnpflow/image_generation/models/ncsnpp.py:233
Method
forward
(self, x)
pnpflow/image_generation/models/layerspp.py:39
Method
forward
(self, x, y)
pnpflow/image_generation/models/layerspp.py:52
Method
forward
(self, x)
pnpflow/image_generation/models/layerspp.py:75
Method
forward
(self, x)
pnpflow/image_generation/models/layerspp.py:114
Method
forward
(self, x)
pnpflow/image_generation/models/layerspp.py:149
Method
forward
(self, x, temb=None)
pnpflow/image_generation/models/layerspp.py:193
Method
forward
(self, x, temb=None)
pnpflow/image_generation/models/layerspp.py:242
Method
forward
(self, x, y)
pnpflow/image_generation/models/ncsnv2.py:106
Method
forward
(self, x, y)
pnpflow/image_generation/models/ncsnv2.py:196
Method
forward
(self, x, y)
pnpflow/image_generation/models/ncsnv2.py:284
Method
forward
(self, x, y)
pnpflow/image_generation/models/ncsnv2.py:386
Method
forward
(self, x, y)
pnpflow/image_generation/models/normalization.py:57
Method
forward
(self, x, y)
pnpflow/image_generation/models/normalization.py:82
Method
forward
(self, x, y)
pnpflow/image_generation/models/normalization.py:101
Method
forward
(self, x)
pnpflow/image_generation/models/normalization.py:118
Method
forward
(self, x, y)
pnpflow/image_generation/models/normalization.py:139
Method
forward
(self, x)
pnpflow/image_generation/models/normalization.py:153
Method
forward
(self, x)
pnpflow/image_generation/models/normalization.py:170
Method
forward
(self, x, y)
pnpflow/image_generation/models/normalization.py:200
Method
forward
(self, x)
pnpflow/image_generation/models/up_or_down_sampling.py:45
Method
forward
(self, t, x)
pnpflow/methods/d_flow.py:209
Method
forward
r""" Add noise to the input :param torch.Tensor input: input tensor :param int seed: the seed for the random number generator.
pnpflow/methods/pnp_diff.py:198
Method
forward
r""" Adds the noise to measurements x :param torch.Tensor x: measurements :param float, torch.Tensor sigma: standard deviatio
pnpflow/methods/pnp_diff.py:269
Function
fused_bias_act
pnpflow/image_generation/op/fused_bias_act.cpp:11
Method
gaussian
(self, img)
pnpflow/methods/d_flow.py:35
Function
gaussian_blur
Blur a tensor image with Gaussian filter x: tensor image, NxCxWxH sigma: standard deviation of the Gaussian kernel
pnpflow/utils.py:313
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train_online.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k_g_1.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/bedroom_rf_gaussian.py:22
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_ddpmpp.py:22
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/celeba_hq_pytorch_rf_gaussian.py:22
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_generate_data.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/church_rf_gaussian.py:22
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1_online.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_train.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/cifar10_rf_gaussian_reflow_distill_k=1.py:23
Function
get_config
()
pnpflow/image_generation/configs/rectified_flow/afhq_cat_pytorch_rf_gaussian.py:22
Function
get_data_inverse_scaler
Inverse data normalizer.
pnpflow/image_generation/datasets.py:29
Function
get_data_scaler
Data normalizer. Assume data are always in [0, 1].
pnpflow/image_generation/datasets.py:20
Function
get_ddpm_params
Get betas and alphas --- parameters used in the original DDPM paper.
pnpflow/image_generation/models/utils.py:65
Function
get_inception_model
(inceptionv3=False)
pnpflow/image_generation/evaluation.py:34
Function
get_likelihood_fn
Create a function to compute the unbiased log-likelihood estimate of a given data point. Args: sde: A `sde_lib.SDE` object that represents the
pnpflow/image_generation/likelihood.py:41
Function
get_likelihood_fn_rf
Create a function to compute the unbiased log-likelihood estimate of a given data point for Rectified Flow. Args: sde: A `sde_lib.SDE` object t
pnpflow/image_generation/likelihood.py:116
Function
get_network
(config)
pnpflow/image_generation/models/ncsnv2.py:31
Function
get_optimizer
Returns a flax optimizer object based on `config`.
pnpflow/image_generation/losses.py:26
Function
get_sampling_fn
Create a sampling function. Args: config: A `ml_collections.ConfigDict` object that contains all configuration information. sde: A `s
pnpflow/image_generation/sampling.py:36
Function
get_save_path_ip
dict_cfg_method contains keys and values of the method. Return path composed of key1=value1/key2=value2/.../keyN=valueN
pnpflow/utils.py:1112
Function
get_score_fn
Wraps `score_fn` so that the model output corresponds to a real time-dependent score function. Args: sde: An `sde_lib.SDE` object that repr
pnpflow/image_generation/models/utils.py:138
Function
get_step_fn
Create a one-step training/evaluation function. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. optimize_fn: An o
pnpflow/image_generation/losses.py:146
Function
get_timestep_embedding
(timesteps, embedding_dim, max_positions=10000)
pnpflow/image_generation/models/layers.py:515
Function
get_weight
Get/create weight tensor for a convolution or fully-connected layer.
pnpflow/image_generation/models/up_or_down_sampling.py:14
Function
group_norm
(out_ch)
pnpflow/models.py:33
Function
hut_estimator
(NO_test, v, inp, t)
pnpflow/utils.py:243
Function
init
(shape, dtype=dtype, device=device)
pnpflow/image_generation/models/layers.py:66
Function
load_dataset_stats
Load the pre-computed dataset statistics.
pnpflow/image_generation/evaluation.py:42
Function
main
(argv)
pnpflow/image_generation/main.py:39
Method
model_forward
(self, x)
pnpflow/methods/pnp_gs.py:19
Function
naive_downsample_2d
(x, factor=2)
pnpflow/image_generation/models/up_or_down_sampling.py:66
Function
naive_upsample_2d
(x, factor=2)
pnpflow/image_generation/models/up_or_down_sampling.py:59
Function
ncsn_conv1x1
1x1 convolution. Same as NCSNv1/v2.
pnpflow/image_generation/models/layers.py:44
Function
ode_func
(t, x)
pnpflow/image_generation/likelihood.py:92
Function
ode_func
(t, x)
pnpflow/image_generation/sampling.py:135
Method
ode_func
(t, x)
pnpflow/image_generation/sde_lib.py:54
Method
operator
(vec)
pnpflow/train_denoiser.py:151
Function
optimization_manager
Returns an optimize_fn based on `config`.
pnpflow/image_generation/losses.py:38
Function
preprocess
(img, args)
pnpflow/utils.py:546
Method
prox
(self, x, y, physics, gamma=1.0)
pnpflow/methods/pnp_diff.py:103
Method
prox
(self, x, y, physics, gamma=1.0)
pnpflow/methods/pnp_diff.py:128
Method
prox
(self, x, y, physics, gamma=1.0)
pnpflow/methods/pnp_diff.py:154
Method
prox
( self, x, y, physics, *args, gamma=1.0, stepsize=None
pnpflow/methods/pnp_diff.py:331
Method
rand_like
r""" Equivalent to `torch.rand_like` but supports a pseudorandom number generator argument. :param int seed: the seed for the random n
pnpflow/methods/pnp_diff.py:244
Function
register_model
A decorator for registering model classes.
pnpflow/image_generation/models/utils.py:27
Method
reset_rng
r""" Reset the random number generator to its initial state.
pnpflow/methods/pnp_diff.py:238
Function
rk45_sampler
The probability flow ODE sampler with black-box ODE solver. Args: model: A velocity model. z: If present, generate sample
pnpflow/image_generation/sampling.py:111
Function
run_inception_distributed
Distribute the inception network computation to all available TPUs. Args: input_tensor: The input images. Assumed to be within [0, 255].
pnpflow/image_generation/evaluation.py:107
Method
run_method
(self, data_loaders, degradation, sigma_noise)
pnpflow/methods/d_flow.py:178
Method
run_method
(self, data_loaders, degradation, sigma_noise)
pnpflow/methods/pnp_gs.py:253
← previous
next →
301–400 of 418, ranked by callers