Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Thmen/EGVSR
/ functions
Functions
408 in github.com/Thmen/EGVSR
⨍
Functions
408
◇
Types & classes
85
Method
forward_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
Method
forward_sequence
(self, data, args_dict)
codes/models/networks/tecogan_nets.py:494
Method
generate_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
Method
generate_dummy_input
(self, lr_size)
codes/models/networks/vespcn_nets.py:107
Method
generate_dummy_input
(self, lr_size)
codes/models/networks/tecogan_nets_bk.py:160
Method
generate_dummy_input
(self, lr_size)
codes/models/networks/tecogan_nets.py:161
Method
generate_dummy_input
(self, lr_size)
codes/models/networks/espcn_nets.py:31
Method
generate_dummy_input
(self, lr_size)
codes/models/networks/sofvsr_nets.py:209
Method
get_current_errors
(self)
codes/metrics/LPIPS/models/dist_model.py:167
Method
get_current_errors
(self)
codes/metrics/LPIPS/models/base_model.py:30
Method
get_current_errors
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:210
Method
get_current_errors
(self)
codes/official_metrics/LPIPSmodels/base_model.py:33
Method
get_current_log
(self)
codes/models/base_model.py:70
Method
get_current_visuals
(self)
codes/metrics/LPIPS/models/dist_model.py:176
Method
get_current_visuals
(self)
codes/metrics/LPIPS/models/base_model.py:27
Method
get_current_visuals
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:219
Method
get_current_visuals
(self)
codes/official_metrics/LPIPSmodels/base_model.py:30
Method
get_image_paths
(self)
codes/metrics/LPIPS/models/base_model.py:21
Method
get_image_paths
(self)
codes/official_metrics/LPIPSmodels/base_model.py:24
Method
get_lr
(self)
codes/models/optim/lr_schedules.py:24
Function
grab_patch
(img_in, P, yy, xx)
codes/official_metrics/LPIPSmodels/util.py:162
Function
hook_fn_forward
(module, input, output)
codes/metrics/model_summary.py:26
Method
infer
Function of inference Parameters: :param lr_data: a rgb video sequence with shape thwc :return: a rgb vid
codes/models/espcn_model.py:31
Method
infer
(self, lr_data)
codes/models/sofvsr_model.py:35
Method
infer
(self, data)
codes/models/base_model.py:36
Method
infer
(self, lr_data)
codes/models/vespcn_model.py:55
Method
infer_sequence
infer for a whole sequence (for inference)
codes/models/networks/base_nets.py:24
Method
infer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/vespcn_nets.py:118
Method
infer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/tecogan_nets_bk.py:256
Method
infer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/tecogan_nets.py:257
Method
infer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/espcn_nets.py:61
Method
infer_sequence
Parameters: :param lr_data: torch.FloatTensor in shape tchw :param device: torch.device
codes/models/networks/sofvsr_nets.py:220
Function
info
Print methods and doc strings. Takes module, class, list, dictionary, or string.
codes/official_metrics/LPIPSmodels/util.py:208
Method
initialize
(self, use_gpu=True, gpu_ids=[0])
codes/metrics/LPIPS/models/base_model.py:14
Method
initialize
(self, use_gpu=True)
codes/official_metrics/LPIPSmodels/base_model.py:16
Function
initialize_weights
Modify from BasicSR/MMSR
codes/utils/net_utils.py:9
Function
l2
(p0, p1, range=255.)
codes/metrics/LPIPS/models/__init__.py:47
Function
l2
(p0, p1, range=255.)
codes/official_metrics/LPIPSmodels/util.py:44
Function
load_image
(path)
codes/official_metrics/LPIPSmodels/util.py:165
Method
load_network
(self, network, network_label, epoch_label)
codes/metrics/LPIPS/models/base_model.py:43
Method
load_network
(self, network, network_label, epoch_label)
codes/official_metrics/LPIPSmodels/base_model.py:46
Function
mkdirs
(paths)
codes/official_metrics/LPIPSmodels/util.py:242
Function
montage
( imgs, PAD=5, RATIO=16 / 9., EXTRA_PAD=( False, False), MM=-1,
codes/official_metrics/LPIPSmodels/util.py:260
Method
name
(self)
codes/metrics/LPIPS/models/base_model.py:11
Method
name
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:25
Method
name
(self)
codes/official_metrics/LPIPSmodels/base_model.py:13
Function
normalize_tensor
(in_feat,eps=1e-10)
codes/metrics/LPIPS/models/__init__.py:43
Function
on_mouse
(event, x, y, flags, img)
scripts/selectROI.py:25
Method
optimize_parameters
(self)
codes/metrics/LPIPS/models/dist_model.py:116
Method
optimize_parameters
(self)
codes/metrics/LPIPS/models/base_model.py:24
Method
optimize_parameters
(self)
codes/official_metrics/LPIPSmodels/dist_model.py:162
Method
optimize_parameters
(self)
codes/official_metrics/LPIPSmodels/base_model.py:27
Function
prep_display_image
(img, dtype='uint8')
codes/official_metrics/LPIPSmodels/util.py:201
Function
print_network
(net)
codes/metrics/LPIPS/models/networks_basic.py:182
Function
print_network
(net)
codes/official_metrics/LPIPSmodels/networks_basic.py:263
Function
print_numpy
(x, val=True, shp=False)
codes/official_metrics/LPIPSmodels/util.py:231
Function
print_options
(opt, logger, tab='')
codes/utils/base_utils.py:35
Function
psnr
(p0, p1, peak=255.)
codes/metrics/LPIPS/models/__init__.py:50
Function
psnr
(p0, p1, peak=255.)
codes/official_metrics/LPIPSmodels/util.py:47
Function
rand_flip
(input1,input2)
codes/official_metrics/LPIPSmodels/util.py:38
Function
random_swap
(obj0,obj1)
codes/official_metrics/LPIPSmodels/util.py:415
Function
read_csv_file_as_text
(in_path)
codes/official_metrics/LPIPSmodels/util.py:405
Function
read_file
(in_path,list_lines=False)
codes/official_metrics/LPIPSmodels/util.py:387
Function
read_text_file
(in_path)
codes/official_metrics/LPIPSmodels/util.py:22
Function
register_hook
(module)
codes/metrics/model_summary.py:32
Function
resize_image
(img, max_size=256)
codes/official_metrics/LPIPSmodels/util.py:180
Function
resize_image_zoom
(img, zoom_factor=1., order=3)
codes/official_metrics/LPIPSmodels/util.py:190
Function
rgb2lab
(in_img,mean_cent=False)
codes/metrics/LPIPS/models/__init__.py:56
Function
rgb2lab
(in_img,mean_cent=False)
codes/official_metrics/LPIPSmodels/util.py:54
Function
rgb2yCbCr
(input_im)
codes/utils/data_utils.py:66
Function
rgb_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
Method
save
(self, path, label)
codes/metrics/LPIPS/models/dist_model.py:191
Method
save
(self, current_iter)
codes/models/vsrgan_model.py:314
Method
save_done
(self, flag=False)
codes/metrics/LPIPS/models/base_model.py:55
Method
save_done
(self, flag=False)
codes/official_metrics/LPIPSmodels/base_model.py:59
Function
save_image
(image_numpy, image_path, )
codes/official_metrics/LPIPSmodels/util.py:196
Function
save_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
Method
save_training_state
(self, current_epoch, current_iter)
codes/models/base_model.py:84
Function
score_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
Function
score_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
Function
score_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
Function
score_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
Method
set_criterion
(self)
codes/models/base_model.py:30
Method
set_input
(self, data)
codes/metrics/LPIPS/models/dist_model.py:128
Method
set_input
(self, data)
codes/official_metrics/LPIPSmodels/dist_model.py:174
Method
set_network
(self)
codes/models/base_model.py:24
Method
set_network
(self)
codes/models/vsrgan_model.py:26
Function
setup_logger
(name)
codes/utils/base_utils.py:18
Function
setup_paths
(opt, mode)
codes/utils/base_utils.py:71
Function
setup_random_seed
(seed)
codes/utils/base_utils.py:10
Function
tensor2tensorlab
(image_tensor,to_norm=True,mc_only=False)
codes/metrics/LPIPS/models/__init__.py:71
Function
tensor2tensorlab
(image_tensor,to_norm=True,mc_only=False)
codes/official_metrics/LPIPSmodels/util.py:97
Function
tensor2vec
(vector_tensor)
codes/metrics/LPIPS/models/__init__.py:117
Function
tensor2vec
(vector_tensor)
codes/official_metrics/LPIPSmodels/util.py:147
Function
tensorlab2tensor
(lab_tensor,return_inbnd=False)
codes/metrics/LPIPS/models/__init__.py:85
Function
tensorlab2tensor
(lab_tensor,return_inbnd=False)
codes/official_metrics/LPIPSmodels/util.py:113
Function
test
(opt)
codes/torch2onnx.py:14
Method
train
(self, data)
codes/models/base_model.py:33
Method
train
Function for mini-batch training Parameters: :param data: a batch of training tensor with shape NTCHW
codes/models/vsrgan_model.py:120
Method
update_learning_rate
(self,nepoch_decay)
codes/metrics/LPIPS/models/dist_model.py:198
← previous
next →
301–400 of 408, ranked by callers