MCPcopy Create free account

hub / github.com/Thmen/EGVSR / functions

Functions408 in github.com/Thmen/EGVSR

Methodforward_sequence
:param data: should be either hr_data or gt_data :param args_dict: a dict including data/config needed here
codes/models/networks/tecogan_nets.py:366
Methodforward_sequence
(self, data, args_dict)
codes/models/networks/tecogan_nets.py:494
Methodgenerate_dummy_input
use for compute per-step FLOPs and speed return random tensors that can be taken as input of <forward>
codes/models/networks/base_nets.py:8
Methodgenerate_dummy_input
(self, lr_size)
codes/models/networks/vespcn_nets.py:107
Methodgenerate_dummy_input
(self, lr_size)
codes/models/networks/tecogan_nets_bk.py:160
Methodgenerate_dummy_input
(self, lr_size)
codes/models/networks/tecogan_nets.py:161
Methodgenerate_dummy_input
(self, lr_size)
codes/models/networks/espcn_nets.py:31
Methodgenerate_dummy_input
(self, lr_size)
codes/models/networks/sofvsr_nets.py:209
Methodget_current_errors
(self)
codes/metrics/LPIPS/models/dist_model.py:167
Methodget_current_errors
(self)
codes/metrics/LPIPS/models/base_model.py:30
Methodget_current_errors
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:210
Methodget_current_errors
(self)
codes/official_metrics/LPIPSmodels/base_model.py:33
Methodget_current_log
(self)
codes/models/base_model.py:70
Methodget_current_visuals
(self)
codes/metrics/LPIPS/models/dist_model.py:176
Methodget_current_visuals
(self)
codes/metrics/LPIPS/models/base_model.py:27
Methodget_current_visuals
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:219
Methodget_current_visuals
(self)
codes/official_metrics/LPIPSmodels/base_model.py:30
Methodget_image_paths
(self)
codes/metrics/LPIPS/models/base_model.py:21
Methodget_image_paths
(self)
codes/official_metrics/LPIPSmodels/base_model.py:24
Methodget_lr
(self)
codes/models/optim/lr_schedules.py:24
Functiongrab_patch
(img_in, P, yy, xx)
codes/official_metrics/LPIPSmodels/util.py:162
Functionhook_fn_forward
(module, input, output)
codes/metrics/model_summary.py:26
Methodinfer
Function of inference Parameters: :param lr_data: a rgb video sequence with shape thwc :return: a rgb vid
codes/models/espcn_model.py:31
Methodinfer
(self, lr_data)
codes/models/sofvsr_model.py:35
Methodinfer
(self, data)
codes/models/base_model.py:36
Methodinfer
(self, lr_data)
codes/models/vespcn_model.py:55
Methodinfer_sequence
infer for a whole sequence (for inference)
codes/models/networks/base_nets.py:24
Methodinfer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/vespcn_nets.py:118
Methodinfer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/tecogan_nets_bk.py:256
Methodinfer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/tecogan_nets.py:257
Methodinfer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/espcn_nets.py:61
Methodinfer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/sofvsr_nets.py:220
Functioninfo
Print methods and doc strings. Takes module, class, list, dictionary, or string.
codes/official_metrics/LPIPSmodels/util.py:208
Methodinitialize
(self, use_gpu=True, gpu_ids=[0])
codes/metrics/LPIPS/models/base_model.py:14
Methodinitialize
(self, use_gpu=True)
codes/official_metrics/LPIPSmodels/base_model.py:16
Functioninitialize_weights
Modify from BasicSR/MMSR
codes/utils/net_utils.py:9
Functionl2
(p0, p1, range=255.)
codes/metrics/LPIPS/models/__init__.py:47
Functionl2
(p0, p1, range=255.)
codes/official_metrics/LPIPSmodels/util.py:44
Functionload_image
(path)
codes/official_metrics/LPIPSmodels/util.py:165
Methodload_network
(self, network, network_label, epoch_label)
codes/metrics/LPIPS/models/base_model.py:43
Methodload_network
(self, network, network_label, epoch_label)
codes/official_metrics/LPIPSmodels/base_model.py:46
Functionmkdirs
(paths)
codes/official_metrics/LPIPSmodels/util.py:242
Functionmontage
( imgs, PAD=5, RATIO=16 / 9., EXTRA_PAD=( False, False), MM=-1,
codes/official_metrics/LPIPSmodels/util.py:260
Methodname
(self)
codes/metrics/LPIPS/models/base_model.py:11
Methodname
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:25
Methodname
(self)
codes/official_metrics/LPIPSmodels/base_model.py:13
Functionnormalize_tensor
(in_feat,eps=1e-10)
codes/metrics/LPIPS/models/__init__.py:43
Functionon_mouse
(event, x, y, flags, img)
scripts/selectROI.py:25
Methodoptimize_parameters
(self)
codes/metrics/LPIPS/models/dist_model.py:116
Methodoptimize_parameters
(self)
codes/metrics/LPIPS/models/base_model.py:24
Methodoptimize_parameters
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:162
Methodoptimize_parameters
(self)
codes/official_metrics/LPIPSmodels/base_model.py:27
Functionprep_display_image
(img, dtype='uint8')
codes/official_metrics/LPIPSmodels/util.py:201
Functionprint_network
(net)
codes/metrics/LPIPS/models/networks_basic.py:182
Functionprint_network
(net)
codes/official_metrics/LPIPSmodels/networks_basic.py:263
Functionprint_numpy
(x, val=True, shp=False)
codes/official_metrics/LPIPSmodels/util.py:231
Functionprint_options
(opt, logger, tab='')
codes/utils/base_utils.py:35
Functionpsnr
(p0, p1, peak=255.)
codes/metrics/LPIPS/models/__init__.py:50
Functionpsnr
(p0, p1, peak=255.)
codes/official_metrics/LPIPSmodels/util.py:47
Functionrand_flip
(input1,input2)
codes/official_metrics/LPIPSmodels/util.py:38
Functionrandom_swap
(obj0,obj1)
codes/official_metrics/LPIPSmodels/util.py:415
Functionread_csv_file_as_text
(in_path)
codes/official_metrics/LPIPSmodels/util.py:405
Functionread_file
(in_path,list_lines=False)
codes/official_metrics/LPIPSmodels/util.py:387
Functionread_text_file
(in_path)
codes/official_metrics/LPIPSmodels/util.py:22
Functionregister_hook
(module)
codes/metrics/model_summary.py:32
Functionresize_image
(img, max_size=256)
codes/official_metrics/LPIPSmodels/util.py:180
Functionresize_image_zoom
(img, zoom_factor=1., order=3)
codes/official_metrics/LPIPSmodels/util.py:190
Functionrgb2lab
(in_img,mean_cent=False)
codes/metrics/LPIPS/models/__init__.py:56
Functionrgb2lab
(in_img,mean_cent=False)
codes/official_metrics/LPIPSmodels/util.py:54
Functionrgb2yCbCr
(input_im)
codes/utils/data_utils.py:66
Functionrgb_to_ycbcr
Coefficients are taken from the official codes of DUF-VSR This conversion is also the same as that in BasicSR Parameters:
codes/utils/data_utils.py:31
Methodsave
(self, path, label)
codes/metrics/LPIPS/models/dist_model.py:191
Methodsave
(self, current_iter)
codes/models/vsrgan_model.py:314
Methodsave_done
(self, flag=False)
codes/metrics/LPIPS/models/base_model.py:55
Methodsave_done
(self, flag=False)
codes/official_metrics/LPIPSmodels/base_model.py:59
Functionsave_image
(image_numpy, image_path, )
codes/official_metrics/LPIPSmodels/util.py:196
Functionsave_sequence
Save each frame of a sequence to .png image in seq_dir Parameters: :param seq_dir: dir to save results :param seq_da
codes/utils/data_utils.py:109
Methodsave_training_state
(self, current_epoch, current_iter)
codes/models/base_model.py:84
Functionscore_2afc_dataset
Function computes Two Alternative Forced Choice (2AFC) score using distance function 'func' in dataset 'data_loader' INPUTS data_
codes/metrics/LPIPS/models/dist_model.py:208
Functionscore_2afc_dataset
Function computes Two Alternative Forced Choice (2AFC) score using distance function 'func' in dataset 'data_loader' INPUTS data_
codes/official_metrics/LPIPSmodels/dist_model.py:250
Functionscore_jnd_dataset
Function computes JND score using distance function 'func' in dataset 'data_loader' INPUTS data_loader - CustomDatasetDataLoader object -
codes/metrics/LPIPS/models/dist_model.py:243
Functionscore_jnd_dataset
Function computes JND score using distance function 'func' in dataset 'data_loader' INPUTS data_loader - CustomDatasetDataLoader object -
codes/official_metrics/LPIPSmodels/dist_model.py:287
Methodset_criterion
(self)
codes/models/base_model.py:30
Methodset_input
(self, data)
codes/metrics/LPIPS/models/dist_model.py:128
Methodset_input
(self, data)
codes/official_metrics/LPIPSmodels/dist_model.py:174
Methodset_network
(self)
codes/models/base_model.py:24
Methodset_network
(self)
codes/models/vsrgan_model.py:26
Functionsetup_logger
(name)
codes/utils/base_utils.py:18
Functionsetup_paths
(opt, mode)
codes/utils/base_utils.py:71
Functionsetup_random_seed
(seed)
codes/utils/base_utils.py:10
Functiontensor2tensorlab
(image_tensor,to_norm=True,mc_only=False)
codes/metrics/LPIPS/models/__init__.py:71
Functiontensor2tensorlab
(image_tensor,to_norm=True,mc_only=False)
codes/official_metrics/LPIPSmodels/util.py:97
Functiontensor2vec
(vector_tensor)
codes/metrics/LPIPS/models/__init__.py:117
Functiontensor2vec
(vector_tensor)
codes/official_metrics/LPIPSmodels/util.py:147
Functiontensorlab2tensor
(lab_tensor,return_inbnd=False)
codes/metrics/LPIPS/models/__init__.py:85
Functiontensorlab2tensor
(lab_tensor,return_inbnd=False)
codes/official_metrics/LPIPSmodels/util.py:113
Functiontest
(opt)
codes/torch2onnx.py:14
Methodtrain
(self, data)
codes/models/base_model.py:33
Methodtrain
Function for mini-batch training Parameters: :param data: a batch of training tensor with shape NTCHW
codes/models/vsrgan_model.py:120
Methodupdate_learning_rate
(self,nepoch_decay)
codes/metrics/LPIPS/models/dist_model.py:198
← previousnext →301–400 of 408, ranked by callers