MCPcopy Create free account

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

Functions464 in github.com/alexhe101/Pan-Mamba

⨍ Functions 464◇ Types & classes 108
↓ 42 callersMethodsave
docstringls
pan-sharpening/tool/modcrop.py:29
↓ 27 callersFunctionno_ref_evaluate
(pred, pan, hs)
pan-sharpening/py-tra/metrics.py:358
↓ 26 callersFunctionref_evaluate
(pred, gt)
pan-sharpening/py-tra/metrics.py:346
↓ 16 callersFunctionupsample_interp23
(image, ratio)
pan-sharpening/py-tra/utils.py:28
↓ 13 callersMethod__init__
(self,num_channels=None,base_filter=None,args=None)
pan-sharpening/model/panmamba_baseline_finalversion.py:247
↓ 12 callersFunctioncal
(ref, noref)
pan-sharpening/py-tra/demo_all_methods_o.py:56
↓ 12 callersFunctioncal
(ref, noref)
pan-sharpening/py-tra/demo_all_methods.py:56
↓ 12 callersFunctionsave_config
(time, log)
pan-sharpening/utils/utils.py:144
↓ 11 callersMethod__init__
(self, lap_weight=1, angle_weight=1)
pan-sharpening/py-tra/utilsmetric.py:375
↓ 8 callersMethod__init__
(self, in_channels, out_channels, kernel_size=[3, 3], stride=[1, 1], padding
pan-sharpening/model/modules.py:130
↓ 8 callersMethod__init__
(self, n_feat, out_channel)
pan-sharpening/model/refine.py:81
↓ 8 callersFunctiongaussian_filter2d
2D Gaussian filter USAGE h = gaussian_filter2d(shape,sigma) INPUT shape : window size (e.g., (3,3)) sigma : sca
pan-sharpening/py-tra/methods/CNMF.py:546
↓ 7 callersFunctionboxfilter
(img, r)
pan-sharpening/py-tra/methods/GFPCA.py:17
↓ 7 callersFunctioncal
(ref, noref)
pan-sharpening/py-tra/tra_full.py:56
↓ 7 callersFunctionfilter2D
r"""Function that convolves a tensor with a kernel. The function applies a given kernel to a tensor. The kernel is applied independently at e
pan-sharpening/py-tra/utilsmetric.py:566
↓ 6 callersFunction_qindex
Q-index for 2D (one-band) image, shape (H, W); uint or float [0, 1]
pan-sharpening/py-tra/metrics.py:59
↓ 6 callersFunctionis_image_file
(filename)
pan-sharpening/data/dataset.py:18
↓ 6 callersFunctionload_img
(filepath)
pan-sharpening/data/dataset.py:22
↓ 5 callersMethod__init__
(self, mid_ch, theta=0.7)
pan-sharpening/model/CDC.py:13
↓ 5 callersMethod__init__
(self, scale, n_feat, bn=False, activation='prelu', bias=True)
pan-sharpening/model/base_net.py:19
↓ 5 callersMethod__init__
(self, d_model, depth, expansion_factor = 4, dropout = 0.)
pan-sharpening/model/spatial_shift.py:155
↓ 5 callersMethod__init__
(self, dim, LayerNorm_type)
pan-sharpening/model/mamba_module.py:50
↓ 5 callersMethod__init__
(self, dim, LayerNorm_type)
Mamba-block/mamba_module.py:50
↓ 5 callersFunctionget_config
(cfg_path)
pan-sharpening/utils/config.py:12
↓ 5 callersFunctionimresize
(img, size=None, scale_factor=None)
pan-sharpening/py-tra/utilsmetric.py:74
↓ 5 callersMethodstep
(self, hidden_states, conv_state, ssm_state)
Mamba-block/mamba_simple.py:361
↓ 4 callersFunctionIm2Patch
(img, win, stride=1)
pan-sharpening/py-tra/utilsmetric.py:59
↓ 4 callersMethod__init__
(self)
pan-sharpening/utils/utils.py:104
↓ 4 callersFunction_qindex
Q-index for 2D (one-band) image, shape (H, W); uint or float [0, 1]
pan-sharpening/utils/utils.py:307
↓ 4 callersFunctioncalculate_Q
(x, y)
pan-sharpening/utils/utils.py:77
↓ 4 callersFunctionget_path
(subdir)
pan-sharpening/utils/utils.py:141
↓ 4 callersMethodimread
(self, path)
pan-sharpening/tool/pre_processing.py:70
↓ 4 callersFunctionmake_loss
(loss_type)
pan-sharpening/utils/utils.py:46
↓ 4 callersMethodsave_img
(self, img, img_name, mode)
pan-sharpening/solver/testsolver.py:146
↓ 4 callersFunctionsum
(tensor, dim=None, keepdim=False)
pan-sharpening/model/thops.py:15
↓ 4 callersFunctionzoom_nn
Zoom via nearest neighbor interpolation
pan-sharpening/py-tra/methods/CNMF.py:644
↓ 3 callersFunctionBicubic
(pan, hs)
pan-sharpening/py-tra/methods/Bicubic.py:11
↓ 3 callersFunctionBrovey
(pan, hs)
pan-sharpening/py-tra/methods/Brovey.py:18
↓ 3 callersFunctionGFPCA
(pan, hs)
pan-sharpening/py-tra/methods/GFPCA.py:57
↓ 3 callersFunctionGS
(pan, hs)
pan-sharpening/py-tra/methods/GS.py:18
↓ 3 callersFunctionIHS
(pan, hs)
pan-sharpening/py-tra/methods/IHS.py:20
↓ 3 callersFunctionPCA
(pan, hs)
pan-sharpening/py-tra/methods/PCA.py:17
↓ 3 callersFunctionSFIM
(pan, hs)
pan-sharpening/py-tra/methods/SFIM.py:17
↓ 3 callersFunctionaugment
(ms_image, lms_image, pan_image, bms_image, flip_h=True, rot=True)
pan-sharpening/data/dataset.py:57
↓ 3 callersFunctioncreate_window
(window_size, channel)
pan-sharpening/utils/loss_util.py:156
↓ 3 callersFunctiongate_loss
(gate)
pan-sharpening/solver/solver.py:108
↓ 3 callersFunctiongaussian_down_sample
This function downsamples HS image with a Gaussian point spread function. USAGE HSI = gaussian_down_sample(data,w,mask) INPUT
pan-sharpening/py-tra/methods/CNMF.py:569
↓ 3 callersFunctionnormalize_kernel2d
r"""Normalizes both derivative and smoothing kernel.
pan-sharpening/py-tra/utilsmetric.py:629
↓ 3 callersFunctionrescale_img
(img_in, scale)
pan-sharpening/data/dataset.py:26
↓ 3 callersFunctionssim
(img1, img2, pixel_range=255, color_mode='rgb')
pan-sharpening/utils/utils.py:183
↓ 3 callersFunctiontransform
()
pan-sharpening/data/data.py:16
↓ 2 callersFunctionCNMF
COUPLED NONNEGATIVE MATRIX FACTORIZATION (CNMF) Copyright (c) 2016 Naoto Yokoya Email: yokoya@sal.rcast.u-tokyo.ac.jp Update: 2016/0
pan-sharpening/py-tra/methods/CNMF.py:17
↓ 2 callersFunctionD_lambda
Spectral distortion img_fake, generated HRMS img_lm, LRMS
pan-sharpening/utils/utils.py:280
↓ 2 callersFunctionD_lambda
Spectral distortion img_fake, generated HRMS img_lm, LRMS
pan-sharpening/py-tra/metrics.py:270
↓ 2 callersFunctionD_s
Spatial distortion img_fake, generated HRMS img_lm, LRMS pan, HRPan
pan-sharpening/utils/utils.py:362
↓ 2 callersFunctionD_s
Spatial distortion img_fake, generated HRMS img_lm, LRMS pan, HRPan
pan-sharpening/py-tra/metrics.py:297
↓ 2 callersFunctionGNyq2win
Generate a 2D convolutional window from a given GNyq GNyq: Nyquist frequency scale: spatial size of PAN / spatial size of MS
pan-sharpening/utils/utils.py:427
↓ 2 callersFunctionGNyq2win
Generate a 2D convolutional window from a given GNyq GNyq: Nyquist frequency scale: spatial size of PAN / spatial size of MS
pan-sharpening/py-tra/metrics.py:222
↓ 2 callersFunctionGSA
(pan, hs)
pan-sharpening/py-tra/methods/GSA.py:45
↓ 2 callersFunctionMTF_GLP
(pan, hs, sensor='gaussian')
pan-sharpening/py-tra/methods/MTF_GLP.py:57
↓ 2 callersFunctionMTF_GLP_HPM
(pan, hs, sensor='gaussian')
pan-sharpening/py-tra/methods/MTF_GLP_HPM.py:57
↓ 2 callersFunctionWavelet
(pan, hs)
pan-sharpening/py-tra/methods/Wavelet.py:16
↓ 2 callersMethod__init__
(self, window_size = 11, size_average = True)
pan-sharpening/utils/loss_util.py:185
↓ 2 callersMethod__init__
(self, data_dir_ms, data_dir_pan, cfg, transform=None, data_dir_mask=None)
pan-sharpening/data/dataset.py:85
↓ 2 callersMethod__init__
( self, d_model, d_state=16, d_conv=4, expand=2, dt_rank="auto
Mamba-block/mamba_simple.py:35
↓ 2 callersMethod_center
(self, input, reverse=False)
pan-sharpening/model/modules.py:43
↓ 2 callersMethod_check_input_dim
(self, input)
pan-sharpening/model/modules.py:27
↓ 2 callersFunction_qindex
Q-index for 2D (one-band) image, shape (H, W); uint or float [0, 1]
pan-sharpening/utils/loss_util.py:225
↓ 2 callersMethod_scale
(self, input, logdet=None, reverse=False)
pan-sharpening/model/modules.py:49
↓ 2 callersFunction_ssim
SSIM for 2D (one-band) image, shape (H, W); uint8 if 225; uint16 if 2047
pan-sharpening/utils/utils.py:237
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average = True)
pan-sharpening/utils/loss_util.py:162
↓ 2 callersFunction_ssim
SSIM for 2D (one-band) image, shape (H, W); uint8 if 225; uint16 if 2047
pan-sharpening/py-tra/metrics.py:128
↓ 2 callersFunctioncpsnr
PSNR metric, img uint8 if 225; uint16 if 2047
pan-sharpening/utils/utils.py:172
↓ 2 callersFunctioncssim
SSIM for 2D (H, W) or 3D (H, W, C) image; uint8 if 225; uint16 if 2047
pan-sharpening/utils/utils.py:261
↓ 2 callersFunctiondowngrade_images
downgrade MS and PAN by a ratio factor with given sensor's gains
pan-sharpening/py-tra/utils.py:133
↓ 2 callersMethodeval
(self)
pan-sharpening/solver/solver.py:260
↓ 2 callersMethodeval
(self)
pan-sharpening/solver/testsolver.py:119
↓ 2 callersFunctionfir_filter_wind
compute fir filter with window method Hd: desired freqeuncy response (2D) w: window (2D)
pan-sharpening/py-tra/utils.py:118
↓ 2 callersMethodforward
(self, input)
pan-sharpening/model/modules.py:144
↓ 2 callersFunctiongaussian2d
(N, std)
pan-sharpening/py-tra/utils.py:98
↓ 2 callersFunctionget_data
(cfg, mode)
pan-sharpening/data/data.py:21
↓ 2 callersFunctionget_gaussian_kernel1d
r"""Function that returns Gaussian filter coefficients. Args: kernel_size (int): filter size. It should be odd and positive. sigm
pan-sharpening/py-tra/utilsmetric.py:766
↓ 2 callersMethodget_padding
(padding, kernel_size, stride)
pan-sharpening/model/modules.py:116
↓ 2 callersFunctionget_test_data
(cfg, mode)
pan-sharpening/data/data.py:28
↓ 2 callersFunctionim2double
(img)
pan-sharpening/py-tra/utilsmetric.py:50
↓ 2 callersMethodimread
(self, path)
pan-sharpening/tool/real_pre_processing.py:58
↓ 2 callersFunctionlaplacian
r"""Function that returns a tensor using a Laplacian filter. See :class:`~kornia.filters.Laplacian` for details.
pan-sharpening/py-tra/utilsmetric.py:1082
↓ 2 callersFunctionlsqnonneg
Nonnegative least squares via the active set method This function solves the following optimization min |y-Ax|^2 s.t. x>=0
pan-sharpening/py-tra/methods/CNMF.py:677
↓ 2 callersFunctionmaek_optimizer
(opt_type, cfg, params)
pan-sharpening/utils/utils.py:35
↓ 2 callersMethodmodcrop
(self, img, scale = 4)
pan-sharpening/tool/pre_processing.py:74
↓ 2 callersMethodmodcrop
(self, img, scale =3)
pan-sharpening/tool/real_pre_processing.py:62
↓ 2 callersFunctionno_ref_evaluate
(pred, pan, hs)
pan-sharpening/utils/utils.py:469
↓ 2 callersFunctionpannet
(lrhs_size=(16, 16, 3), hrms_size = (64, 64, 1))
pan-sharpening/py-tra/methods/PanNet.py:91
↓ 2 callersFunctionpnn_net
(lrhs_size=(32, 32, 3), hrms_size = (32, 32, 1))
pan-sharpening/py-tra/methods/PNN.py:30
↓ 2 callersFunctionpsnr_loss
r"""Function that computes PSNR See :class:`~kornia.losses.PSNR` for details.
pan-sharpening/py-tra/utilsmetric.py:740
↓ 2 callersMethodread
(self, path)
pan-sharpening/tool/modcrop.py:25
↓ 2 callersMethodrun
(self)
pan-sharpening/solver/solver.py:401
↓ 2 callersFunctionsam
SAM for (N, C, H, W) image; torch.float32 [0.,1.].
pan-sharpening/py-tra/utilsmetric.py:398
↓ 2 callersMethodsample
(mean, logs, eps_std=None)
pan-sharpening/model/modules.py:282
↓ 2 callersMethodsave_checkpoint
(self,epoch)
pan-sharpening/solver/solver.py:369
next →1–100 of 464, ranked by callers