Index

A

affine (neural_de.transformations.diffusion.unet.nn.GroupNorm32 attribute) attention_resolutions (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1]

AttentionBlock (class in neural_de.transformations.diffusion.unet.attention_block) avg_pool_nd() (in module neural_de.transformations.diffusion.unet.nn)

B

backward() (neural_de.transformations.diffusion.unet.nn.CheckpointFunction static method) BaseTransformation (class in neural_de.transformations.transformation) block_gmlp_factor (in module neural_de.external.maxim_tf.maxim.maxim) block_size_hr (in module neural_de.external.maxim_tf.maxim.maxim) block_size_lr (in module neural_de.external.maxim_tf.maxim.maxim)

BlockGatingUnit() (in module neural_de.external.maxim_tf.maxim.blocks.block_gating) BlockGmlpLayer() (in module neural_de.external.maxim_tf.maxim.blocks.block_gating) BlockImages (class in neural_de.external.maxim_tf.maxim.layers) BottleneckBlock() (in module neural_de.external.maxim_tf.maxim.blocks.bottleneck) BrightnessEnhancer (class in neural_de.transformations.brightness_enhancer) build_conv_block() (neural_de.external.derain.blocks.DeformableResnetBlock method)

C

CALayer() (in module neural_de.external.maxim_tf.maxim.blocks.attentions) call() (neural_de.external.maxim_tf.maxim.layers.BlockImages method) (neural_de.external.maxim_tf.maxim.layers.Resizing method) (neural_de.external.maxim_tf.maxim.layers.SwapAxes method) (neural_de.external.maxim_tf.maxim.layers.UnblockImages method) camel_to_snake() (in module neural_de.transformations.transformation_pipeline) CenteredZoom (class in neural_de.transformations.centered_zoom) channel_mult (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] channels_reduction (in module neural_de.external.maxim_tf.maxim.maxim) check_device_validity() (neural_de.transformations.transformation.BaseTransformation method)

checkpoint() (in module neural_de.transformations.diffusion.unet.nn) CheckpointFunction (class in neural_de.transformations.diffusion.unet.nn) Conv2d (class in neural_de.external.derain.blocks) conv_nd() (in module neural_de.transformations.diffusion.unet.nn) conv_resample (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] convert_module_to_f16() (in module neural_de.transformations.diffusion.unet.utils) convert_module_to_f32() (in module neural_de.transformations.diffusion.unet.utils) convert_to_fp16() (neural_de.transformations.diffusion.unet.unet_model.UNetModel method) convert_to_fp32() (neural_de.transformations.diffusion.unet.unet_model.UNetModel method) crop_image() (in module neural_de.utils.math) CrossGatingBlock() (in module neural_de.external.maxim_tf.maxim.blocks.misc_gating)

D

DecoderBlock (class in neural_de.external.derain.blocks) DeformableConv2d (class in neural_de.external.derain.blocks) DeformableResnetBlock (class in neural_de.external.derain.blocks) depth (in module neural_de.external.maxim_tf.maxim.maxim) DeRainConfig (class in neural_de.transformations.de_rain_enhancer) DeRainEnhancer (class in neural_de.transformations.de_rain_enhancer) DeSnowEnhancer (class in neural_de.transformations.de_snow_enhancer)

DiffPureConfig (class in neural_de.transformations.diffusion.diffpure_config) DiffusionEnhancer (class in neural_de.transformations.diffusion.diffusion_enhancer) dims (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] download_model() (neural_de.utils.model_manager.ModelManager method) Downsample (class in neural_de.transformations.diffusion.unet.downsample) dropout (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] dropout_rate (in module neural_de.external.maxim_tf.maxim.maxim) (neural_de.transformations.night_image_enhancer.NightConfig attribute)

E

enhance_brightness() (neural_de.transformations.brightness_enhancer.BrightnessEnhancer method)

eps (neural_de.transformations.diffusion.unet.nn.GroupNorm32 attribute)

F

