MCPcopy Create free account

hub / github.com/cidautai/DarkIR / types & classes

Types & classes37 in github.com/cidautai/DarkIR

↓ 14 callersClassMyDataset_Crop
A Dataset of the low and high light images with data values in each channel in the range 0-1 (normalized).
data/dataset_reader/datapipeline.py:96
↓ 4 callersClassCustomSequential
Similar to nn.Sequential, but it lets us introduce a second argument in the forward method so adaptors can be considered in the inference.
archs/arch_util.py:47
↓ 4 callersClassIGAB
archs/retinexformer.py:203
↓ 4 callersClassLayerNorm2d
archs/arch_util.py:35
↓ 3 callersClassEBlock
Change this block using Branch
archs/arch_model.py:141
↓ 3 callersClassGELU
archs/retinexformer.py:71
↓ 3 callersClassSimpleGate
archs/arch_model.py:12
↓ 2 callersClassBranch
Branch that lasts lonly the dilated convolutions
archs/arch_model.py:57
↓ 2 callersClassDBlock
Change this block using Branch
archs/arch_model.py:72
↓ 2 callersClassDarkIR
archs/DarkIR.py:11
↓ 2 callersClassVGGLoss
losses/loss.py:351
↓ 1 callersClassCharbonnierLoss
Charbonnier loss (one variant of Robust L1Loss, a differentiable variant of L1Loss). Described in "Deep Laplacian Pyramid Networks for Fast a
losses/loss.py:155
↓ 1 callersClassCropTo4
A function that crops an image into 4 patches (top-left, top-right, bottom-left, bottom-right), adn returns a list of the patches.
data/dataset_reader/datapipeline.py:23
↓ 1 callersClassDenoiser
archs/retinexformer.py:233
↓ 1 callersClassEdgeLoss
losses/loss.py:379
↓ 1 callersClassEnhanceLoss
Applies the enhanceLoss. This loss is the l1 loss of the image downsampled at the middle of the encoder-decoder plus the l1 of the features o
losses/loss.py:502
↓ 1 callersClassFeedForward
archs/retinexformer.py:182
↓ 1 callersClassFreMLP
archs/arch_model.py:36
↓ 1 callersClassFrequencyLoss
Calculates the amplitude of frequencies loss.
losses/loss.py:117
↓ 1 callersClassIG_MSA
archs/retinexformer.py:125
↓ 1 callersClassIllumination_Estimator
archs/retinexformer.py:95
↓ 1 callersClassL1Loss
L1 (mean absolute error, MAE) loss. Args: loss_weight (float): Loss weight for L1 loss. Default: 1.0. reduction (str): Specifies
losses/loss.py:57
↓ 1 callersClassMSELoss
MSE (L2) loss. Args: loss_weight (float): Loss weight for MSE loss. Default: 1.0. reduction (str): Specifies the reduction to app
losses/loss.py:87
↓ 1 callersClassPreNorm
archs/retinexformer.py:60
↓ 1 callersClassRandomCropSame
A function that random crops a pair of images with a fixed size (and the same crop).
data/dataset_reader/datapipeline.py:54
↓ 1 callersClassRetinexFormer
archs/retinexformer.py:343
↓ 1 callersClassRetinexFormer_Single_Stage
archs/retinexformer.py:324
↓ 1 callersClassSSIM
losses/loss.py:424
↓ 1 callersClassVGG19
losses/loss.py:318
ClassAdapter
archs/arch_model.py:17
ClassL1_Charbonnier_loss
L1 Charbonnierloss.
losses/loss.py:452
ClassL_deblur
L_deblur.
losses/loss.py:464
ClassL_enhance
L_enhance.
losses/loss.py:478
ClassL_reblur
L_enhance.
losses/loss.py:491
ClassLayerNormFunction
archs/arch_util.py:8
ClassPSNRLoss
losses/loss.py:41
ClassSSIMloss
losses/loss.py:433