MCPcopy Create free account

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

Functions288 in github.com/FZU-N/URWKV

⨍ Functions 288◇ Types & classes 59
↓ 11 callersMethodstep
(self, epoch=None, metrics=None)
custom_utils/warmup_scheduler/scheduler.py:53
↓ 10 callersMethod__init__
(self, debug=0, ui_info=None, visdom_info=None, port=None)
custom_utils/visdom.py:381
↓ 7 callersMethod__init__
(self)
custom_utils/lr_scheduler.py:173
↓ 5 callersMethod__init__
(self, n_embd, n_layer, layer_id, shift_mode='q_shift', channel_gamma=1/4, shift_pixel=1, dro
model/modules/urwkv.py:349
↓ 5 callersFunctionaggd_features
(imdata)
custom_utils/metrics/niqe.py:21
↓ 5 callersMethodfilter2D
( self, input: torch.Tensor, kernel: torch.Tensor, channel: in
custom_utils/losses.py:85
↓ 4 callersMethod__init__
(self, conv_index='54', rgb_range=1)
model/loss.py:29
↓ 4 callersFunctionaggd_features
(imdata)
custom_utils/metrics/brisque.py:16
↓ 4 callersFunctioncheck_keys
(model, pretrained_state_dict)
custom_utils/model_load.py:16
↓ 4 callersFunctionis_image_file
(filename)
custom_utils/data_loaders/lol.py:13
↓ 4 callersFunctionsegmentEdge
(blockEdge, nSegments, blockSize, windowSize)
custom_utils/metrics/piqe.py:19
↓ 3 callersMethod__init__
(self, loss_weight=1.0, reduction='mean', toY=False)
custom_utils/losses.py:40
↓ 3 callersFunctionbitget
(byteval, idx)
custom_utils/plotting.py:95
↓ 3 callersMethoddraw_data
(self)
custom_utils/visdom.py:29
↓ 3 callersFunctiondraw_figure
(fig)
custom_utils/plotting.py:7
↓ 3 callersFunctioneval
()
tools/test.py:63
↓ 3 callersMethodget_lr
(self)
custom_utils/lr_scheduler.py:38
↓ 3 callersFunctionremove_prefix
Old style model is stored with all names of parameters share common prefix 'module.'
custom_utils/model_load.py:39
↓ 2 callersMethod__init__
(self, num_feats, encode_channels, target_channels)
model/modules/SSF.py:38
↓ 2 callersMethod__init__
(self, images_path, mode='train')
custom_utils/data_loaders/lol.py:150
↓ 2 callersFunction_get_patches_generic
(img, patch_size, is_train, stride)
custom_utils/metrics/niqe.py:167
↓ 2 callersFunctioncompute_image_mscn_transform
(image, C=1, avg_window=None, extend_mode='constant')
custom_utils/metrics/niqe.py:105
↓ 2 callersMethodconv_gauss
(self, img)
custom_utils/losses.py:147
↓ 2 callersFunctioncreate_window
(window_size, channel)
model/loss.py:64
↓ 2 callersMethoddynamicInterpolate
(self, x, spatial_shift_list)
model/modules/urwkv.py:115
↓ 2 callersFunctionextract_brisque_feats
(mscncoefs)
custom_utils/metrics/brisque.py:103
↓ 2 callersFunctionextract_on_patches
(img, patch_size)
custom_utils/metrics/niqe.py:148
↓ 2 callersFunctionfiFindByWildcard
(wildcard)
tools/measure.py:127
↓ 2 callersFunctionget_gaussian_kernel
(ksize: int, sigma: float)
custom_utils/losses.py:14
↓ 2 callersMethodget_params
(self, low)
custom_utils/data_loaders/mit5k.py:67
↓ 2 callersFunctionimread
(path)
tools/measure.py:131
↓ 2 callersMethodlaplacian_kernel
(self, current)
custom_utils/losses.py:152
↓ 2 callersFunctionload_data
(train_dir, train_patchsize, val_dir, val_patchsize, train_batch_size, val_batch_size, shuffle=True, num_worke
tools/train.py:136
↓ 2 callersFunctionmkdir
(path)
custom_utils/dir_utils.py:14
↓ 2 callersMethodoutput_features
(self, x)
model/loss.py:147
↓ 2 callersFunctionsample_patch
Sample an image patch. args: im: Image pos: center position of crop sample_sz: size to crop output_sz: size to re
custom_utils/preprocessing.py:55
↓ 2 callersMethodshow_image
(self, data=None)
custom_utils/visdom.py:221
↓ 2 callersFunctionshow_image_with_boxes
(im, boxes, iou_pred=None, disp_ids=None)
custom_utils/plotting.py:63
↓ 2 callersFunctiont
(img)
tools/measure.py:110
↓ 1 callersMethodFLIP_aug
(self, low, high)
custom_utils/data_loaders/exposure.py:53
↓ 1 callersMethodFLIP_aug
(self, low, high)
custom_utils/data_loaders/mit5k.py:55
↓ 1 callersFunctionRUN_CUDA
(B, T, C, w, u, k, v)
model/modules/urwkv.py:90
↓ 1 callersMethod__init__
(self)
custom_utils/WT/transform.py:41
↓ 1 callersMethod_apply_convolutions
Applies convolution operations with different kernel sizes.
model/modules/LAN.py:71
↓ 1 callersMethod_apply_linear_layers
Apply Linear layers to match the feature channels with C.
model/modules/LAN.py:82
↓ 1 callersMethod_build_lr
(self, start_lr, steps, mult, epochs)
custom_utils/lr_scheduler.py:93
↓ 1 callersMethod_build_lr
(self, start_lr, end_lr, epochs)
custom_utils/lr_scheduler.py:121
↓ 1 callersMethod_gap_and_pad_features
Applies GAP to input features and pads channels to match the maximum.
model/modules/LAN.py:51
↓ 1 callersMethod_init_weights
(self, init_mode)
model/modules/urwkv.py:160
↓ 1 callersMethod_init_weights
(self, init_mode)
model/modules/urwkv.py:276
↓ 1 callersMethod_initialize_weights
(self)
model/modules/SSF.py:18
↓ 1 callersMethod_initialize_weights
(self)
model/modules/SSF.py:55
↓ 1 callersMethod_initialize_weights
(self)
model/modules/LAN.py:47
↓ 1 callersFunction_niqe_extract_subband_feats
(mscncoefs)
custom_utils/metrics/niqe.py:124
↓ 1 callersFunction_pascal_color_map
Python implementation of the color map function for the PASCAL VOC data set. Official Matlab version can be found in the PASCAL VOC devkit
custom_utils/plotting.py:88
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
model/loss.py:71
↓ 1 callersMethodaugData
(self,data,target)
custom_utils/UHD_load/train_data_aug_local.py:53
↓ 1 callersMethodbackward
(ctx, gy)
model/modules/urwkv.py:56
↓ 1 callersFunctionbrisque
(im)
custom_utils/metrics/brisque.py:120
↓ 1 callersFunctioncalculate_mscn
(dis_image)
custom_utils/metrics/piqe.py:6
↓ 1 callersFunctioncalculate_mscn
(dis_image)
custom_utils/metrics/brisque.py:83
↓ 1 callersFunctioncenterSurDev
(Block, blockSize)
custom_utils/metrics/piqe.py:84
↓ 1 callersMethodcompute_zero_padding
Computes zero padding.
custom_utils/losses.py:81
↓ 1 callersMethoddraw_data
(self)
custom_utils/visdom.py:97
↓ 1 callersMethoddraw_data
(self)
custom_utils/visdom.py:322
↓ 1 callersMethoddraw_data
(self)
custom_utils/visdom.py:363
↓ 1 callersMethoddraw_grid
(self, data)
custom_utils/visdom.py:194
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for E
model/modules/drop.py:4
↓ 1 callersFunctiondwt_init
(x)
custom_utils/WT/transform.py:8
↓ 1 callersFunctionformat_result
(psnr, ssim, lpips, mae, niqe)
tools/measure.py:135
↓ 1 callersMethodforward
(self, sr, hr)
model/loss.py:45
↓ 1 callersFunctiongauss_fcn
(x)
custom_utils/losses.py:8
↓ 1 callersFunctiongaussian
(window_size, sigma)
model/loss.py:59
↓ 1 callersFunctiongaussian
(window_size, sigma)
custom_utils/losses.py:7
↓ 1 callersFunctiongen_gauss_window
(lw, sigma)
custom_utils/metrics/niqe.py:88
↓ 1 callersMethodgenerate_display_text
(self, data)
custom_utils/visdom.py:244
↓ 1 callersFunctionget_gaussian_kernel2d
(ksize: Tuple[int, int], sigma: Tuple[float, float])
custom_utils/losses.py:21
↓ 1 callersMethodget_images
(self, index)
custom_utils/UHD_load/train_data_aug_local.py:36
↓ 1 callersMethodget_images
(self, index)
custom_utils/UHD_load/val_data.py:27
↓ 1 callersMethodget_images
(self, index)
custom_utils/UHD_load/val_data_train.py:24
↓ 1 callersMethodget_params
(self, low)
custom_utils/data_loaders/exposure.py:64
↓ 1 callersFunctionget_patches_test_features
(img, patch_size, stride=8)
custom_utils/metrics/niqe.py:144
↓ 1 callersFunctionggd_features
(imdata)
custom_utils/metrics/brisque.py:60
↓ 1 callersFunctioniwt_init
(x)
custom_utils/WT/transform.py:22
↓ 1 callersMethodjit_func
(self, x, spatial_shift_list, patch_resolution)
model/modules/urwkv.py:198
↓ 1 callersFunctionload_checkpoint
(model, weights)
tools/test.py:50
↓ 1 callersMethodmeasure
(self, imgA, imgB)
tools/measure.py:46
↓ 1 callersFunctionmeasure_dirs
(dirA, dirB, dataset_name, txt_path, use_gpu, verbose=False)
tools/measure.py:138
↓ 1 callersFunctionnoiseCriterion
(Block, blockSize, blockVar)
custom_utils/metrics/piqe.py:71
↓ 1 callersFunctionnoticeDistCriterion
(Block, nSegments, blockSize, windowSize, blockImpairedThreshold, N)
custom_utils/metrics/piqe.py:30
↓ 1 callersFunctionnumpy_to_torch
(a: np.ndarray)
custom_utils/preprocessing.py:6
↓ 1 callersFunctionoverlay_mask
Overlay mask over image. Source: https://github.com/albertomontesg/davis-interactive/blob/master/davisinteractive/utils/visualization.py This
custom_utils/plotting.py:115
↓ 1 callersFunctionpaired_product
(new_im)
custom_utils/metrics/niqe.py:74
↓ 1 callersFunctionpaired_product
(new_im)
custom_utils/metrics/brisque.py:69
↓ 1 callersFunctionpopulate_train_list
(images_path, mode='train')
custom_utils/data_loaders/exposure.py:20
↓ 1 callersFunctionpopulate_train_list
(images_path, mode='train')
custom_utils/data_loaders/lol.py:139
↓ 1 callersFunctionpopulate_train_list
(images_path, mode='train')
custom_utils/data_loaders/mit5k.py:24
↓ 1 callersMethodsave_data
(self, data, **kwargs)
custom_utils/visdom.py:26
↓ 1 callersMethodset_zoom_pos
(self, slice_pos)
custom_utils/visdom.py:125
↓ 1 callersMethodshade_cell
(self, data)
custom_utils/visdom.py:207
next →1–100 of 288, ranked by callers