f() (neural_de.transformations.diffusion.rev_vpsde.RevVPSDE method) features (in module neural_de.external.maxim_tf.maxim.maxim) forward() (neural_de.external.derain.blocks.Conv2d method) (neural_de.external.derain.blocks.DecoderBlock method) (neural_de.external.derain.blocks.DeformableConv2d method) (neural_de.external.derain.blocks.DeformableResnetBlock method) (neural_de.external.derain.blocks.ResNetModified method) (neural_de.external.derain.blocks.UpConv2d method) (neural_de.external.prenet.networks.PReNet method) (neural_de.external.prenet.networks.PReNet_GRU method) (neural_de.external.prenet.networks.PReNet_LSTM method) (neural_de.external.prenet.networks.PReNet_r method) (neural_de.external.prenet.networks.PReNet_x method) (neural_de.external.prenet.networks.PRN method) (neural_de.external.prenet.networks.PRN_r method) (neural_de.transformations.diffusion.diffusion_enhancer.DiffusionEnhancer method) (neural_de.transformations.diffusion.unet.attention_block.AttentionBlock method) (neural_de.transformations.diffusion.unet.downsample.Downsample method) (neural_de.transformations.diffusion.unet.nn.CheckpointFunction static method) (neural_de.transformations.diffusion.unet.nn.GroupNorm32 method) (neural_de.transformations.diffusion.unet.qkv_attention.QKVAttention method) (neural_de.transformations.diffusion.unet.qkv_attention_legacy.QKVAttentionLegacy method) (neural_de.transformations.diffusion.unet.res_block.ResBlock method) (neural_de.transformations.diffusion.unet.timestep_block.TimestepBlock method) (neural_de.transformations.diffusion.unet.timestep_embed_sequential.TimestepEmbedSequential method) (neural_de.transformations.diffusion.unet.unet_model.UNetModel method) (neural_de.transformations.diffusion.unet.upsample.Upsample method)

G

g() (neural_de.transformations.diffusion.rev_vpsde.RevVPSDE method) get_config() (neural_de.external.maxim_tf.maxim.layers.BlockImages method) (neural_de.external.maxim_tf.maxim.layers.Resizing method) (neural_de.external.maxim_tf.maxim.layers.SwapAxes method) (neural_de.external.maxim_tf.maxim.layers.UnblockImages method) get_logger() (in module neural_de.utils.twe_logger) get_pad_value() (in module neural_de.utils.math) getG() (in module neural_de.external.nplie.nplie)

GetSpatialGatingWeights() (in module neural_de.external.maxim_tf.maxim.blocks.misc_gating) grid_gmlp_factor (in module neural_de.external.maxim_tf.maxim.maxim) grid_size_hr (in module neural_de.external.maxim_tf.maxim.maxim) grid_size_lr (in module neural_de.external.maxim_tf.maxim.maxim) GridGatingUnit() (in module neural_de.external.maxim_tf.maxim.blocks.grid_gating) GridGmlpLayer() (in module neural_de.external.maxim_tf.maxim.blocks.grid_gating) GroupNorm32 (class in neural_de.transformations.diffusion.unet.nn) guideFilter() (in module neural_de.external.nplie.nplie)

H

high_res_stages (in module neural_de.external.maxim_tf.maxim.maxim)

I

image_editing_sample() (neural_de.transformations.diffusion.rev_guided_diffusion.RevGuidedDiffusion method) img_shape (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] imgrad() (in module neural_de.external.nplie.nplie) initIlluminunce() (in module neural_de.external.nplie.nplie) initJm() (in module neural_de.external.nplie.nplie)

input_nc (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) input_proj_factor (in module neural_de.external.maxim_tf.maxim.maxim) is_batch_valid() (in module neural_de.utils.validation) is_device_valid() (in module neural_de.utils.validation) is_power_of_two() (in module neural_de.utils.validation) is_scaled() (in module neural_de.utils.math)

K

KernelDeblurringEnhancer (class in neural_de.transformations.kernel_deblurring_enhancer)

L

learn_sigma (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] linear() (in module neural_de.transformations.diffusion.unet.nn) log_and_raise() (in module neural_de.utils.twe_logger) LOG_FILE (in module neural_de.utils.twe_logger)

LOG_LEVEL (in module neural_de.utils.twe_logger) LOG_OUTPUT (in module neural_de.utils.twe_logger) log_str_to_level() (in module neural_de.utils.twe_logger) LOGGER_DEFAULT_NAME (in module neural_de.utils.twe_logger) lrelu_slope (in module neural_de.external.maxim_tf.maxim.maxim)

M

