MCPcopy Create free account

hub / github.com/alexhe101/Pan-Mamba / functions

Functions464 in github.com/alexhe101/Pan-Mamba

Method__init__
( self, window_size: int, reduction: str = "none", max_val: fl
pan-sharpening/py-tra/utilsmetric.py:898
Method__init__
(self, kernel_size: int, border_type: str = 'reflect', normalized: bool = Tr
pan-sharpening/py-tra/utilsmetric.py:1061
Method__init__
(self, **kwargs)
pan-sharpening/py-tra/methods/PanNet.py:32
Method__init__
Simple block wrapping a mixer class with LayerNorm/RMSNorm and residual connection" This Block has a slightly different structure co
Mamba-block/mamba_simple.py:451
Method__init__
( self, d_model, d_state=16, d_conv=4, expand=2, dt_rank="auto
Mamba-block/mamba_simple.py:511
Method__init__
(self, normalized_shape)
Mamba-block/mamba_module.py:14
Method__init__
(self, normalized_shape)
Mamba-block/mamba_module.py:31
Method__init__
(self, dim)
Mamba-block/mamba_module.py:78
Method__init__
(self, dim)
Mamba-block/mamba_module.py:95
Method__init__
(self, dim)
Mamba-block/mamba_module.py:106
Method__len__
(self)
pan-sharpening/data/dataset.py:129
Method__len__
(self)
pan-sharpening/data/dataset.py:175
Method__len__
(self)
pan-sharpening/data/dataset.py:216
Method__len__
(self)
pan-sharpening/py-tra/utilsmetric.py:159
Method__len__
(self)
pan-sharpening/py-tra/utilsmetric.py:177
Method__len__
(self)
pan-sharpening/py-tra/utilsmetric.py:194
Method__len__
(self)
pan-sharpening/py-tra/utilsmetric.py:213
Method__len__
(self)
pan-sharpening/py-tra/utilsmetric.py:241
Method__repr__
(self)
pan-sharpening/py-tra/utilsmetric.py:691
Method__repr__
(self)
pan-sharpening/py-tra/utilsmetric.py:1073
Method_check_input_dim
(self, input)
pan-sharpening/model/modules.py:86
Method_forward
(x)
pan-sharpening/utils/vgg.py:32
Method_wgan_loss
wgan loss. Args: input (Tensor): Input tensor. target (bool): Target label. Returns: Tensor: wga
pan-sharpening/utils/utils.py:548
Method_wgan_softplus_loss
wgan loss with soft plus. softplus is a smooth approximation to the ReLU function. In StyleGAN2, it is called: Logistic l
pan-sharpening/utils/utils.py:560
Methodallocate_inference_cache
(self, batch_size, max_seqlen, dtype=None, **kwargs)
Mamba-block/mamba_simple.py:505
Methodallocate_inference_cache
(self, batch_size, max_seqlen, dtype=None, **kwargs)
Mamba-block/mamba_simple.py:884
Functionbox_blur
r"""Blurs an image using the box filter. See :class:`~kornia.filters.BoxBlur` for details.
pan-sharpening/py-tra/utilsmetric.py:701
Functioncalculate_ds
(pred,p,lrms)
pan-sharpening/utils/utils.py:123
Functioncalculate_psnr
(img1, img2, pixel_range=255, color_mode='rgb')
pan-sharpening/utils/utils.py:154
Functioncalculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]
pan-sharpening/utils/utils.py:216
Methodcall
(self, inputs)
pan-sharpening/py-tra/methods/PanNet.py:35
Methodcall
(self, inputs)
pan-sharpening/py-tra/methods/PanNet.py:68
Functioncat_feature
(tensor_a, tensor_b)
pan-sharpening/model/thops.py:58
Methodcompute_output_shape
(self, input_shape)
pan-sharpening/py-tra/methods/PanNet.py:55
Methodcompute_output_shape
(self, input_shape)
pan-sharpening/py-tra/methods/PanNet.py:72
Functioncreate_grid
(resolution)
pan-sharpening/model/utils.py:50
Functiondefault_conv
(in_channels, out_channels, kernel_size,stride=1, bias=True)
pan-sharpening/model/CDC.py:184
Functiondwt_init
(x)
pan-sharpening/solver/unisolver.py:13
Functionfeature_save
(tensor,name,i=0)
pan-sharpening/solver/solver.py:71
Functionfeature_save
(tensor,name,i=0)
pan-sharpening/solver/unisolver.py:85
Functionfeature_save
(tensor,name,i=0)
pan-sharpening/solver/testsolver.py:20
Methodforward
(self, input, label)
pan-sharpening/utils/SAM_loss.py:21
Methodforward
(self, pred, target, **kwargs)
pan-sharpening/utils/SAM_loss.py:38
Methodforward
(self, input,target)
pan-sharpening/utils/utils.py:27
Methodforward
(self, pred, target,p,lrms)
pan-sharpening/utils/utils.py:106
Methodforward
Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
pan-sharpening/utils/utils.py:487
Methodforward
(self, pred, target)
pan-sharpening/utils/utils.py:501
Methodforward
Args: input (Tensor): The input for the loss module, i.e., the network prediction. target_is_real (bo
pan-sharpening/utils/utils.py:594
Methodforward
(self, sr, hr)
pan-sharpening/utils/vgg.py:31
Methodforward
(self, img1, img2)
pan-sharpening/utils/loss_util.py:192
Methodforward
(self, pred, target)
pan-sharpening/utils/loss_util.py:276
Methodforward
(self, pred, img_lrms)
pan-sharpening/utils/loss_util.py:292
Methodforward
(self, x)
pan-sharpening/model/CDC.py:28
Methodforward
(self, x)
pan-sharpening/model/CDC.py:61
Methodforward
(self, x)
pan-sharpening/model/CDC.py:85
Methodforward
(self, x)
pan-sharpening/model/CDC.py:101
Methodforward
(self, x)
pan-sharpening/model/CDC.py:126
Methodforward
(self, x)
pan-sharpening/model/CDC.py:159
Methodforward
(self, x)
pan-sharpening/model/base_net.py:48
Methodforward
(self, x)
pan-sharpening/model/base_net.py:102
Methodforward
(self, x)
pan-sharpening/model/base_net.py:127
Methodforward
(self, x)
pan-sharpening/model/base_net.py:191
Methodforward
(self, x)
pan-sharpening/model/base_net.py:225
Methodforward
(self, x)
pan-sharpening/model/spatial_shift.py:17
Methodforward
(self, x)
pan-sharpening/model/spatial_shift.py:40
Methodforward
(self, x_all)
pan-sharpening/model/spatial_shift.py:86
Methodforward
(self, x)
pan-sharpening/model/spatial_shift.py:139
Methodforward
(self, x)
pan-sharpening/model/spatial_shift.py:171
Methodforward
(self, x)
pan-sharpening/model/spatial_shift.py:273
Methodforward
(self, x)
pan-sharpening/model/mamba_module.py:25
Methodforward
(self, x)
pan-sharpening/model/mamba_module.py:43
Methodforward
(self, x)
pan-sharpening/model/mamba_module.py:57
Methodforward
(self,ms,ms_resi,pan)
pan-sharpening/model/mamba_module.py:84
Methodforward
(self,ipt)
pan-sharpening/model/mamba_module.py:100
Methodforward
(self, ms,pan ,ms_residual,pan_residual)
pan-sharpening/model/mamba_module.py:112
Methodforward
(self, input, logdet=None, reverse=False)
pan-sharpening/model/modules.py:66
Methodforward
(self, input)
pan-sharpening/model/modules.py:104
Methodforward
(self, input)
pan-sharpening/model/modules.py:164
Methodforward
(self, input, reverse=False)
pan-sharpening/model/modules.py:185
Methodforward
log-det = log|abs(|W|)| * pixels
pan-sharpening/model/modules.py:247
Methodforward
(self, input, logdet=0., reverse=False, eps_std=None)
pan-sharpening/model/modules.py:298
Methodforward
(self, input, logdet=None, reverse=False)
pan-sharpening/model/modules.py:350
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:24
Methodforward
(self, ms, pan)
pan-sharpening/model/panmamba_baseline_finalversion.py:42
Methodforward
(self, ms,pan)
pan-sharpening/model/panmamba_baseline_finalversion.py:78
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:96
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:114
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:128
Methodforward
(self, x,x_size)
pan-sharpening/model/panmamba_baseline_finalversion.py:151
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:167
Methodforward
(self,ipt)
pan-sharpening/model/panmamba_baseline_finalversion.py:183
Methodforward
(self, ms,pan ,ms_residual,pan_residual)
pan-sharpening/model/panmamba_baseline_finalversion.py:195
Methodforward
(self,ms,ms_resi,pan)
pan-sharpening/model/panmamba_baseline_finalversion.py:218
Methodforward
(self, x)
pan-sharpening/model/panmamba_baseline_finalversion.py:240
Methodforward
(self,ms,_,pan)
pan-sharpening/model/panmamba_baseline_finalversion.py:272
Methodforward
(self, x)
pan-sharpening/model/refine.py:22
Methodforward
(self, x)
pan-sharpening/model/refine.py:48
Methodforward
(self, x)
pan-sharpening/model/refine.py:71
Methodforward
(self, x)
pan-sharpening/model/refine.py:91
Methodforward
(self, x)
pan-sharpening/model/refine.py:130
← previousnext →301–400 of 464, ranked by callers