Functions — github.com/GuoLanqing/ShadowFormer

CodeHub MCPcopy Create free account

hub / github.com/GuoLanqing/ShadowFormer / functions

Functions140 in github.com/GuoLanqing/ShadowFormer

⨍ Functions 140◇ Types & classes 31

↓ 20 callersMethod__init__

(self, n_feat, kernel_size, reduction, bias, act)

↓ 18 callersMethodflops

↓ 8 callersMethodstep

(self, epoch=None, metrics=None)

warmup_scheduler/scheduler.py:53

↓ 6 callersFunctionis_png_file

utils/image_utils.py:13

↓ 4 callersFunctionload_img

utils/image_utils.py:32

↓ 4 callersFunctionwindow_partition

(x, win_size, dilation_rate=1)

↓ 2 callersMethod__len__

↓ 2 callersFunctionconv

(in_channels, out_channels, kernel_size, bias=False, stride = 1)

↓ 2 callersMethodget_lr

warmup_scheduler/scheduler.py:25

↓ 2 callersFunctionget_validation_data

utils/loader.py:8

↓ 2 callersFunctionload_mask

utils/image_utils.py:46

↓ 2 callersFunctionmkdir

utils/dir_utils.py:12

↓ 2 callersMethodtensor_size

↓ 1 callersMethod__init__

(self, rgb_dir, target_transform=None)

↓ 1 callersMethod__init__

(self, tv_loss_weight=1)

↓ 1 callersMethod__init__

(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)

utils/antialias.py:20

↓ 1 callersMethodaug

(self, rgb_gt, rgb_noisy, gray_mask)

utils/dataset_utils.py:38

↓ 1 callersFunctionget_pad_layer

utils/antialias.py:61

↓ 1 callersFunctionget_pad_layer_1d

utils/antialias.py:116

↓ 1 callersFunctionget_training_data

(rgb_dir, img_options)

utils/loader.py:4

↓ 1 callersMethodinit

↓ 1 callersFunctionmyPSNR

(tar_img, prd_img)

utils/image_utils.py:71

↓ 1 callersMethodstep_ReduceLROnPlateau

(self, metrics, epoch=None)

warmup_scheduler/scheduler.py:39

↓ 1 callersFunctionwindow_reverse

(windows, win_size, H, W, dilation_rate=1)

Method__getitem__

Method__getitem__

(self, rgb_dir, img_options=None, target_transform=None)

(self, eps=1e-3)

(self, in_channel, out_channel, strides=1)

(self, block=ConvBlock,dim=32)

(self, in_chans, embed_dim=768, s=1)

(self, channel, reduction=16)

(self, in_channels, out_channels, kernel_size,

(self, channel, reduction=16, bias=False)

(self, dim, heads = 8, dim_head = 64, kernel_size=3, q_stride=1, k_stride=1, v_stride=1, dropout = 0.,

(self, dim, heads = 8, dim_head = 64, dropout = 0., bias=True)

(self, dim, heads = 8, dim_head = 64, dropout = 0., bias=True)

(self, dim, win_size, num_heads, token_projection='linear', qkv_bias=True, qk_scale=None, attn_drop=0.,

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

(self, dim=32, hidden_dim=128, act_layer=nn.GELU,drop = 0.)

(self, in_channel, out_channel)

(self, in_channel, out_channel)

(self, in_channel=3, out_channel=64, kernel_size=3, stride=1, norm_layer=None,act_layer=nn.LeakyReLU)

(self, in_channel=64, out_channel=3, kernel_size=3, stride=1, norm_layer=None,act_layer=None)

(self, dim, input_resolution, num_heads, win_size=10, shift_size=0, mlp_ratio=4., qkv_bias=T

(self, dim, input_resolution, num_heads, win_size=10, shift_size=0, mlp_ratio=4., qkv_bias=T

(self, dim, output_dim, input_resolution, depth, num_heads, win_size, mlp_ratio=4., qkv_bias

(self, img_size=256, in_chans=3, embed_dim=32, depths=[2, 2, 2, 2, 2, 2, 2, 2, 2], num_heads

(self, optimizer, multiplier, total_epoch, after_scheduler=None)

warmup_scheduler/scheduler.py:16

(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)

utils/antialias.py:74

utils/dataset_utils.py:6

utils/dataset_utils.py:35

Method_init_weights

Functionbatch_PSNR

(img1, img2, average=True)

utils/image_utils.py:77

Functionbundle_submissions_raw

Bundles submission data for raw denoising submission_folder Folder where denoised images reside Output is written to <submission_folder>/

utils/bundle_submissions.py:13

Functionbundle_submissions_srgb

Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_fo

utils/bundle_submissions.py:44

Functionbundle_submissions_srgb_v1

Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_fo

utils/bundle_submissions.py:76

Functioncalc_RMSE

(real_img, fake_img)

utils/image_utils.py:107

Methodextra_repr

Methodextra_repr

Methodextra_repr

Methodextra_repr

Methodextra_repr

(self, x, H=None, W=None)

(self, x, attn_kv=None)

(self, x, attn_kv=None)

(self, x, attn_kv=None)

(self, x, xm, attn_kv=None, mask=None)

(self, x, img_size=(128,128))

(self, x, img_size=(128,128))

(self, x, img_size=(128,128))

(self, x, img_size=(128,128))

(self, x, xm, mask=None, img_size=(128, 128))

next →1–100 of 140, ranked by callers