main() (in module neural_de.main) MAXIM() (in module neural_de.external.maxim_tf.maxim.maxim) MlpBlock() (in module neural_de.external.maxim_tf.maxim.blocks.others) Model() (in module neural_de.external.maxim_tf.create_maxim_model) ModelManager (class in neural_de.utils.model_manager) module neural_de neural_de.external neural_de.external.derain neural_de.external.derain.blocks neural_de.external.maxim_tf neural_de.external.maxim_tf.create_maxim_model neural_de.external.maxim_tf.maxim neural_de.external.maxim_tf.maxim.blocks neural_de.external.maxim_tf.maxim.blocks.attentions neural_de.external.maxim_tf.maxim.blocks.block_gating neural_de.external.maxim_tf.maxim.blocks.bottleneck neural_de.external.maxim_tf.maxim.blocks.grid_gating neural_de.external.maxim_tf.maxim.blocks.misc_gating neural_de.external.maxim_tf.maxim.blocks.others neural_de.external.maxim_tf.maxim.blocks.unet neural_de.external.maxim_tf.maxim.configs neural_de.external.maxim_tf.maxim.layers neural_de.external.maxim_tf.maxim.maxim neural_de.external.nplie neural_de.external.nplie.nplie neural_de.external.prenet neural_de.external.prenet.networks neural_de.main neural_de.transformations neural_de.transformations.brightness_enhancer neural_de.transformations.centered_zoom neural_de.transformations.de_rain_enhancer neural_de.transformations.de_snow_enhancer neural_de.transformations.diffusion neural_de.transformations.diffusion.diffpure_config neural_de.transformations.diffusion.diffusion_enhancer neural_de.transformations.diffusion.rev_guided_diffusion neural_de.transformations.diffusion.rev_vpsde neural_de.transformations.diffusion.unet neural_de.transformations.diffusion.unet.attention_block neural_de.transformations.diffusion.unet.downsample neural_de.transformations.diffusion.unet.nn neural_de.transformations.diffusion.unet.qkv_attention neural_de.transformations.diffusion.unet.qkv_attention_legacy neural_de.transformations.diffusion.unet.res_block neural_de.transformations.diffusion.unet.timestep_block neural_de.transformations.diffusion.unet.timestep_embed_sequential neural_de.transformations.diffusion.unet.unet_model neural_de.transformations.diffusion.unet.upsample neural_de.transformations.diffusion.unet.utils neural_de.transformations.kernel_deblurring_enhancer neural_de.transformations.night_image_enhancer neural_de.transformations.resolution_enhancer neural_de.transformations.transformation neural_de.transformations.transformation_pipeline neural_de.utils neural_de.utils.math neural_de.utils.model_manager neural_de.utils.twe_logger neural_de.utils.validation

N

n_blocks (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) neural_de module neural_de.external module neural_de.external.derain module neural_de.external.derain.blocks module neural_de.external.maxim_tf module neural_de.external.maxim_tf.create_maxim_model module neural_de.external.maxim_tf.maxim module neural_de.external.maxim_tf.maxim.blocks module neural_de.external.maxim_tf.maxim.blocks.attentions module neural_de.external.maxim_tf.maxim.blocks.block_gating module neural_de.external.maxim_tf.maxim.blocks.bottleneck module neural_de.external.maxim_tf.maxim.blocks.grid_gating module neural_de.external.maxim_tf.maxim.blocks.misc_gating module neural_de.external.maxim_tf.maxim.blocks.others module neural_de.external.maxim_tf.maxim.blocks.unet module neural_de.external.maxim_tf.maxim.configs module neural_de.external.maxim_tf.maxim.layers module neural_de.external.maxim_tf.maxim.maxim module neural_de.external.nplie module neural_de.external.nplie.nplie module neural_de.external.prenet module neural_de.external.prenet.networks module neural_de.main module neural_de.transformations module neural_de.transformations.brightness_enhancer module neural_de.transformations.centered_zoom module neural_de.transformations.de_rain_enhancer module neural_de.transformations.de_snow_enhancer module neural_de.transformations.diffusion module neural_de.transformations.diffusion.diffpure_config module neural_de.transformations.diffusion.diffusion_enhancer module neural_de.transformations.diffusion.rev_guided_diffusion module neural_de.transformations.diffusion.rev_vpsde module

