Functions — github.com/Lizhe1228/MambaDFuse

CodeHub MCPcopy Create free account

hub / github.com/Lizhe1228/MambaDFuse / functions

Functions470 in github.com/Lizhe1228/MambaDFuse

⨍ Functions 470◇ Types & classes 81

models/loss.py:147

models/loss.py:172

models/network.py:70

models/network.py:78

models/network.py:107

models/network.py:120

models/network.py:133

(self, I1,I2 ,I1_residual,I2_residual)

models/network.py:147

(self,I1,fusion_resi,I2,fusion,test_h,test_w)

models/network.py:171

models/network.py:213

(self, x, x_size)

models/network.py:251

models/network.py:481

models/loss_gt.py:15

(self, GT, fused)

models/loss_gt.py:31

(self, image_A, image_B, image_fused)

models/loss_gt.py:45

models/loss_gt.py:69

(self, image_GT, image_fused)

models/loss_gt.py:77

(self, org , enhance )

models/loss_gt.py:95

models/loss_gt.py:132

models/loss_gt.py:164

(self, image_A, image_B, image_fused, image_GT)

models/loss_gt.py:184

models/loss_med.py:16

(self, image_A, image_B, image_fused)

models/loss_med.py:34

(self, image_A, image_B, image_fused)

models/loss_med.py:43

models/loss_med.py:69

(self, image_A, image_B, image_fused)

models/loss_med.py:78

(self, image_A, image_B, image_fused)

models/loss_med.py:91

models/basicblock.py:145

models/basicblock.py:165

models/basicblock.py:180

models/basicblock.py:197

models/basicblock.py:221

models/basicblock.py:259

models/basicblock.py:287

models/basicblock.py:317

models/basicblock.py:345

models/basicblock.py:364

models/basicblock.py:384

models/basicblock.py:403

models/basicblock.py:424

:param x: (b, c, t, h, w) :return:

models/basicblock.py:568

models/network_feature.py:40

(self, img1, img2)

models/loss_ssim.py:73

models/loss_vif.py:18

(self, image_A, image_B, image_fused)

models/loss_vif.py:35

(self, image_A, image_B, image_fused)

models/loss_vif.py:48

models/loss_vif.py:68

(self, image_A, image_B, image_fused)

models/loss_vif.py:77

(self, image_A, image_B, image_fused)

models/loss_vif.py:91

models/refine.py:22

models/refine.py:48

models/refine.py:71

models/refine.py:91

models/refine.py:130

models/refine.py:193

models/refine.py:284

models/refine.py:302

models/refine.py:322

(ctx, grad_output, out, negative_slope, scale)

models/op/fused_act.py:23

(ctx, input, bias, negative_slope, scale)

models/op/fused_act.py:55

models/op/fused_act.py:83

( ctx, grad_output, kernel, grad_kernel, up, down, pad, g_pad, in_size, out_size )

models/op/upfirdn2d.py:21

(ctx, input, kernel, up, down, pad)

models/op/upfirdn2d.py:90

Functionfun_reshape

utils/utils_sisr.py:616

Functionfused_bias_act

models/op/fused_bias_act.cpp:11

Functiong_path_regularize

(fake_img, latents, mean_path_length, decay=0.01)

models/loss.py:196

Functiongather_grad

utils/utils_dist.py:131

Functiongen_kernel

# modified version of https://github.com/assafshocher/BlindSR_dataset_generator # Kai Zhang # min_var = 0.175 * sf # variance of the gau

utils/utils_sisr.py:172

Functionget_image_pair

(args, path, a_dir=None, b_dir=None)

test_MambaDFuse.py:109

Functionget_model_activation

(model, input_res, input_constructor=None)

utils/utils_modelsummary.py:48

Functionget_model_complexity_info

(model, input_res, print_per_layer_stat=True, as_strings=True, input_constructor

utils/utils_modelsummary.py:67

Functionget_model_flops

(model, input_res, print_per_layer_stat=True, input_constructor=None)

utils/utils_modelsummary.py:27

Functionget_rank

utils/utils_dist.py:80

Functiongradient_penalty_loss

Calculate gradient penalty for wgan-gp. Args: discriminator (nn.Module): Network for the discriminator. real_data (Tensor): Real i

models/loss.py:211

utils/utils_sisr.py:421

Functionimresize

(img, scale, antialiasing=True)

utils/utils_image.py:878

Functionimresize_np

(img, scale, antialiasing=True)

utils/utils_image.py:951

(img, img_path)

utils/utils_image.py:208

(x, title=None, cbar=False, figsize=None)

utils/utils_image.py:37

Functionimwrite

(img, img_path)

utils/utils_image.py:214

Functioninfo_model

utils/utils_model.py:255

Methodinfo_network

models/model_base.py:80

Methodinfo_network

models/model_plain.py:339

Functioninfo_params

utils/utils_model.py:263

Methodinfo_params

models/model_base.py:86

Methodinfo_params

models/model_plain.py:346

Functioninit_fn

(m, init_type='xavier_uniform', init_bn_type='uniform', gain=1)

models/select_network.py:161

Methodinit_train

models/model_plain.py:49

Functioninterpolation_down

(x, sf, center=False)

utils/utils_sisr.py:579

utils/utils_sisr.py:413

Functionlinear_flops_counter_hook

(module, input, output)

utils/utils_modelsummary.py:305

(*args, **kwargs)

utils/utils_logger.py:16

Functionlogger_info

set up logger modified by Kai Zhang (github: https://github.com/cszn)

utils/utils_logger.py:27

Functionmaster_only

utils/utils_dist.py:100

Functionmat2json

Converts .mat file to .json and writes new file Parameters ---------- mat_path: Str path/filename .mat存放路径 filepath: Str

utils/utils_mat.py:58

Methodmerge_bnorm_test

models/model_base.py:214

Methodmerge_bnorm_train

models/model_base.py:205

Functionmkdir_and_rename

utils/utils_image.py:171

utils/utils_image.py:163

Functionmodcrop

img: tensor image, NxCxWxH or CxWxH or WxH sf: scale factor

utils/utils_sisr.py:478

← previous next →301–400 of 470, ranked by callers