Functions — github.com/cschenxiang/HCT-FFN

CodeHub MCPcopy Create free account

hub / github.com/cschenxiang/HCT-FFN / functions

Functions152 in github.com/cschenxiang/HCT-FFN

⨍ Functions 152◇ Types & classes 42

↓ 12 callersMethod__init__

(self, n_feats, nm='in')

code/model/hct-ffn.py:108

↓ 12 callersMethod__init__

(self, channel, reduction=16)

code/model/common.py:100

↓ 6 callersMethodget_model

code/model/__init__.py:57

↓ 6 callersMethodload

(self, apath, cpu=False)

code/loss/__init__.py:147

↓ 5 callersMethodstate_dict

(self, **kwargs)

code/model/__init__.py:63

↓ 5 callersMethodstep

code/loss/__init__.py:101

↓ 5 callersMethodwrite_log

(self, log, refresh=False)

code/utility.py:93

↓ 4 callersMethod_check_and_load

(self, ext, l, f, verbose=True, load=True)

code/data/srdata.py:128

↓ 3 callersFunctioncreate_window

(window_size, channel)

code/loss/ssim.py:11

↓ 3 callersFunctionget_residue

return residue_channle (RGB)

code/model/hct-ffn.py:68

↓ 3 callersMethodprepare

code/trainer.py:133

↓ 3 callersMethodset_scale

(self, idx_scale)

code/data/demo.py:39

↓ 3 callersMethodtic

code/utility.py:23

↓ 2 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.ReLU, drop=0.)

code/util/rlutrans.py:11

↓ 2 callersMethod_scan

code/data/srdata.py:90

↓ 2 callersFunction_ssim

(img1, img2, window, window_size, channel, size_average = True)

code/loss/ssim.py:17

↓ 2 callersMethodhold

code/utility.py:29

↓ 2 callersMethodrelease

code/utility.py:32

↓ 2 callersMethodtest

code/trainer.py:81

↓ 2 callersMethodtoc

code/utility.py:26

↓ 1 callersFunction_augment

code/data/common.py:63

↓ 1 callersMethod_get_index

code/data/srdata.py:165

↓ 1 callersMethod_load_file

code/data/srdata.py:171

↓ 1 callersMethod_name_hrbin

code/data/srdata.py:114

↓ 1 callersMethod_name_lrbin

code/data/srdata.py:121

↓ 1 callersFunction_np2Tensor

code/data/common.py:49

↓ 1 callersFunction_set_channel

code/data/common.py:34

↓ 1 callersMethod_set_filesystem

(self, dir_data)

code/data/srdata.py:108

↓ 1 callersMethodadd_log

code/utility.py:90

↓ 1 callersMethoddisplay_loss

code/loss/__init__.py:112

↓ 1 callersMethoddone

code/utility.py:100

↓ 1 callersMethodend_log

(self, n_batches)

code/loss/__init__.py:109

↓ 1 callersFunctionextract_image_patches

Extract patches from images and put them in the C output dimension. :param padding: :param images: [batch, channels, in_rows, in_cols]. A

code/util/tools.py:30

↓ 1 callersMethodforward_chop

(self, x, shave=10, min_size=160000)

code/model/__init__.py:118

↓ 1 callersMethodforward_x8

(self, x, forward_function)

code/model/__init__.py:159

↓ 1 callersFunctiongaussian

(window_size, sigma)

code/loss/ssim.py:7

↓ 1 callersMethodget_loss_module

code/loss/__init__.py:137

↓ 1 callersMethodget_patch

code/data/srdata.py:191

↓ 1 callersMethodload

(self, apath, pre_train='.', resume=-1, cpu=False)

code/model/__init__.py:85

↓ 1 callersMethodplot_loss

(self, apath, epoch)

code/loss/__init__.py:120

↓ 1 callersMethodplot_psnr

code/utility.py:103

↓ 1 callersFunctionprint_network

code/main.py:12

↓ 1 callersFunctionreverse_patches

Extract patches from images and put them in the C output dimension. :param padding: :param images: [batch, channels, in_rows, in_cols]. A

code/util/tools.py:60

↓ 1 callersFunctionsame_padding

(images, ksizes, strides, rates)

code/util/tools.py:11