neural_de.transformations.diffusion.unet module neural_de.transformations.diffusion.unet.attention_block module neural_de.transformations.diffusion.unet.downsample module neural_de.transformations.diffusion.unet.nn module neural_de.transformations.diffusion.unet.qkv_attention module neural_de.transformations.diffusion.unet.qkv_attention_legacy module neural_de.transformations.diffusion.unet.res_block module neural_de.transformations.diffusion.unet.timestep_block module neural_de.transformations.diffusion.unet.timestep_embed_sequential module neural_de.transformations.diffusion.unet.unet_model module neural_de.transformations.diffusion.unet.upsample module neural_de.transformations.diffusion.unet.utils module neural_de.transformations.kernel_deblurring_enhancer module neural_de.transformations.night_image_enhancer module neural_de.transformations.resolution_enhancer module neural_de.transformations.transformation module neural_de.transformations.transformation_pipeline module neural_de.utils module neural_de.utils.math module neural_de.utils.model_manager module neural_de.utils.twe_logger module neural_de.utils.validation module ngf (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) NightConfig (class in neural_de.transformations.night_image_enhancer) NightImageEnhancer (class in neural_de.transformations.night_image_enhancer) normalization() (in module neural_de.transformations.diffusion.unet.nn) normValue() (in module neural_de.external.nplie.nplie) NPLIE() (in module neural_de.external.nplie.nplie) num_bottleneck_blocks (in module neural_de.external.maxim_tf.maxim.maxim) num_channels (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] (neural_de.transformations.diffusion.unet.nn.GroupNorm32 attribute) num_classes (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] num_groups (in module neural_de.external.maxim_tf.maxim.maxim) (neural_de.transformations.diffusion.unet.nn.GroupNorm32 attribute) num_head_channels (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] num_heads (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] num_heads_upsample (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] num_outputs (in module neural_de.external.maxim_tf.maxim.maxim) (neural_de.transformations.night_image_enhancer.NightConfig attribute) num_res_blocks (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] num_stages (in module neural_de.external.maxim_tf.maxim.maxim) num_supervision_scales (in module neural_de.external.maxim_tf.maxim.maxim) (neural_de.transformations.night_image_enhancer.NightConfig attribute)

O

optimizAlgo() (in module neural_de.external.nplie.nplie)

output_nc (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute)

P

padding_type (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) PReNet (class in neural_de.external.prenet.networks) PReNet_GRU (class in neural_de.external.prenet.networks) PReNet_LSTM (class in neural_de.external.prenet.networks)

PReNet_r (class in neural_de.external.prenet.networks) PReNet_x (class in neural_de.external.prenet.networks) PRN (class in neural_de.external.prenet.networks) PRN_r (class in neural_de.external.prenet.networks) psf2otf() (in module neural_de.external.nplie.nplie)

Q

QKVAttention (class in neural_de.transformations.diffusion.unet.qkv_attention)

QKVAttentionLegacy (class in neural_de.transformations.diffusion.unet.qkv_attention_legacy)

R

rand_t (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] RCAB() (in module neural_de.external.maxim_tf.maxim.blocks.attentions) RDCAB() (in module neural_de.external.maxim_tf.maxim.blocks.attentions) ResBlock (class in neural_de.transformations.diffusion.unet.res_block) resblock_updown (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] ResidualSplitHeadMultiAxisGmlpLayer() (in module neural_de.external.maxim_tf.maxim.blocks.misc_gating)

Resizing (class in neural_de.external.maxim_tf.maxim.layers) ResNetModified (class in neural_de.external.derain.blocks) ResolutionEnhancer (class in neural_de.transformations.resolution_enhancer) RevGuidedDiffusion (class in neural_de.transformations.diffusion.rev_guided_diffusion) RevVPSDE (class in neural_de.transformations.diffusion.rev_vpsde) rvpsde_fn() (neural_de.transformations.diffusion.rev_vpsde.RevVPSDE method)

S

SAM() (in module neural_de.external.maxim_tf.maxim.blocks.attentions) sample_step (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1]

shrink() (in module neural_de.external.nplie.nplie) sign() (in module neural_de.external.nplie.nplie) SwapAxes (class in neural_de.external.maxim_tf.maxim.layers)

T

