Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/FZU-N/URWKV
/ functions
Functions
288 in github.com/FZU-N/URWKV
⨍
Functions
288
◇
Types & classes
59
Method
cv_ui_handler
(self, data)
custom_utils/visdom.py:158
Method
draw_data
(self)
custom_utils/visdom.py:52
Method
draw_data
(self)
custom_utils/visdom.py:64
Method
draw_data
(self)
custom_utils/visdom.py:150
Method
draw_data
(self)
custom_utils/visdom.py:235
Method
draw_data
(self)
custom_utils/visdom.py:261
Method
draw_data
(self)
custom_utils/visdom.py:274
Method
draw_data
(self)
custom_utils/visdom.py:286
Method
extra_repr
(self)
model/modules/drop.py:35
Function
forward
model/modules/cuda/wkv_op.cpp:6
Method
forward
(self, img1, img2)
model/loss.py:103
Method
forward
(self, X, Y)
model/loss.py:129
Method
forward
(self, pred_im, gt)
model/loss.py:155
Method
forward
(self, x, encode_list, inter_feat)
model/decoder.py:51
Method
forward
(self, x, inter_feat)
model/encoder.py:34
Method
forward
(self, x)
model/model_builder.py:38
Method
forward
(self, x)
model/modules/drop.py:32
Method
forward
(self, x)
model/modules/SSF.py:26
Method
forward
(self, x, feat_list)
model/modules/SSF.py:62
Method
forward
(self, x)
model/modules/SSF.py:101
Method
forward
(ctx, B, T, C, w, u, k, v)
model/modules/urwkv.py:33
Method
forward
(self, x, spatial_shift_list, patch_resolution=None)
model/modules/urwkv.py:226
Method
forward
(self, x, channel_shift_list, patch_resolution=None)
model/modules/urwkv.py:294
Method
forward
(self, x)
model/modules/urwkv.py:342
Method
forward
(self, x, spatial_shift_list, channel_shift_list, inter_feat, patch_resolution=None)
model/modules/urwkv.py:375
Method
forward
(self, x, inter_feat)
model/modules/urwkv.py:493
Method
forward
(self, x, inter_feat, patch_resolution)
model/modules/LAN.py:92
Method
forward
(self, pred, target)
custom_utils/losses.py:49
Method
forward
(self, img1: torch.Tensor, img2: torch.Tensor)
custom_utils/losses.py:92
Method
forward
(self, x, y)
custom_utils/losses.py:132
Method
forward
(self, x, y)
custom_utils/losses.py:161
Method
forward
(self, x)
custom_utils/WT/transform.py:45
Method
forward
(self, x)
custom_utils/WT/transform.py:54
Function
freeze
(model)
custom_utils/model_utils.py:6
Method
get_cur_lr
(self)
custom_utils/lr_scheduler.py:35
Function
get_gaussian_kernel
(kernel_size=21, sigma=5, channels=3)
custom_utils/GaussianBlur.py:4
Function
get_last_path
(path, session)
custom_utils/dir_utils.py:19
Method
get_lr
(self)
custom_utils/warmup_scheduler/scheduler.py:25
Function
get_patches_train_features
(img, patch_size, stride=8)
custom_utils/metrics/niqe.py:140
Function
ggd_features
(imdata)
custom_utils/metrics/niqe.py:65
Function
hsl2rgb_torch
(hsl: torch.Tensor)
custom_utils/image_utils.py:146
Function
hsv2rgb_torch
(hsv: torch.Tensor)
custom_utils/image_utils.py:127
Function
is_frozen
(model)
custom_utils/model_utils.py:16
Function
load_checkpoint
(model, weights)
custom_utils/model_utils.py:27
Function
load_checkpoint_multigpu
(model, weights)
custom_utils/model_utils.py:40
Function
load_img
(filepath)
custom_utils/image_utils.py:54
Function
load_img
(filepath)
custom_utils/data_loaders/lol.py:16
Function
load_optim
(optimizer, weights)
custom_utils/model_utils.py:56
Function
load_pretrain
(model, pretrained_path)
custom_utils/model_load.py:47
Function
load_start_epoch
(weights)
custom_utils/model_utils.py:50
Method
lpips
(self, imgA, imgB, model=None)
tools/measure.py:63
Method
mae
(self, imgA, imgB)
tools/measure.py:70
Function
mkdirs
(paths)
custom_utils/dir_utils.py:6
Function
network_parameters
(nets)
custom_utils/model_utils.py:63
Function
niqe
(inputImgData)
custom_utils/metrics/niqe.py:200
Method
niqe
(self, gt, enhance)
tools/measure.py:84
Function
numpyPSNR
(tar_img, prd_img)
custom_utils/image_utils.py:76
Function
pad_input
(net_input, local_window_dim=2)
custom_utils/img_resize.py:13
Function
piqe
(im)
custom_utils/metrics/piqe.py:107
Function
plot_graph
Plot graph. Data is a 1D tensor. args: fig_num: Figure number. title: Title of figure.
custom_utils/plotting.py:45
Method
psnr
(self, imgA, imgB)
tools/measure.py:50
Function
q_shift
(input, shift_pixel=1, gamma=1/4, patch_resolution=None)
model/modules/urwkv.py:93
Method
register
(self, data, mode, debug_level=0, title='Data', **kwargs)
custom_utils/visdom.py:403
Function
restore_from
(model, optimizer, ckpt_path)
custom_utils/model_load.py:73
Function
rgb2hsl_torch
(rgb: torch.Tensor)
custom_utils/image_utils.py:87
Function
rgb2hsv_torch
(rgb: torch.Tensor)
custom_utils/image_utils.py:111
Function
sample_patch_multiscale
Extract image patches at multiple scales. args: im: Image. pos: Center position for extraction. scales: Image scales to ex
custom_utils/preprocessing.py:33
Function
sample_patch_transformed
Extract transformed image samples. args: im: Image. pos: Center position for extraction. scale: Image scale to extract fea
custom_utils/preprocessing.py:14
Function
save_checkpoint
(model_dir, state, session)
custom_utils/model_utils.py:21
Method
save_data
(self, data)
custom_utils/visdom.py:48
Method
save_data
(self, data)
custom_utils/visdom.py:60
Method
save_data
(self, data)
custom_utils/visdom.py:83
Method
save_data
(self, data)
custom_utils/visdom.py:146
Method
save_data
(self, data)
custom_utils/visdom.py:229
Method
save_data
(self, data)
custom_utils/visdom.py:257
Method
save_data
(self, data)
custom_utils/visdom.py:271
Method
save_data
(self, data)
custom_utils/visdom.py:283
Method
save_data
(self, data)
custom_utils/visdom.py:301
Method
save_data
(self, data)
custom_utils/visdom.py:357
Function
save_img
(filepath, img)
custom_utils/image_utils.py:73
Function
show_tensor
Display a 2D tensor. args: fig_num: Figure number. title: Title of figure.
custom_utils/plotting.py:13
Method
ssim
(self, imgA, imgB, gray_scale=False)
tools/measure.py:55
Method
tensor_to_image
(self,tensor)
custom_utils/UHD_load/train_data_aug_local.py:96
Function
torchSSIM
(tar_img, prd_img)
custom_utils/image_utils.py:70
Function
torch_to_numpy
(a: torch.Tensor)
custom_utils/preprocessing.py:10
Function
unfreeze
(model)
custom_utils/model_utils.py:11
Function
validation
(model, val_loader)
custom_utils/validation.py:39
Function
validation_shadow
(model, val_loader)
custom_utils/validation.py:67
← previous
201–288 of 288, ranked by callers