MCPcopy Create free account

hub / github.com/LujiaJin/Basis-Prediction-Networks-PyTorch / functions

Functions53 in github.com/LujiaJin/Basis-Prediction-Networks-PyTorch

↓ 9 callersMethod__init__
(self, color=True, burst_length=8, blind_est=True, kernel_size=15, basis_size=90, upMode='bil
model.py:114
↓ 6 callersMethodget_value
(self)
utils/train_utils.py:25
↓ 6 callersFunctionnormalize
(Img)
utils/train_utils.py:141
↓ 6 callersFunctiontorch2numpy
(tensor, gamma=None)
utils/train_utils.py:81
↓ 5 callersMethodpad_before_cat
Prevent the image dimensions in the encoder and the decoder from being different due to the odd image dimension, which will lead to sk
model.py:176
↓ 4 callersFunctioncalculate_psnr
(output_img, target_img)
utils/train_utils.py:90
↓ 4 callersFunctioncalculate_ssim
(output_img, target_img)
utils/train_utils.py:103
↓ 4 callersMethodpool_before_cat
In the decoder for basis, the features are pooled to 1*1 size and then enlarged by replication before skip concatenation.
model.py:187
↓ 3 callersMethodupdate
(self, val)
utils/train_utils.py:19
↓ 2 callersMethod__len__
(self)
data_provider.py:80
↓ 2 callersFunction_represent_int
(s)
utils/train_utils.py:42
↓ 2 callersFunctioncalculate_pearsonr
(output_img, target_img)
utils/train_utils.py:130
↓ 2 callersFunctioncalculate_rmse
(output_img, target_img)
utils/train_utils.py:118
↓ 2 callersFunctionload_checkpoint
(checkpoint_dir, best_or_latest='best')
utils/train_utils.py:50
↓ 2 callersFunctionsetup_seed
(seed)
utils/train_utils.py:148
↓ 1 callersMethodcrop_random
(tensor, patch_size)
data_provider.py:52
↓ 1 callersFunctionevaluate
(model, device, global_step, data_loader, loss_func)
train_and_eval.py:242
↓ 1 callersFunctionexclude_too_small_images
(img_path_list, threshold)
utils/train_utils.py:181
↓ 1 callersFunctionis_image
(filename)
utils/train_utils.py:156
↓ 1 callersMethodkernel_predict
return size: (batch_size, burst_length, kernel_size ** 2, color_channel, height, width)
model.py:205
↓ 1 callersFunctionrandom_crop
(im, num_patches, w, h=None)
utils/train_utils.py:161
↓ 1 callersFunctionread_config
(config_file, config_spec)
utils/train_utils.py:71
↓ 1 callersFunctionsave_checkpoint
(state, is_best, checkpoint_dir, n_iter, max_keep=10)
utils/train_utils.py:29
↓ 1 callersFunctiontest
(config, args)
train_and_eval.py:264
↓ 1 callersFunctiontrain
(config, args)
train_and_eval.py:48
↓ 1 callersFunctiontrain_initialization
(args, model, optimizer, scheduler, checkpoint_dir)
train_and_eval.py:13
Method__call__
(self, tensor)
data_provider.py:12
Method__call__
(self, tensor)
data_provider.py:22
Method__getitem__
(self, index)
data_provider.py:56
Method__init__
(self, p=0.5)
data_provider.py:9
Method__init__
(self, p=0.5)
data_provider.py:19
Method__init__
(self, config, eval)
data_provider.py:29
Method__init__
(self, in_ch, out_ch)
model.py:8
Method__init__
(self, in_ch, out_ch)
model.py:23
Method__init__
(self, in_ch, out_ch)
model.py:41
Method__init__
(self, in_ch, out_ch)
model.py:63
Method__init__
(self, kernel_size=15)
model.py:79
Method__init__
(self, coeff_basic=1.0, coeff_anneal=1.0, gradient_L1=True, alpha=0.9998, beta=100)
model.py:327
Method__init__
(self, gradient_L1=True)
model.py:354
Method__init__
(self, alpha=0.9998, beta=100)
model.py:370
Method__init__
(self, L1=True)
model.py:396
Method__init__
(self, n)
utils/train_utils.py:14
Method_init_weights
(m)
model.py:167
Methodforward
(self, data)
model.py:16
Methodforward
(self, data)
model.py:34
Methodforward
(self, data)
model.py:55
Methodforward
(self, data)
model.py:71
Methodforward
compute the pred image according to core and frames :param data: [batch_size, burst_length, color_channel, height, width] :pa
model.py:83
Methodforward
forward and obtain pred image directly :param data_with_est: if not blind estimation, it is same as data :param data:
model.py:216
Methodforward
forward function of loss_func :param pred_burst: shape [batch_size, burst_length, color_channel, height, width] :param pred:
model.py:335
Methodforward
(self, pred, ground_truth)
model.py:360
Methodforward
:param global_step: int :param pred_burst: [batch_size, burst_length, color_channel, height, width] :param ground_truth: [bat
model.py:377
Methodforward
(self, img)
model.py:400