MCPcopy Create free account

hub / github.com/FZU-N/URWKV / functions

Functions288 in github.com/FZU-N/URWKV

↓ 1 callersMethodshow_cost_volume
(self)
custom_utils/visdom.py:114
↓ 1 callersMethodshow_cost_volume_slice
(self)
custom_utils/visdom.py:134
↓ 1 callersMethodstep_ReduceLROnPlateau
(self, metrics, epoch=None)
custom_utils/warmup_scheduler/scheduler.py:39
↓ 1 callersFunctionto_4d
(img)
tools/measure.py:111
↓ 1 callersFunctionto_CHW
(img)
tools/measure.py:118
↓ 1 callersFunctionto_tensor
(img)
tools/measure.py:121
↓ 1 callersMethodtoggle_display
(self, new_mode=None)
custom_utils/visdom.py:32
↓ 1 callersMethodtoggle_show_slice
(self, new_mode=None)
custom_utils/visdom.py:128
↓ 1 callersFunctiontorchPSNR
(tar_img, prd_img)
custom_utils/image_utils.py:57
↓ 1 callersMethodupdate
(self, data, **kwargs)
custom_utils/visdom.py:20
FunctionPYBIND11_MODULE
model/modules/cuda/wkv_op.cpp:13
MethodRandom_Crop
(self, low, high)
custom_utils/data_loaders/exposure.py:76
MethodRandom_Crop
(self, low, high)
custom_utils/data_loaders/mit5k.py:79
FunctionTORCH_LIBRARY
model/modules/cuda/wkv_op.cpp:18
Method__call__
(self, a, b)
custom_utils/validation.py:27
Method__getitem__
(self, index)
custom_utils/UHD_load/train_data_aug_local.py:68
Method__getitem__
(self, index)
custom_utils/UHD_load/val_data.py:72
Method__getitem__
(self, index)
custom_utils/UHD_load/val_data_train.py:35
Method__getitem__
(self, index)
custom_utils/data_loaders/exposure.py:84
Method__getitem__
(self, index)
custom_utils/data_loaders/lol.py:37
Method__getitem__
(self, index)
custom_utils/data_loaders/lol.py:116
Method__getitem__
(self, index)
custom_utils/data_loaders/lol.py:158
Method__getitem__
(self, index)
custom_utils/data_loaders/mit5k.py:87
Method__init__
(self, net='alex', use_gpu=False)
tools/measure.py:39
Method__init__
(self, rgb_range, rgb_mean, rgb_std, sign=-1)
model/loss.py:18
Method__init__
(self, window_size=11, size_average=True)
model/loss.py:96
Method__init__
(self)
model/loss.py:125
Method__init__
(self, vgg_model)
model/loss.py:138
Method__init__
(self, dim)
model/decoder.py:16
Method__init__
(self, dim)
model/encoder.py:7
Method__init__
(self, dim)
model/model_builder.py:20
Method__init__
(self, drop_prob: float = 0., scale_by_keep: bool = True)
model/modules/drop.py:27
Method__init__
(self, in_channels, out_channels)
model/modules/SSF.py:6
Method__init__
(self, num_channels)
model/modules/SSF.py:94
Method__init__
(self, alpha=0.5)
model/modules/urwkv.py:111
Method__init__
(self, n_embd, n_layer, layer_id, shift_mode='q_shift', channel_gamma=1/4, shift_pixel=1, ini
model/modules/urwkv.py:125
Method__init__
(self, n_embd, n_layer, layer_id, shift_mode='q_shift', channel_gamma=1/4, shift_pixel=1, hid
model/modules/urwkv.py:246
Method__init__
(self, normalized_shape)
model/modules/urwkv.py:328
Method__init__
(self, img_size=32, patch_size=16, in_channels=3,
model/modules/urwkv.py:412
Method__init__
(self, dim, channel_first=True, seed=42)
model/modules/LAN.py:27
Method__init__
(self, window_size: int = 11, reduction: str = 'mean', max_val: float = 1.0)
custom_utils/losses.py:67
Method__init__
(self, eps=1e-3)
custom_utils/losses.py:128
Method__init__
(self)
custom_utils/losses.py:139
Method__init__
(self)
custom_utils/dataset_utils.py:6
Method__init__
(self, optimizer, last_epoch=-1)
custom_utils/lr_scheduler.py:30
Method__init__
(self, optimizer, start_lr=0.03, end_lr=5e-4, epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:49
Method__init__
(self, optimizer, start_lr=0.01, end_lr=None, step=10, mult=0.1, epochs=50, last_epoch=-1, **
custom_utils/lr_scheduler.py:62
Method__init__
(self, optimizer, start_lr=0.01, end_lr=None, steps=[10, 20, 30, 40], mult=0.5, epochs=50,
custom_utils/lr_scheduler.py:78
Method__init__
(self, optimizer, start_lr=0.01, end_lr=0.005, epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:104
Method__init__
(self, optimizer, start_lr=0.01, end_lr=0.005, epochs=50, last_epoch=-1, **kwargs)
custom_utils/lr_scheduler.py:113
Method__init__
(self, optimizer, warmup, normal, epochs=50, last_epoch=-1)
custom_utils/lr_scheduler.py:129
Method__init__
(self, max_val=0)
custom_utils/validation.py:18
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:14
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:45
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:57
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:74
Method__init__
(self, visdom, show_data, title, flip=False)
custom_utils/visdom.py:108
Method__init__
(self, visdom, show_data, title, feat_shape, registered_blocks)
custom_utils/visdom.py:185
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:240
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:268
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:280
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:298
Method__init__
(self, visdom, show_data, title)
custom_utils/visdom.py:348
Method__init__
(self, optimizer, multiplier, total_epoch, after_scheduler=None)
custom_utils/warmup_scheduler/scheduler.py:16
Method__init__
(self)
custom_utils/WT/transform.py:50
Method__init__
(self, crop_size, train_data_dir)
custom_utils/UHD_load/train_data_aug_local.py:19
Method__init__
(self, dataset_name,val_data_dir)
custom_utils/UHD_load/val_data.py:11
Method__init__
(self, val_data_dir)
custom_utils/UHD_load/val_data_train.py:12
Method__init__
(self, images_path, mode='train', expert='c', normalize=False)
custom_utils/data_loaders/exposure.py:32
Method__init__
(self, rgb_dir, img_options=None)
custom_utils/data_loaders/lol.py:20
Method__init__
(self, rgb_dir, img_options=None, rgb_dir2=None)
custom_utils/data_loaders/lol.py:100
Method__init__
(self, images_path, mode='train', normalize=False)
custom_utils/data_loaders/mit5k.py:37
Method__iter__
(self)
custom_utils/dataset_utils.py:23
Method__len__
(self)
custom_utils/UHD_load/train_data_aug_local.py:72
Method__len__
(self)
custom_utils/UHD_load/val_data.py:76
Method__len__
(self)
custom_utils/UHD_load/val_data_train.py:39
Method__len__
(self)
custom_utils/data_loaders/exposure.py:112
Method__len__
(self)
custom_utils/data_loaders/lol.py:34
Method__len__
(self)
custom_utils/data_loaders/lol.py:113
Method__len__
(self)
custom_utils/data_loaders/lol.py:227
Method__len__
(self)
custom_utils/data_loaders/mit5k.py:114
Method__repr__
(self)
custom_utils/lr_scheduler.py:43
Function_build_lr_scheduler
(optimizer, config, epochs=50, last_epoch=-1)
custom_utils/lr_scheduler.py:146
Method_forward
(x)
model/loss.py:46
Method_init_weights
(self, m)
model/model_builder.py:29
Method_inner_forward
(x)
model/modules/urwkv.py:227
Method_inner_forward
(x, channel_shift_list)
model/modules/urwkv.py:295
Method_inner_forward
(x, spatial_shift_list, channel_shift_list, inter_feat)
model/modules/urwkv.py:376
Functionadjust_learning_rate
(optimizer, epoch, lr_decay=0.5)
custom_utils/lr_scheduler.py:15
Methodaug
(self, rgb_gt, rgb_noisy)
custom_utils/dataset_utils.py:9
Functionbackward
model/modules/cuda/wkv_op.cpp:9
Functionbatch_PSNR
(img1, img2, data_range=None)
custom_utils/image_utils.py:63
Methodblock_list_callback_handler
(self, data)
custom_utils/visdom.py:78
Methodblock_list_callback_handler
(self, data)
custom_utils/visdom.py:352
Methodblock_list_callback_handler
(self, data)
custom_utils/visdom.py:394
Functionbrisque_val
(im)
custom_utils/metrics/brisque.py:128
Functioncalculate_psnr
(img1, img2, border=0)
custom_utils/image_utils.py:7
Functioncalculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]
custom_utils/image_utils.py:28
Functioncrop_output
(net_input, net_output)
custom_utils/img_resize.py:34
Methodcutblur
(self, im1, im2, prob=1.0, alpha=1.0)
custom_utils/UHD_load/train_data_aug_local.py:76
← previousnext →101–200 of 288, ranked by callers