MCPcopy Create free account

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

Functions159 in github.com/cvlab-yonsei/PISR

⨍ Functions 159◇ Types & classes 18
↓ 19 callersFunctionquantize
(img, rgb_range)
utils/utils.py:23
↓ 17 callersFunctionfloat2uint8
(image, normalize=False, rgb_range=1, margin=1)
visualizers/visualizer_factory.py:15
↓ 7 callersMethod__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:52
↓ 7 callersMethodforward
(self, img1, img2)
utils/metrics.py:91
↓ 4 callersFunctionget_model
(config, model_type)
models/model_factory.py:9
↓ 3 callersMethod__init__
(self, scale, n_colors, d=56, s=12, m=4, fsrcnn_weight_init=False)
models/fsrcnn.py:8
↓ 3 callersMethod_get_imagepath_list
(self, hr_dir, lr_dir, scale, ext='png')
datasets/base_dataset.py:47
↓ 3 callersFunctioncreate_window
(window_size, channel)
utils/metrics.py:53
↓ 3 callersFunctionget_psnr
(sr, hr, scale, rgb_range, benchmark=False)
utils/metrics.py:9
↓ 2 callersMethod__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None, patch_size=192, a
datasets/base_dataset.py:256
↓ 2 callersMethod_get_index
(self, idx)
datasets/base_dataset.py:229
↓ 2 callersMethod_get_index
(self, idx)
datasets/base_dataset.py:356
↓ 2 callersMethod_init_data_paths
(self, img_paths, img_dir, bin_dir, reset, verbose)
datasets/base_dataset.py:147
↓ 2 callersMethod_init_data_paths
(self, img_paths, img_dir, bin_dir, reset, verbose)
datasets/base_dataset.py:291
↓ 2 callersFunction_merge_config
(src, dst)
utils/config.py:67
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average = True)
utils/metrics.py:60
↓ 2 callersFunctionaugment
(*img_pair, hflip=True, rot=True)
datasets/data_utils.py:54
↓ 2 callersMethodfreeze_modules
(self)
models/base.py:19
↓ 2 callersFunctionget_adaptation_layer
(in_channels, out_channels, adap_avg_pool)
models/vid_module.py:41
↓ 2 callersFunctionget_figure_basic
(rgb_range, scale)
visualizers/visualizer_factory.py:35
↓ 2 callersFunctionget_last_checkpoint
(checkpoint_dir)
utils/checkpoint.py:12
↓ 2 callersFunctionget_loss
(config)
losses/loss_factory.py:15
↓ 2 callersFunctionget_optimizer
(config, model)
optimizers/optimizer_factory.py:23
↓ 2 callersFunctionget_patch
(*img_pair, patch_size=96, scale=2)
datasets/data_utils.py:12
↓ 2 callersFunctionget_scheduler
(config, optimizer, last_epoch)
schedulers/scheduler_factory.py:30
↓ 2 callersFunctionget_train_dataloader
(config, transform=None)
datasets/dataset_factory.py:30
↓ 2 callersFunctionget_valid_dataloader
(config, transform=None)
datasets/dataset_factory.py:64
↓ 2 callersMethodget_vid_module_dict
(self)
models/base.py:46
↓ 2 callersFunctionget_visualizer
(config)
visualizers/visualizer_factory.py:161
↓ 2 callersMethodload_pretrained_model
(self)
models/base.py:27
↓ 2 callersFunctionnp2Tensor
(*img_pair, rgb_range=255)
datasets/data_utils.py:44
↓ 2 callersFunctionset_channel
(*img_pair, n_channels=3)
datasets/data_utils.py:29
↓ 1 callersFunction_augment
(img)
datasets/data_utils.py:59
↓ 1 callersMethod_check_and_save
(self, img_path, bin_path, reset=False, verbose=False)
datasets/base_dataset.py:181
↓ 1 callersMethod_check_and_save
(self, img_path, bin_path, reset=False, verbose=False)
datasets/base_dataset.py:325
↓ 1 callersMethod_convert2binarypath
(self, img_dir, bin_dir, paths, ext='pt')
datasets/base_dataset.py:157
↓ 1 callersMethod_convert2binarypath
(self, img_dir, bin_dir, paths, ext='pt')
datasets/base_dataset.py:301
↓ 1 callersFunction_get_default_config
()
utils/config.py:10
↓ 1 callersMethod_get_hr_and_lr_binary_dir
(self, base_dir)
datasets/base_dataset.py:172
↓ 1 callersMethod_get_hr_and_lr_binary_dir
(self, base_dir)
datasets/base_dataset.py:316
↓ 1 callersMethod_get_hr_and_lr_img_dir
(self, base_dir)
datasets/base_dataset.py:163
↓ 1 callersMethod_get_hr_and_lr_img_dir
(self, base_dir)
datasets/base_dataset.py:307
↓ 1 callersMethod_get_hr_and_lr_img_dir
(self, base_dir)
datasets/base_dataset.py:426
↓ 1 callersMethod_get_index
(self, idx)
datasets/base_dataset.py:415
↓ 1 callersMethod_get_lr_hr_filename
(self, idx)
datasets/base_dataset.py:201
↓ 1 callersMethod_init_repeat
(self)
datasets/base_dataset.py:140
↓ 1 callersMethod_init_repeat
(self)
datasets/base_dataset.py:284
↓ 1 callersMethod_initialize
(self)
datasets/base_dataset.py:110
↓ 1 callersMethod_initialize
(self)
datasets/base_dataset.py:272
↓ 1 callersMethod_initialize
(self)
datasets/base_dataset.py:389
↓ 1 callersMethod_load_file
(self, idx)
datasets/base_dataset.py:75
↓ 1 callersFunction_np2Tensor
(img)
datasets/data_utils.py:45
↓ 1 callersFunction_set_channel
(img)
datasets/data_utils.py:30
↓ 1 callersFunctioncount_parameters
(model)
step2_train_student.py:191
↓ 1 callersFunctioncount_parameters
(model)
step1_train_teacher.py:175
↓ 1 callersFunctioncount_parameters
(model)
evaluate.py:88
↓ 1 callersFunctionevaluate
(config, student_model, dataloaders, start_epoch)
evaluate.py:80
↓ 1 callersFunctionevaluate_single_epoch
(config, student_model, teacher_model, dataloader, criterion, epoch, writer,
step2_train_student.py:92
↓ 1 callersFunctionevaluate_single_epoch
(config, model, dataloader, criterion, epoch, writer, visualizer,
step1_train_teacher.py:84
↓ 1 callersFunctionevaluate_single_epoch
(config, student_model, dataloader_dict, eval_type)
evaluate.py:42
↓ 1 callersMethodfsrcnn_weight_init
(self, mean=0.0, std=0.001)
models/fsrcnn.py:50
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/metrics.py:48
↓ 1 callersFunctionget_encoder
(name, **params)
models/encoder.py:4
↓ 1 callersFunctionget_mean_and_variance
(in_channels, homoscedasticity)
models/vid_module.py:16
↓ 1 callersMethodget_patch
(self, lr, hr)
datasets/base_dataset.py:43
↓ 1 callersMethodget_patch
(self, lr, hr)
datasets/base_dataset.py:236
↓ 1 callersFunctionget_test_dataloader
(config, transform=None)
datasets/dataset_factory.py:93
↓ 1 callersFunctionget_test_dataset
(config, transform=None)
datasets/dataset_factory.py:77
↓ 1 callersFunctionget_train_dataset
(config, transform=None)
datasets/dataset_factory.py:10
↓ 1 callersFunctionget_transform
(config)
transforms/transform_factory.py:13
↓ 1 callersFunctionget_valid_dataset
(config, transform=None)
datasets/dataset_factory.py:44
↓ 1 callersFunctionget_vid_module_dict
(network, layers, homoscedasticities)
models/vid_module.py:3
↓ 1 callersFunctionmain
()
step2_train_student.py:248
↓ 1 callersFunctionmain
()
step1_train_teacher.py:217
↓ 1 callersFunctionmain
()
evaluate.py:122
↓ 1 callersFunctionparse_args
()
step2_train_student.py:240
↓ 1 callersFunctionparse_args
()
step1_train_teacher.py:209
↓ 1 callersFunctionparse_args
()
evaluate.py:114
↓ 1 callersFunctionrun
(config)
step2_train_student.py:195
↓ 1 callersFunctionrun
(config)
step1_train_teacher.py:179
↓ 1 callersFunctionrun
(config)
evaluate.py:92
↓ 1 callersFunctionset_model_weight_dirs
(config)
utils/config.py:94
↓ 1 callersFunctiontrain
(config, student_model, teacher_model, dataloaders, criterion, optimizer, scheduler, writer, visuali
step2_train_student.py:148
↓ 1 callersFunctiontrain
(config, model, dataloaders, criterion, optimizer, scheduler, writer, visualizer, start_epoch)
step1_train_teacher.py:136
↓ 1 callersFunctiontrain_single_epoch
(config, student_model, teacher_model, dataloader, criterion, optimizer, epoch, writer,
step2_train_student.py:42
↓ 1 callersFunctiontrain_single_epoch
(config, model, dataloader, criterion, optimizer, epoch, writer,
step1_train_teacher.py:38
↓ 1 callersFunctionvid_loss_fn
(mu, std, tl)
losses/loss_factory.py:46
Method__get_index__
(self, idx)
datasets/base_dataset.py:35
Method__getitem__
(self, idx)
datasets/base_dataset.py:66
Method__getitem__
(self, idx)
datasets/base_dataset.py:192
Method__init__
(self, window_size=11, size_average=True, channel=1)
utils/metrics.py:84
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, base_dir=None, name='')
datasets/base_dataset.py:16
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None, patch_size=192, a
datasets/base_dataset.py:93
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, base_dir=None, name='')
datasets/base_dataset.py:382
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:10
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:25
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:39
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:66
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:79
Method__init__
(self, scale=2, n_colors=1, rgb_range=255, batch_size=128, test_every=None,
datasets/dataset.py:93
next →1–100 of 159, ranked by callers