MCPcopy Create free account

hub / github.com/ZhendongWang6/Uformer / functions

Functions188 in github.com/ZhendongWang6/Uformer

⨍ Functions 188◇ Types & classes 38
Methodextra_repr
(self)
model.py:1051
Methodextra_repr
(self)
model.py:1266
Methodflops
(self, H, W)
model.py:49
Methodflops
(self, H, W)
model.py:103
Methodflops
(self, H, W)
model.py:176
Methodflops
(self)
model.py:273
Methodflops
(self)
model.py:305
Methodflops
(self)
model.py:338
Methodflops
(self, HW)
model.py:373
Methodflops
(self, q_L, kv_L=None)
model.py:412
Methodflops
(self, q_L, kv_L=None)
model.py:444
Methodflops
(self, H, W)
model.py:527
Methodflops
(self, q_num, kv_num)
model.py:600
Methodflops
(self, H, W)
model.py:644
Methodflops
(self, H, W)
model.py:687
Methodflops
(self, H, W)
model.py:748
Methodflops
(self, H, W)
model.py:773
Methodflops
(self, H, W)
model.py:802
Methodflops
(self, H, W)
model.py:838
Methodflops
(self)
model.py:991
Methodflops
(self)
model.py:1062
Methodflops
(self)
model.py:1307
Methodforward
(self, x)
losses.py:25
Methodforward
(self, x, y)
losses.py:48
Methodforward
(self, x)
model.py:29
Methodforward
(self, x, x_img)
model.py:73
Methodforward
(self, x)
model.py:97
Methodforward
(self, x)
model.py:140
Methodforward
(self, x)
model.py:214
Methodforward
(self, x, H=None, W=None)
model.py:236
Methodforward
(self, x)
model.py:264
Methodforward
(self, x)
model.py:293
Methodforward
(self, x)
model.py:325
Methodforward
(self, x)
model.py:367
Methodforward
(self, x, attn_kv=None)
model.py:394
Methodforward
(self, x, attn_kv=None)
model.py:431
Methodforward
(self, x, attn_kv=None, mask=None)
model.py:494
Methodforward
(self, x, attn_kv=None, mask=None)
model.py:567
Methodforward
(self, x)
model.py:636
Methodforward
(self, x)
model.py:666
Methodforward
(self, x)
model.py:739
Methodforward
(self, x)
model.py:765
Methodforward
(self, x)
model.py:795
Methodforward
(self, x)
model.py:828
Methodforward
(self, x, mask=None)
model.py:908
Methodforward
(self, x, mask=None)
model.py:1054
Methodforward
(self, x, mask=None)
model.py:1269
Methodforward
(self, inp)
utils/antialias.py:40
Methodforward
(self, inp)
utils/antialias.py:94
Functionfreeze
(model)
utils/model_utils.py:6
Functionget_arch
(opt)
utils/model_utils.py:56
Functionget_last_path
(path, session)
utils/dir_utils.py:16
Functionget_test_data
(rgb_dir, img_options=None)
dataset/dataset_denoise.py:148
Functionget_test_data
(rgb_dir, img_options=None)
dataset/dataset_motiondeblur.py:188
Functionget_test_data_SR
(rgb_dir)
utils/loader.py:18
Functionget_training_data
(rgb_dir, img_options)
dataset/dataset_denoise.py:139
Functionget_training_data
(rgb_dir, img_options)
dataset/dataset_motiondeblur.py:179
Functionget_validation_data
(rgb_dir)
dataset/dataset_denoise.py:144
Functionis_frozen
(model)
utils/model_utils.py:14
Functionis_image_file
(filename)
utils/image_utils.py:9
Functionis_numpy_file
(filename)
utils/image_utils.py:6
Functionis_pkl_file
(filename)
utils/image_utils.py:15
Functionload_checkpoint
(model, weights)
utils/model_utils.py:23
Functionload_checkpoint_multigpu
(model, weights)
utils/model_utils.py:36
Functionload_npy
(filepath)
utils/image_utils.py:27
Functionload_optim
(optimizer, weights)
utils/model_utils.py:50
Functionload_pkl
(filename_)
utils/image_utils.py:18
Functionload_start_epoch
(weights)
utils/model_utils.py:45
Functionmkdirs
(paths)
utils/dir_utils.py:5
Methodno_weight_decay
(self)
model.py:249
Methodno_weight_decay
(self)
model.py:1259
Methodno_weight_decay_keywords
(self)
model.py:1263
Functionproc
(filename)
test/test_realblur.py:144
Functionsave_checkpoint
(model_dir, state, session)
utils/model_utils.py:18
Functionsave_dict
(dict_, filename_)
utils/image_utils.py:23
Functionsave_files
(i)
generate_patches_SIDD.py:46
Functionsave_img
(filepath, img)
utils/image_utils.py:37
Functiontensor2uint
(img)
utils/caculate_psnr_ssim.py:6
Methodtransform0
(self, torch_tensor)
utils/dataset_utils.py:8
Methodtransform1
(self, torch_tensor)
utils/dataset_utils.py:10
Methodtransform2
(self, torch_tensor)
utils/dataset_utils.py:13
Methodtransform3
(self, torch_tensor)
utils/dataset_utils.py:16
Methodtransform4
(self, torch_tensor)
utils/dataset_utils.py:19
Methodtransform5
(self, torch_tensor)
utils/dataset_utils.py:22
Methodtransform6
(self, torch_tensor)
utils/dataset_utils.py:25
Methodtransform7
(self, torch_tensor)
utils/dataset_utils.py:28
Functiontv_loss
Calculates TV loss for an image `x`. Args: x: image, torch.Variable of torch.Tensor beta: See https://arxiv.org/abs/1412.
losses.py:8
Functionunfreeze
(model)
utils/model_utils.py:10
← previous101–188 of 188, ranked by callers