Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/cidautai/DarkIR
/ types & classes
Types & classes
37 in github.com/cidautai/DarkIR
⨍
Functions
170
◇
Types & classes
37
↓ 14 callers
Class
MyDataset_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 callers
Class
CustomSequential
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 callers
Class
IGAB
archs/retinexformer.py:203
↓ 4 callers
Class
LayerNorm2d
archs/arch_util.py:35
↓ 3 callers
Class
EBlock
Change this block using Branch
archs/arch_model.py:141
↓ 3 callers
Class
GELU
archs/retinexformer.py:71
↓ 3 callers
Class
SimpleGate
archs/arch_model.py:12
↓ 2 callers
Class
Branch
Branch that lasts lonly the dilated convolutions
archs/arch_model.py:57
↓ 2 callers
Class
DBlock
Change this block using Branch
archs/arch_model.py:72
↓ 2 callers
Class
DarkIR
archs/DarkIR.py:11
↓ 2 callers
Class
VGGLoss
losses/loss.py:351
↓ 1 callers
Class
CharbonnierLoss
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 callers
Class
CropTo4
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 callers
Class
Denoiser
archs/retinexformer.py:233
↓ 1 callers
Class
EdgeLoss
losses/loss.py:379
↓ 1 callers
Class
EnhanceLoss
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 callers
Class
FeedForward
archs/retinexformer.py:182
↓ 1 callers
Class
FreMLP
archs/arch_model.py:36
↓ 1 callers
Class
FrequencyLoss
Calculates the amplitude of frequencies loss.
losses/loss.py:117
↓ 1 callers
Class
IG_MSA
archs/retinexformer.py:125
↓ 1 callers
Class
Illumination_Estimator
archs/retinexformer.py:95
↓ 1 callers
Class
L1Loss
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 callers
Class
MSELoss
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 callers
Class
PreNorm
archs/retinexformer.py:60
↓ 1 callers
Class
RandomCropSame
A function that random crops a pair of images with a fixed size (and the same crop).
data/dataset_reader/datapipeline.py:54
↓ 1 callers
Class
RetinexFormer
archs/retinexformer.py:343
↓ 1 callers
Class
RetinexFormer_Single_Stage
archs/retinexformer.py:324
↓ 1 callers
Class
SSIM
losses/loss.py:424
↓ 1 callers
Class
VGG19
losses/loss.py:318
Class
Adapter
archs/arch_model.py:17
Class
L1_Charbonnier_loss
L1 Charbonnierloss.
losses/loss.py:452
Class
L_deblur
L_deblur.
losses/loss.py:464
Class
L_enhance
L_enhance.
losses/loss.py:478
Class
L_reblur
L_enhance.
losses/loss.py:491
Class
LayerNormFunction
archs/arch_util.py:8
Class
PSNRLoss
losses/loss.py:41
Class
SSIMloss
losses/loss.py:433