MCPcopy Create free account

hub / github.com/PSRben/FusionMamba / functions

Functions51 in github.com/PSRben/FusionMamba

⨍ Functions 51◇ Types & classes 15
↓ 6 callersMethod__init__
(self, in_channels, out_channels, scale, upsample='default', combine='add')
model/u2net.py:37
↓ 3 callersFunctioncreate_window
(window_size, channel)
utils/tools.py:31
↓ 3 callersMethodstep
(self, hidden_states, conv_state, ssm_state)
Mamba-block/mamba_simple.py:983
↓ 2 callersMethod__init__
(self, dim, H, W, depth=1, final=False)
model/fusion_mamba.py:48
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average = True)
utils/tools.py:38
↓ 2 callersFunctionget_edge_h5py
(data)
utils/load_test_data.py:24
↓ 2 callersFunctionget_edge_mat
(data)
utils/load_test_data.py:14
↓ 1 callersMethod__init__
(self, window_size = 11, size_average = True)
utils/tools.py:62
↓ 1 callersMethod__init__
( self, d_model, d_state=16, d_conv=4, expand=2, dt_rank="auto
Mamba-block/mamba_simple.py:49
↓ 1 callersMethod_get_states_from_cache
(self, inference_params, batch_size, initialize_states=False)
Mamba-block/mamba_simple.py:1043
↓ 1 callersMethodallocate_inference_cache
(self, batch_size, max_seqlen, dtype=None, **kwargs)
Mamba-block/mamba_simple.py:1030
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/tools.py:26
↓ 1 callersFunctionget_edge
(data)
utils/load_train_data.py:8
↓ 1 callersFunctionload_h5py_with_hp
(file_path)
utils/load_test_data.py:68
↓ 1 callersFunctionprepare_training_data
(args)
train.py:33
↓ 1 callersFunctionsave_checkpoint
(args, model, epoch)
train.py:26
↓ 1 callersFunctiontest
(args)
test.py:10
↓ 1 callersFunctiontrain
(args, training_data_loader, validate_data_loader)
train.py:43
Method__getitem__
(self, index)
utils/load_train_data.py:45
Method__init__
(self, ratio=4)
utils/tools.py:11
Method__init__
(self, file_path)
utils/load_train_data.py:20
Method__init__
(self, dim, res_se_ratio)
model/u2net.py:10
Method__init__
(self, dim, scale)
model/u2net.py:25
Method__init__
(self, in_channels, out_channels, scale, downsample='default')
model/u2net.py:89
Method__init__
(self, in_channels, out_channels, H, W, scale=2, sample_mode='down')
model/u2net.py:109
Method__init__
(self, spe_channels, se_ratio=8, mode='mamba', channels=32)
model/u2net.py:132
Method__init__
(self, dim, pan_dim, ms_dim, H=64, W=64, scale=4)
model/u2net.py:162
Method__init__
(self, dim, H, W)
model/fusion_mamba.py:12
Method__init__
(self, dim, H, W)
model/fusion_mamba.py:29
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:1073
Method__len__
(self)
utils/load_train_data.py:52
Methodallocate_inference_cache
(self, batch_size, max_seqlen, dtype=None, **kwargs)
Mamba-block/mamba_simple.py:1127
Methodforward
(self, img, gt)
utils/tools.py:15
Methodforward
(self, img1, img2)
utils/tools.py:69
Methodforward
(self, x)
model/u2net.py:17
Methodforward
(self, x)
model/u2net.py:32
Methodforward
(self, x1, x2)
model/u2net.py:79
Methodforward
(self, x)
model/u2net.py:104
Methodforward
(self, pan, ms, pan_pre=None, ms_pre=None)
model/u2net.py:117
Methodforward
(self, input)
model/u2net.py:152
Methodforward
(self, ms, pan)
model/u2net.py:194
Methodforward
(self, input)
model/fusion_mamba.py:19
Methodforward
(self, input0, input1)
model/fusion_mamba.py:37
Methodforward
(self, pan, ms)
model/fusion_mamba.py:60
Methodforward
hidden_states: (B, L, D) extra_emb: (B, L, D) Returns: same shape as hidden_states
Mamba-block/mamba_simple.py:241
Methodforward
r"""Pass the input through the encoder layer. Args: hidden_states: the sequence to the encoder layer (required). resi
Mamba-block/mamba_simple.py:1099
Functiongettime
()
utils/tools.py:100
Functionload_h5py
(file_path)
utils/load_test_data.py:54
Functionload_mat
(file_path)
utils/load_test_data.py:35
Functionload_mat_with_hp
(file_path)
utils/load_test_data.py:45
Functionssim
(img1, img2, window_size = 11, size_average = True)
utils/tools.py:89