MCPcopy Create free account

hub / github.com/Lizhe1228/MambaDFuse / types & classes

Types & classes81 in github.com/Lizhe1228/MambaDFuse

↓ 6 callersClassLayerNorm
models/network.py:112
↓ 5 callersClassM3
models/network.py:162
↓ 4 callersClassMamba
mamba_simple.py:34
↓ 3 callersClassChannelAttention
models/refine.py:30
↓ 3 callersClassResidualDenseBlock_5C
models/basicblock.py:393
↓ 3 callersClassSingleMambaBlock
models/network.py:127
↓ 2 callersClassFourierShift
models/refine.py:98
↓ 2 callersClassRubikCube_multiply
models/refine.py:148
↓ 2 callersClassSSIMLoss
models/loss_ssim.py:65
↓ 2 callersClassSobelxy
models/loss_med.py:56
↓ 2 callersClassSobelxy
models/loss_vif.py:55
↓ 2 callersClassTokenSwapMamba
models/network.py:139
↓ 1 callersClassBiasFree_LayerNorm
models/network.py:58
↓ 1 callersClassCALayer
models/basicblock.py:333
↓ 1 callersClassCALayer
models/refine.py:53
↓ 1 callersClassChannelExchange
models/network.py:74
↓ 1 callersClassCharbonnierLoss
Charbonnier Loss (L1)
models/loss.py:165
↓ 1 callersClassESA
models/basicblock.py:271
↓ 1 callersClassL_Grad
models/loss_gt.py:40
↓ 1 callersClassL_Grad
models/loss_med.py:38
↓ 1 callersClassL_Grad
models/loss_vif.py:30
↓ 1 callersClassL_Intensity
models/loss_gt.py:73
↓ 1 callersClassL_Intensity
models/loss_med.py:74
↓ 1 callersClassL_Intensity
models/loss_vif.py:73
↓ 1 callersClassL_SSIM
models/loss_med.py:29
↓ 1 callersClassL_SSIM
models/loss_vif.py:43
↓ 1 callersClassL_TV
models/loss_gt.py:27
↓ 1 callersClassMambaDFuse
models/network.py:307
↓ 1 callersClassNet
utils/utils_model.py:305
↓ 1 callersClassNoneDict
utils/utils_mat.py:53
↓ 1 callersClassNoneDict
utils/utils_option.py:242
↓ 1 callersClassPatchEmbed
r""" Image to Patch Embedding Args: img_size (int): Image size. Default: 224. patch_size (int): Patch token size. Default: 4.
models/network.py:184
↓ 1 callersClassPatchUnEmbed
r""" Image to Patch Unembedding Args: img_size (int): Image size. Default: 224. patch_size (int): Patch token size. Default: 4.
models/network.py:227
↓ 1 callersClassRCABlock
models/basicblock.py:354
↓ 1 callersClassSobelxy
models/loss_gt.py:56
↓ 1 callersClassVGGFeatureExtractor
models/loss.py:12
↓ 1 callersClassVGGFeatureExtractor
models/network_feature.py:16
↓ 1 callersClassWithBias_LayerNorm
models/network.py:94
↓ 1 callersClassfusion_loss_gt
models/loss_gt.py:176
↓ 1 callersClassfusion_loss_med
models/loss_med.py:83
↓ 1 callersClassfusion_loss_vif
models/loss_vif.py:83
ClassBlock
mamba_simple.py:536
ClassCFRB
models/basicblock.py:298
ClassConcatBlock
models/basicblock.py:175
ClassConditionalBatchNorm2d
models/basicblock.py:156
ClassDWMamba
mamba_simple.py:596
ClassDataset
# ----------------------------------------- # Get L/H for denosing on AWGN with fixed sigma. # Only dataroot_H is needed. # ---------
data/dataloder.py:9
ClassDataset
# ----------------------------------------- # Get L/H for denosing on AWGN with fixed sigma. # Only dataroot_H is needed. # ---------
data/dataset_wogt.py:9
ClassDenseModule
models/refine.py:13
ClassFusedLeakyReLU
models/op/fused_act.py:75
ClassFusedLeakyReLUFunction
models/op/fused_act.py:53
ClassFusedLeakyReLUFunctionBackward
models/op/fused_act.py:21
ClassGANLoss
models/loss.py:93
ClassIMDBlock
@inproceedings{hui2019lightweight, title={Lightweight Image Super-Resolution with Information Multi-distillation Network}, author={Hu
models/basicblock.py:230
ClassL_color
models/loss_gt.py:10
ClassL_color
models/loss_med.py:11
ClassL_color
models/loss_vif.py:13
ClassL_exp
models/loss_gt.py:126
ClassL_spa
models/loss_gt.py:82
ClassModelBase
models/model_base.py:8
ClassModelPlain
Train with pixel loss
models/model_plain.py:19
ClassNonLocalBlock2D
models/basicblock.py:543
ClassPerceptualLoss
VGG Perceptual loss
models/loss.py:57
ClassPixelUnShuffle
r"""Rearranges elements in a Tensor of shape :math:`(C, rH, rW)` to a tensor of shape :math:`(*, r^2C, H, W)`. Authors: Zhaoyi Yan, h
models/basicblock.py:129
ClassRCAGroup
models/basicblock.py:373
ClassRGB_HSV
utils/utils_color.py:9
ClassRGB_YCbCr
utils/utils_color.py:89
ClassRRDB
models/basicblock.py:416
ClassRefine
models/refine.py:79
ClassRefine1
models/refine.py:309
ClassRefineRubik
models/refine.py:271
ClassRefineShift
models/refine.py:289
ClassResBlock
models/basicblock.py:211
ClassShortcutBlock
models/basicblock.py:191
ClassTVLoss
models/loss.py:136
ClassUpFirDn2d
models/op/upfirdn2d.py:88
ClassUpFirDn2dBackward
models/op/upfirdn2d.py:19
ClassUpsample
Upsample module. Args: scale (int): Scale factor. Supported scales: 2^n and 3. num_feat (int): Channel number of intermediate fea
models/network.py:261
ClassUpsampleOneStep
UpsampleOneStep module (the difference with Upsample is that it always only has 1conv + 1pixelshuffle) Used in lightweight SR to save parameter
models/network.py:283
Classlogger_print
utils/utils_logger.py:56
Classperception_loss
models/loss_gt.py:142