MCPcopy Create free account

hub / github.com/ZhendongWang6/Uformer / functions

Functions188 in github.com/ZhendongWang6/Uformer

⨍ Functions 188◇ Types & classes 38
↓ 39 callersMethodflops
(self, H, W)
model.py:223
↓ 22 callersMethod__init__
(self, n_feat, kernel_size=3, bias=True)
model.py:67
↓ 13 callersMethodstep
(self, epoch=None, metrics=None)
warmup_scheduler/scheduler.py:53
↓ 10 callersFunctionis_png_file
(filename)
utils/image_utils.py:12
↓ 8 callersFunctionload_img
(filepath)
utils/image_utils.py:31
↓ 6 callersMethodget_lr
(self)
warmup_scheduler/scheduler.py:25
↓ 6 callersFunctionreorder_image
(img, input_order='HWC')
utils/caculate_psnr_ssim.py:155
↓ 6 callersFunctionto_y_channel
(img)
utils/caculate_psnr_ssim.py:165
↓ 4 callersMethod__len__
(self)
dataset/dataset_denoise.py:96
↓ 3 callersMethod__init__
(self, rgb_dir, target_transform=None)
dataset/dataset_motiondeblur.py:78
↓ 3 callersFunctionconv
(in_channels, out_channels, kernel_size, bias=False, stride = 1)
model.py:60
↓ 3 callersFunctionwindow_partition
(x, win_size, dilation_rate=1)
model.py:704
↓ 2 callersMethod__init__
(self, rgb_dir, target_transform=None)
dataset/dataset_denoise.py:78
↓ 2 callersFunctionget_test_data
(rgb_dir)
utils/loader.py:13
↓ 2 callersFunctionget_training_data
(rgb_dir, img_options)
utils/loader.py:4
↓ 2 callersFunctionget_validation_deblur_data
(rgb_dir, img_options=None)
dataset/dataset_motiondeblur.py:184
↓ 2 callersMethodinit
(self, parser)
options.py:8
↓ 2 callersFunctionmkdir
(path)
utils/dir_utils.py:12
↓ 2 callersMethodtensor_size
(t)
losses.py:36
↓ 1 callersMethod__init__
(self, tv_loss_weight=1)
losses.py:21
↓ 1 callersMethod__init__
(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)
utils/antialias.py:8
↓ 1 callersFunction_blocking_effect_factor
(im)
utils/caculate_psnr_ssim.py:84
↓ 1 callersFunction_convert_input_type_range
(img)
utils/caculate_psnr_ssim.py:175
↓ 1 callersFunction_convert_output_type_range
(img, dst_type)
utils/caculate_psnr_ssim.py:187
↓ 1 callersFunction_ssim
(img1, img2)
utils/caculate_psnr_ssim.py:35
↓ 1 callersMethodaug
(self, rgb_gt, rgb_noisy)
utils/dataset_utils.py:38
↓ 1 callersFunctionbundle_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:69
↓ 1 callersFunctioncompute_psnr
(image_true, image_test, image_mask, data_range=None)
test/test_realblur.py:126
↓ 1 callersFunctioncompute_ssim
(tar_img, prd_img, cr1)
test/test_realblur.py:132
↓ 1 callersFunctionexpand2square
(timg,factor=16.0)
test/test_gopro_hide.py:77
↓ 1 callersFunctionexpand2square
(timg,factor=16.0)
test/test_realblur.py:65
↓ 1 callersFunctionexpand2square
(timg,factor=16.0)
test/test_dnd.py:79
↓ 1 callersFunctionexpand2square
(timg,factor=16.0)
test/test_sidd.py:79
↓ 1 callersFunctionget_pad_layer
(pad_type)
utils/antialias.py:49
↓ 1 callersFunctionget_pad_layer_1d
(pad_type)
utils/antialias.py:104
↓ 1 callersFunctionget_validation_data
(rgb_dir)
utils/loader.py:8
↓ 1 callersFunctionimage_align
(deblurred, gt)
test/test_realblur.py:81
↓ 1 callersFunctionis_image_file
(filename)
dataset/dataset_denoise.py:16
↓ 1 callersFunctionis_image_file
(filename)
dataset/dataset_motiondeblur.py:16
↓ 1 callersFunctionmyPSNR
(tar_img, prd_img)
utils/image_utils.py:40
↓ 1 callersFunctionrgb2ycbcr
(img, y_only=False)
utils/caculate_psnr_ssim.py:197
↓ 1 callersMethodstep_ReduceLROnPlateau
(self, metrics, epoch=None)
warmup_scheduler/scheduler.py:39
↓ 1 callersFunctionwindow_reverse
(windows, win_size, H, W, dilation_rate=1)
model.py:717
↓ 1 callersMethodwith_pos_embed
(self, tensor, pos)
model.py:901
Method__getitem__
(self, index)
dataset/dataset_denoise.py:42
Method__getitem__
(self, index)
dataset/dataset_denoise.py:99
Method__getitem__
(self, index)
dataset/dataset_denoise.py:129
Method__getitem__
(self, index)
dataset/dataset_motiondeblur.py:42
Method__getitem__
(self, index)
dataset/dataset_motiondeblur.py:99
Method__getitem__
(self, index)
dataset/dataset_motiondeblur.py:132
Method__getitem__
(self, index)
dataset/dataset_motiondeblur.py:169
Method__init__
(self)
options.py:5
Method__init__
(self, eps=1e-3)
losses.py:44
Method__init__
(self, dim=32, hidden_dim=128, act_layer=nn.GELU,drop = 0.)
model.py:16
Method__init__
(self, in_channel, out_channel, strides=1)
model.py:84
Method__init__
(self, block=ConvBlock,dim=32)
model.py:108
Method__init__
(self, in_channels, out_channels, stride = 1)
model.py:207
Method__init__
(self, in_chans, embed_dim=768, s=1)
model.py:231
Method__init__
(self, channel, reduction=16)
model.py:253
Method__init__
(self, channel, k_size=3)
model.py:285
Method__init__
(self, channel, k_size=3)
model.py:317
Method__init__
(self, in_channels, out_channels, kernel_size,
model.py:345
Method__init__
(self, dim, heads = 8, dim_head = 64, kernel_size=3, q_stride=1, k_stride=1, v_stride=1, dropout = 0.,
model.py:382
Method__init__
(self, dim, heads = 8, dim_head = 64, dropout = 0., bias=True)
model.py:422
Method__init__
(self, dim, win_size,num_heads, token_projection='linear', qkv_bias=True, qk_scale=None, attn_drop=0., proj_dr
model.py:453
Method__init__
(self, dim,num_heads, token_projection='linear', qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
model.py:550
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model.py:624
Method__init__
(self, dim=32, hidden_dim=128, act_layer=nn.GELU,drop = 0., use_eca=False)
model.py:655
Method__init__
(self, in_channel, out_channel)
model.py:731
Method__init__
(self, in_channel, out_channel)
model.py:757
Method__init__
(self, in_channel=3, out_channel=64, kernel_size=3, stride=1, norm_layer=None,act_layer=nn.LeakyReLU)
model.py:782
Method__init__
(self, in_channel=64, out_channel=3, kernel_size=3, stride=1, norm_layer=None,act_layer=None)
model.py:814
Method__init__
(self, dim, input_resolution, num_heads, win_size=8, shift_size=0, mlp_ratio=4., qkv_bias=Tru
model.py:851
Method__init__
(self, dim, output_dim, input_resolution, depth, num_heads, win_size, mlp_ratio=4., qkv_bias=
model.py:1014
Method__init__
(self, img_size=256, in_chans=3, dd_in=3, embed_dim=32, depths=[2, 2, 2, 2, 2, 2, 2, 2, 2], n
model.py:1070
Method__init__
(self, optimizer, multiplier, total_epoch, after_scheduler=None)
warmup_scheduler/scheduler.py:16
Method__init__
(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)
utils/antialias.py:62
Method__init__
(self)
utils/dataset_utils.py:6
Method__init__
(self)
utils/dataset_utils.py:35
Method__init__
(self, rgb_dir, img_options=None, target_transform=None)
dataset/dataset_denoise.py:21
Method__init__
(self, inp_dir, img_options)
dataset/dataset_denoise.py:117
Method__init__
(self, rgb_dir, img_options=None, target_transform=None)
dataset/dataset_motiondeblur.py:21
Method__init__
(self, rgb_dir, img_options=None, rgb_dir2=None)
dataset/dataset_motiondeblur.py:115
Method__init__
(self, inp_dir, img_options)
dataset/dataset_motiondeblur.py:157
Method__len__
(self)
dataset/dataset_denoise.py:39
Method__len__
(self)
dataset/dataset_denoise.py:126
Method__len__
(self)
dataset/dataset_motiondeblur.py:39
Method__len__
(self)
dataset/dataset_motiondeblur.py:96
Method__len__
(self)
dataset/dataset_motiondeblur.py:129
Method__len__
(self)
dataset/dataset_motiondeblur.py:166
Method_init_weights
(self, m)
model.py:1249
Functionbatch_PSNR
(img1, img2, average=True)
utils/image_utils.py:46
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:6
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:37
Functioncalculate_psnr
(img1, img2, crop_border=0, input_order='HWC', test_y_channel=False)
utils/caculate_psnr_ssim.py:12
Functioncalculate_psnrb
(img1, img2, crop_border, input_order='HWC', test_y_channel=False)
utils/caculate_psnr_ssim.py:124
Functioncalculate_ssim
(img1, img2, crop_border=0, input_order='HWC', test_y_channel=False)
utils/caculate_psnr_ssim.py:57
Methodextra_repr
(self)
model.py:524
Methodextra_repr
(self)
model.py:597
Methodextra_repr
(self)
model.py:904
next →1–100 of 188, ranked by callers