Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LujiaJin/Basis-Prediction-Networks-PyTorch
/ functions
Functions
53 in github.com/LujiaJin/Basis-Prediction-Networks-PyTorch
⨍
Functions
53
◇
Types & classes
14
↓ 9 callers
Method
__init__
(self, color=True, burst_length=8, blind_est=True, kernel_size=15, basis_size=90, upMode='bil
model.py:114
↓ 6 callers
Method
get_value
(self)
utils/train_utils.py:25
↓ 6 callers
Function
normalize
(Img)
utils/train_utils.py:141
↓ 6 callers
Function
torch2numpy
(tensor, gamma=None)
utils/train_utils.py:81
↓ 5 callers
Method
pad_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 callers
Function
calculate_psnr
(output_img, target_img)
utils/train_utils.py:90
↓ 4 callers
Function
calculate_ssim
(output_img, target_img)
utils/train_utils.py:103
↓ 4 callers
Method
pool_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 callers
Method
update
(self, val)
utils/train_utils.py:19
↓ 2 callers
Method
__len__
(self)
data_provider.py:80
↓ 2 callers
Function
_represent_int
(s)
utils/train_utils.py:42
↓ 2 callers
Function
calculate_pearsonr
(output_img, target_img)
utils/train_utils.py:130
↓ 2 callers
Function
calculate_rmse
(output_img, target_img)
utils/train_utils.py:118
↓ 2 callers
Function
load_checkpoint
(checkpoint_dir, best_or_latest='best')
utils/train_utils.py:50
↓ 2 callers
Function
setup_seed
(seed)
utils/train_utils.py:148
↓ 1 callers
Method
crop_random
(tensor, patch_size)
data_provider.py:52
↓ 1 callers
Function
evaluate
(model, device, global_step, data_loader, loss_func)
train_and_eval.py:242
↓ 1 callers
Function
exclude_too_small_images
(img_path_list, threshold)
utils/train_utils.py:181
↓ 1 callers
Function
is_image
(filename)
utils/train_utils.py:156
↓ 1 callers
Method
kernel_predict
return size: (batch_size, burst_length, kernel_size ** 2, color_channel, height, width)
model.py:205
↓ 1 callers
Function
random_crop
(im, num_patches, w, h=None)
utils/train_utils.py:161
↓ 1 callers
Function
read_config
(config_file, config_spec)
utils/train_utils.py:71
↓ 1 callers
Function
save_checkpoint
(state, is_best, checkpoint_dir, n_iter, max_keep=10)
utils/train_utils.py:29
↓ 1 callers
Function
test
(config, args)
train_and_eval.py:264
↓ 1 callers
Function
train
(config, args)
train_and_eval.py:48
↓ 1 callers
Function
train_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
Method
forward
(self, data)
model.py:16
Method
forward
(self, data)
model.py:34
Method
forward
(self, data)
model.py:55
Method
forward
(self, data)
model.py:71
Method
forward
compute the pred image according to core and frames :param data: [batch_size, burst_length, color_channel, height, width] :pa
model.py:83
Method
forward
forward and obtain pred image directly :param data_with_est: if not blind estimation, it is same as data :param data:
model.py:216
Method
forward
forward function of loss_func :param pred_burst: shape [batch_size, burst_length, color_channel, height, width] :param pred:
model.py:335
Method
forward
(self, pred, ground_truth)
model.py:360
Method
forward
:param global_step: int :param pred_burst: [batch_size, burst_length, color_channel, height, width] :param ground_truth: [bat
model.py:377
Method
forward
(self, img)
model.py:400