t (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] t_delta (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] timestep_embedding() (in module neural_de.transformations.diffusion.unet.nn) TimestepBlock (class in neural_de.transformations.diffusion.unet.timestep_block) TimestepEmbedSequential (class in neural_de.transformations.diffusion.unet.timestep_embed_sequential) training (neural_de.external.derain.blocks.Conv2d attribute) (neural_de.external.derain.blocks.DecoderBlock attribute) (neural_de.external.derain.blocks.DeformableConv2d attribute) (neural_de.external.derain.blocks.DeformableResnetBlock attribute) (neural_de.external.derain.blocks.ResNetModified attribute) (neural_de.external.derain.blocks.UpConv2d attribute) (neural_de.external.prenet.networks.PReNet attribute) (neural_de.external.prenet.networks.PReNet_GRU attribute) (neural_de.external.prenet.networks.PReNet_LSTM attribute) (neural_de.external.prenet.networks.PReNet_r attribute) (neural_de.external.prenet.networks.PReNet_x attribute) (neural_de.external.prenet.networks.PRN attribute) (neural_de.external.prenet.networks.PRN_r attribute) (neural_de.transformations.diffusion.rev_guided_diffusion.RevGuidedDiffusion attribute) (neural_de.transformations.diffusion.rev_vpsde.RevVPSDE attribute) (neural_de.transformations.diffusion.unet.attention_block.AttentionBlock attribute) (neural_de.transformations.diffusion.unet.downsample.Downsample attribute) (neural_de.transformations.diffusion.unet.qkv_attention.QKVAttention attribute) (neural_de.transformations.diffusion.unet.qkv_attention_legacy.QKVAttentionLegacy attribute) (neural_de.transformations.diffusion.unet.res_block.ResBlock attribute) (neural_de.transformations.diffusion.unet.timestep_block.TimestepBlock attribute) (neural_de.transformations.diffusion.unet.unet_model.UNetModel attribute) (neural_de.transformations.diffusion.unet.upsample.Upsample attribute)

transform() (neural_de.transformations.brightness_enhancer.BrightnessEnhancer method) (neural_de.transformations.centered_zoom.CenteredZoom method) (neural_de.transformations.de_rain_enhancer.DeRainEnhancer method) (neural_de.transformations.de_snow_enhancer.DeSnowEnhancer method) (neural_de.transformations.diffusion.diffusion_enhancer.DiffusionEnhancer method) (neural_de.transformations.kernel_deblurring_enhancer.KernelDeblurringEnhancer method) (neural_de.transformations.night_image_enhancer.NightImageEnhancer method) (neural_de.transformations.resolution_enhancer.ResolutionEnhancer method) (neural_de.transformations.transformation_pipeline.TransformationPipeline method) transform_with_annotations() (neural_de.transformations.centered_zoom.CenteredZoom method) TransformationPipeline (class in neural_de.transformations.transformation_pipeline)

U

UnblockImages (class in neural_de.external.maxim_tf.maxim.layers) UNetDecoderBlock() (in module neural_de.external.maxim_tf.maxim.blocks.unet) UNetEncoderBlock() (in module neural_de.external.maxim_tf.maxim.blocks.unet) UNetModel (class in neural_de.transformations.diffusion.unet.unet_model) UpConv2d (class in neural_de.external.derain.blocks) updateI() (in module neural_de.external.nplie.nplie) updateK() (in module neural_de.external.nplie.nplie) updateLw() (in module neural_de.external.nplie.nplie) Upsample (class in neural_de.transformations.diffusion.unet.upsample) upsample_mode (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) UpSampleRatio() (in module neural_de.external.maxim_tf.maxim.blocks.others)

UPSCALE_MODEL (in module neural_de.transformations.resolution_enhancer) use_bias (in module neural_de.external.maxim_tf.maxim.maxim) (neural_de.transformations.night_image_enhancer.NightConfig attribute) use_bm (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] use_checkpoint (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] use_cross_gating (in module neural_de.external.maxim_tf.maxim.maxim) use_dropout (neural_de.transformations.de_rain_enhancer.DeRainConfig attribute) use_fp16 (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] use_global_mlp (in module neural_de.external.maxim_tf.maxim.maxim) use_new_attention_order (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1] use_scale_shift_norm (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1]

V

variant (neural_de.transformations.night_image_enhancer.NightConfig attribute)

vpsde_fn() (neural_de.transformations.diffusion.rev_vpsde.RevVPSDE method)

W

weights_path (neural_de.transformations.diffusion.diffpure_config.DiffPureConfig attribute), [1]

Z

zero_module() (in module neural_de.transformations.diffusion.unet.nn)