↓ 1 callersMethodsave

(self, trainer, epoch, is_best=False)

code/utility.py:78

↓ 1 callersMethodsave_results

(self, filename, save_list, scale)

code/utility.py:121

↓ 1 callersMethodstart_log

code/loss/__init__.py:106

↓ 1 callersMethodterminate

code/trainer.py:141

↓ 1 callersMethodtrain

code/trainer.py:37

Method__getitem__

code/data/srdata.py:149

Method__getitem__

code/data/demo.py:27

code/utility.py:19

code/utility.py:42

code/dataloader.py:56

( self, args, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None,

code/dataloader.py:128

(self, args, loader, my_model, my_loss, ckp)

code/trainer.py:16

(self, C, stride)

code/model/hct-ffn.py:12

(self, steps, C)

code/model/hct-ffn.py:29

(self, channel, k, num_ops)

code/model/hct-ffn.py:50

(self, inputchannel, outchannel, kernel_size, stride)

code/model/hct-ffn.py:79

(self, in_channels, out_channels, kernel_size, stride)

code/model/hct-ffn.py:93

(self, n_feats, n_blocks, nm='in')

code/model/hct-ffn.py:125

(self, n_feats, blocks=2)

code/model/hct-ffn.py:140

(self, inchannel=64, outchannel=64)

code/model/hct-ffn.py:152

(self, in_dim=32)

code/model/hct-ffn.py:169

(self, n_feats,layer_num ,steps=4)

code/model/hct-ffn.py:201

(self, n_feats, blocks=2)

code/model/hct-ffn.py:234

code/model/hct-ffn.py:266

(self, args, ckp)

code/model/__init__.py:9

(self, rgb_range, rgb_mean, rgb_std, sign=-1)

code/model/common.py:15

( self, rgb_range, rgb_mean=(0.4488, 0.4371, 0.4040), rgb_std=(1.0, 1.0, 1.0), sign=-1)

code/model/common.py:25

( self, in_channels, out_channels, kernel_size, stride=1, bias=False, bn=True, act=nn.ReLU(Tru

code/model/common.py:37

( self, conv, n_feats, kernel_size, bias=True, bn=False, act=nn.ReLU(True), res_scale=1)

code/model/common.py:50

(self, conv, scale, n_feats, bn=False, act=False, bias=True)

code/model/common.py:71

(self, channels, reduction=16)

code/model/common.py:128

(self, C_in, C_out, kernel_size, stride, padding, affine=True)

code/model/common.py:167

(self, C_in, C_out, kernel_size, stride, padding, affine=True)

code/model/common.py:178

(self, C_in, C_out, kernel_size, stride, padding, dilation, affine=True)

code/model/common.py:188

(self, C_in, C_out, kernel_size, stride, padding, affine=True)

code/model/common.py:198

(self, C_in, C_out, kernel_size, stride, padding, affine=True)

code/model/common.py:236

(self, args, name='', train=True, benchmark=False)

code/data/srdata.py:13

(self, args, name='', train=True, benchmark=True)

code/data/benchmark.py:12

(self, args, name='RainHeavy', train=True, benchmark=False)

code/data/rainheavy.py:5

code/data/__init__.py:7

(self, args, name='RainHeavyTest', train=True, benchmark=False)

code/data/rainheavytest.py:5

(self, args, name='Demo', train=False, benchmark=False)

code/data/demo.py:12

(self, args, ckp)

code/loss/__init__.py:15

(self, window_size = 11, size_average = True)

code/loss/ssim.py:40

(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)

code/util/rlutrans.py:29

( self, n_feat = 64,dim=64, num_heads=8, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_dr

code/util/rlutrans.py:63

code/dataloader.py:143

code/data/srdata.py:159

code/data/demo.py:36

Method_initialize_weights

code/model/hct-ffn.py:308

Method_make_dir

code/utility.py:63

Function_ms_loop

(dataset, index_queue, data_queue, collate_fn, scale, seed, init_fn, worker_id)

code/dataloader.py:29

code/trainer.py:135

code/data/rainheavy.py:10

code/data/rainheavytest.py:10

Method_set_filesystem

(self, dir_data)

code/data/benchmark.py:17

next →1–100 of 152, ranked by callers