Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Lizhe1228/MambaDFuse
/ types & classes
Types & classes
81 in github.com/Lizhe1228/MambaDFuse
⨍
Functions
470
◇
Types & classes
81
↓ 6 callers
Class
LayerNorm
models/network.py:112
↓ 5 callers
Class
M3
models/network.py:162
↓ 4 callers
Class
Mamba
mamba_simple.py:34
↓ 3 callers
Class
ChannelAttention
models/refine.py:30
↓ 3 callers
Class
ResidualDenseBlock_5C
models/basicblock.py:393
↓ 3 callers
Class
SingleMambaBlock
models/network.py:127
↓ 2 callers
Class
FourierShift
models/refine.py:98
↓ 2 callers
Class
RubikCube_multiply
models/refine.py:148
↓ 2 callers
Class
SSIMLoss
models/loss_ssim.py:65
↓ 2 callers
Class
Sobelxy
models/loss_med.py:56
↓ 2 callers
Class
Sobelxy
models/loss_vif.py:55
↓ 2 callers
Class
TokenSwapMamba
models/network.py:139
↓ 1 callers
Class
BiasFree_LayerNorm
models/network.py:58
↓ 1 callers
Class
CALayer
models/basicblock.py:333
↓ 1 callers
Class
CALayer
models/refine.py:53
↓ 1 callers
Class
ChannelExchange
models/network.py:74
↓ 1 callers
Class
CharbonnierLoss
Charbonnier Loss (L1)
models/loss.py:165
↓ 1 callers
Class
ESA
models/basicblock.py:271
↓ 1 callers
Class
L_Grad
models/loss_gt.py:40
↓ 1 callers
Class
L_Grad
models/loss_med.py:38
↓ 1 callers
Class
L_Grad
models/loss_vif.py:30
↓ 1 callers
Class
L_Intensity
models/loss_gt.py:73
↓ 1 callers
Class
L_Intensity
models/loss_med.py:74
↓ 1 callers
Class
L_Intensity
models/loss_vif.py:73
↓ 1 callers
Class
L_SSIM
models/loss_med.py:29
↓ 1 callers
Class
L_SSIM
models/loss_vif.py:43
↓ 1 callers
Class
L_TV
models/loss_gt.py:27
↓ 1 callers
Class
MambaDFuse
models/network.py:307
↓ 1 callers
Class
Net
utils/utils_model.py:305
↓ 1 callers
Class
NoneDict
utils/utils_mat.py:53
↓ 1 callers
Class
NoneDict
utils/utils_option.py:242
↓ 1 callers
Class
PatchEmbed
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 callers
Class
PatchUnEmbed
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 callers
Class
RCABlock
models/basicblock.py:354
↓ 1 callers
Class
Sobelxy
models/loss_gt.py:56
↓ 1 callers
Class
VGGFeatureExtractor
models/loss.py:12
↓ 1 callers
Class
VGGFeatureExtractor
models/network_feature.py:16
↓ 1 callers
Class
WithBias_LayerNorm
models/network.py:94
↓ 1 callers
Class
fusion_loss_gt
models/loss_gt.py:176
↓ 1 callers
Class
fusion_loss_med
models/loss_med.py:83
↓ 1 callers
Class
fusion_loss_vif
models/loss_vif.py:83
Class
Block
mamba_simple.py:536
Class
CFRB
models/basicblock.py:298
Class
ConcatBlock
models/basicblock.py:175
Class
ConditionalBatchNorm2d
models/basicblock.py:156
Class
DWMamba
mamba_simple.py:596
Class
Dataset
# ----------------------------------------- # Get L/H for denosing on AWGN with fixed sigma. # Only dataroot_H is needed. # ---------
data/dataloder.py:9
Class
Dataset
# ----------------------------------------- # Get L/H for denosing on AWGN with fixed sigma. # Only dataroot_H is needed. # ---------
data/dataset_wogt.py:9
Class
DenseModule
models/refine.py:13
Class
FusedLeakyReLU
models/op/fused_act.py:75
Class
FusedLeakyReLUFunction
models/op/fused_act.py:53
Class
FusedLeakyReLUFunctionBackward
models/op/fused_act.py:21
Class
GANLoss
models/loss.py:93
Class
IMDBlock
@inproceedings{hui2019lightweight, title={Lightweight Image Super-Resolution with Information Multi-distillation Network}, author={Hu
models/basicblock.py:230
Class
L_color
models/loss_gt.py:10
Class
L_color
models/loss_med.py:11
Class
L_color
models/loss_vif.py:13
Class
L_exp
models/loss_gt.py:126
Class
L_spa
models/loss_gt.py:82
Class
ModelBase
models/model_base.py:8
Class
ModelPlain
Train with pixel loss
models/model_plain.py:19
Class
NonLocalBlock2D
models/basicblock.py:543
Class
PerceptualLoss
VGG Perceptual loss
models/loss.py:57
Class
PixelUnShuffle
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
Class
RCAGroup
models/basicblock.py:373
Class
RGB_HSV
utils/utils_color.py:9
Class
RGB_YCbCr
utils/utils_color.py:89
Class
RRDB
models/basicblock.py:416
Class
Refine
models/refine.py:79
Class
Refine1
models/refine.py:309
Class
RefineRubik
models/refine.py:271
Class
RefineShift
models/refine.py:289
Class
ResBlock
models/basicblock.py:211
Class
ShortcutBlock
models/basicblock.py:191
Class
TVLoss
models/loss.py:136
Class
UpFirDn2d
models/op/upfirdn2d.py:88
Class
UpFirDn2dBackward
models/op/upfirdn2d.py:19
Class
Upsample
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
Class
UpsampleOneStep
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
Class
logger_print
utils/utils_logger.py:56
Class
perception_loss
models/loss_gt.py:142