MCPcopy Create free account

hub / github.com/cvlab-yonsei/PISR / functions

Functions159 in github.com/cvlab-yonsei/PISR

⨍ Functions 159◇ Types & classes 18
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=1, base_dir=None, name='')
datasets/dataset.py:106
Method__init__
(self)
models/base.py:11
Method__init__
(self, scale, n_colors, d=56, s=12, m=4, k=1, encoder='inv_fsrcnn')
models/fsrcnn.py:67
Method__init__
(self, scale, n_colors, d=56, s=12, m=4, vid_info=None, modules_to_freeze=None, initialize_fro
models/fsrcnn.py:93
Method__init__
(self, scale, n_colors, d=56, s=12, m=4, k=1, vid_info=None, modules_to_freeze=None, initial
models/fsrcnn.py:131
Method__len__
(self)
datasets/base_dataset.py:31
Method__len__
(self)
datasets/base_dataset.py:223
Method__len__
(self)
datasets/base_dataset.py:350
Method__len__
(self)
datasets/base_dataset.py:411
Method_get_hr_and_lr_img_dir
(self, base_dir)
datasets/base_dataset.py:39
Method_initialize
(self)
datasets/base_dataset.py:27
Method_load_file
(self, idx)
datasets/base_dataset.py:209
Method_load_file
(self, idx)
datasets/base_dataset.py:336
Method_load_file
(self, idx)
datasets/base_dataset.py:399
Functionadam
(parameters, betas=(0.9, 0.999), weight_decay=0, amsgrad=False, **_)
optimizers/optimizer_factory.py:11
Functionadjust_learning_rate
(config, epoch)
step2_train_student.py:37
Functionadjust_learning_rate
(config, epoch)
step1_train_teacher.py:34
Functionadjust_learning_rate
(config, epoch)
evaluate.py:37
Functionbase_transform
()
transforms/transform_factory.py:22
Functioncheck_same
(stride)
utils/utils.py:30
Functioncopy_last_n_checkpoints
(config, n, name, model_type)
utils/checkpoint.py:34
Functioncosine
(optimizer, last_epoch, T_max=50, eta_min=0.00001, **_)
schedulers/scheduler_factory.py:23
Methodforward
(self, x)
models/fsrcnn.py:62
Methodforward
(self, x)
models/fsrcnn.py:87
Methodforward
(self, LR, HR=None, teacher_pred_dict=None)
models/fsrcnn.py:113
Methodforward
(self, HR, LR=None)
models/fsrcnn.py:150
Functionget_checkpoint
(config, name, model_type='teacher')
utils/checkpoint.py:28
Functionget_figure
(LR, HR, pred, pred_teacher=None)
visualizers/visualizer_factory.py:61
Functionget_figure_basic_fn
(LR, HR, pred)
visualizers/visualizer_factory.py:36
Functionget_fsrcnn_student
(scale, n_colors, **kwargs)
models/fsrcnn.py:174
Functionget_fsrcnn_teacher
(scale, n_colors, **kwargs)
models/fsrcnn.py:170
Functionget_initial_checkpoint
(config, model_type)
utils/checkpoint.py:22
Functionget_lcscc_encoder
(scale=2, d=56, s=12, k=1, n_colors=1)
models/encoder.py:9
Methodget_patch
(self, lr, hr)
datasets/base_dataset.py:362
Methodget_patch
(self, lr, hr)
datasets/base_dataset.py:419
Functionget_ssim
(pred, hr, scale, rgb_range, benchmark=False)
utils/metrics.py:29
Functionhook
(module, input, output)
utils/utils.py:50
Functionl1loss
(reduction='mean', **_)
losses/loss_factory.py:23
Functionload
(config_path)
utils/config.py:80
Functionload_checkpoint
(model, optimizer, checkpoint, model_type)
utils/checkpoint.py:48
Functionloss_fn
(pred_dict, HR, **_)
losses/loss_factory.py:26
Functionmulti_step
(optimizer, last_epoch, milestones=[500, 5000], gamma=0.1, **_)
schedulers/scheduler_factory.py:12
Functionnone
(optimizer, last_epoch, **_)
schedulers/scheduler_factory.py:19
Functionprepare_train_directories
(config, model_type)
utils/utils.py:18
Functionreceptive_field
:parameter 'input_size': tuple of (Channel, Height, Width) :return OrderedDict of `Layername`->OrderedDict of receptive field stats {'j'
utils/utils.py:36
Functionreceptive_field_for_unit
Utility function to calculate the receptive field for a specific unit in a layer using the dictionary calculated above :parameter
utils/utils.py:171
Functionregister_hook
(module)
utils/utils.py:48
Functionsave_checkpoint
(config, model, optimizer, epoch, step, model_type, weights_dict=None, name=None)
utils/checkpoint.py:63
Functionsgd
(parameters, momentum=0.9, weight_decay=0, nesterov=True, **_)
optimizers/optimizer_factory.py:19
Functionsigma_visualizer
(rgb_range, scale)
visualizers/visualizer_factory.py:60
Functionssim
(img1, img2, window_size = 11, size_average = True)
utils/metrics.py:109
Functionstep
(optimizer, last_epoch, step_size=80, gamma=0.1, **_)
schedulers/scheduler_factory.py:8
Functionstep0_visualizer
(rgb_range, scale)
visualizers/visualizer_factory.py:89
Functionstep1_visualizer
(rgb_range, scale)
visualizers/visualizer_factory.py:93
Functionstep2_attention_visualizer
(rgb_range, scale)
visualizers/visualizer_factory.py:128
Functionstep2_visualizer
(rgb_range, scale)
visualizers/visualizer_factory.py:97
Functionteacher_LR_constraint_loss
(reduction='mean', lambda1=1, lambda2=1, **_)
losses/loss_factory.py:86
Functiontransform_fn
(lr, hr, hflip=None, vflip=None, rot90=None, count=None)
transforms/transform_factory.py:24
Functionvid_loss
(reduction='mean', lambda1=1, lambda2=1, epsilon=1e-8, pdf='gaussian', **_)
losses/loss_factory.py:40
← previous101–159 of 159, ranked by callers