MCPcopy Create free account

hub / github.com/Mukosame/Zooming-Slow-Mo-CVPR-2020 / functions

Functions175 in github.com/Mukosame/Zooming-Slow-Mo-CVPR-2020

↓ 8 callersMethodload
(self)
codes/models/base_model.py:35
↓ 5 callersMethodbackward
(ctx, grad_output)
codes/models/modules/DCNv2/dcn_v2.py:33
↓ 4 callersMethod__init__
(self, nf=64, groups=8)
codes/models/modules/Sakuya_arch.py:157
↓ 4 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride, padding, dilation=1, deformable_groups
codes/models/modules/DCNv2/dcn_v2.py:88
↓ 4 callersMethod_read_img_mc
Return BGR, HWC, [0, 255], uint8
codes/data/Vimeo7_dataset.py:94
↓ 4 callersMethodsave
(self, label)
codes/models/base_model.py:32
↓ 4 callersFunctionsingle_forward
(model, imgs_in)
codes/test.py:68
↓ 3 callersFunctionssim
(img1, img2)
codes/utils/util.py:159
↓ 2 callersMethod__init__
(self, input_size, input_dim, hidden_dim, kernel_size, num_layers, batch_first=False, bias=Tr
codes/models/modules/convlstm.py:70
↓ 2 callersMethod_extend_for_multilayer
(param, num_layers)
codes/models/modules/convlstm.py:172
↓ 2 callersFunctioncalculate_weights_indices
(in_length, out_length, scale, kernel, kernel_width, antialiasing)
codes/data/util.py:220
↓ 2 callersFunctioncubic
(x)
codes/data/util.py:211
↓ 2 callersFunctionget_timestamp
()
codes/utils/util.py:50
↓ 2 callersFunctionimresize_np
(img, scale, antialiasing=True)
codes/data/util.py:275
↓ 2 callersFunctionlaplacian_pyramid
(img, kernel, max_levels=5)
codes/models/modules/loss.py:45
↓ 2 callersFunctionmkdir
(path)
codes/utils/util.py:54
↓ 1 callersFunctionOrderedYaml
yaml orderedDict support
codes/utils/util.py:26
↓ 1 callersMethod__init__
(self, optimizer, milestones, restarts=None, weights=None, gamma=0.1, clear_state=False, last
codes/models/lr_scheduler.py:9
↓ 1 callersMethod__init__
(self, max_levels=5, k_size=5, sigma=2.0)
codes/models/modules/loss.py:60
↓ 1 callersFunction_augment
(img)
codes/data/util.py:101
↓ 1 callersMethod_check_kernel_size_consistency
(kernel_size)
codes/models/modules/convlstm.py:166
↓ 1 callersMethod_ensure_memcached
(self)
codes/data/Vimeo7_dataset.py:86
↓ 1 callersMethod_get_init_lr
(self)
codes/models/base_model.py:45
↓ 1 callersMethod_get_max_bar_width
(self)
codes/utils/util.py:217
↓ 1 callersFunction_get_paths_from_images
get image path list from image folder
codes/data/util.py:22
↓ 1 callersFunction_get_paths_from_lmdb
get image path list from lmdb meta info
codes/data/util.py:35
↓ 1 callersMethod_init_hidden
(self, batch_size, tensor_size)
codes/models/modules/convlstm.py:158
↓ 1 callersMethod_init_hidden
(self, batch_size, tensor_size)
codes/models/modules/Sakuya_arch.py:272
↓ 1 callersMethod_init_lmdb
(self)
codes/data/Vimeo7_dataset.py:79
↓ 1 callersFunction_read_img_lmdb
read image from lmdb with key (w/ and w/o fixed size) size: (C, H, W) tuple
codes/data/util.py:62
↓ 1 callersMethod_read_img_mc_BGR
Read BGR channels separately and then combine for 1M limits in cluster
codes/data/Vimeo7_dataset.py:103
↓ 1 callersMethod_set_lr
set learning rate for warmup, lr_groups_l: list for lr_groups. each for a optimizer
codes/models/base_model.py:38
↓ 1 callersFunctionbgr2ycbcr
bgr version of rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
codes/data/util.py:151
↓ 1 callersFunctionbuild_gauss_kernel
(size=5, sigma=1.0, n_channels=1, cuda=False)
codes/models/modules/loss.py:21
↓ 1 callersFunctioncheck_gradient_dconv
()
codes/models/modules/DCNv2/test.py:70
↓ 1 callersFunctioncheck_gradient_dpooling
()
codes/models/modules/DCNv2/test.py:136
↓ 1 callersFunctioncheck_pooling_zero_offset
()
codes/models/modules/DCNv2/test.py:102
↓ 1 callersFunctioncheck_zero_offset
()
codes/models/modules/DCNv2/test.py:32
↓ 1 callersFunctioncombine_frames
(pathIn, pathOut, fps)
codes/utils/make_video.py:24
↓ 1 callersFunctionconv_gauss
convolve img with a gaussian kernel that has been built with build_gauss_kernel
codes/models/modules/loss.py:38
↓ 1 callersFunctionconv_identify
(weight, bias)
codes/models/modules/DCNv2/test.py:20
↓ 1 callersFunctioncreate_dataloader
(dataset, dataset_opt, opt, sampler)
codes/data/__init__.py:7
↓ 1 callersFunctioncreate_dataset
(dataset_opt)
codes/data/__init__.py:28
↓ 1 callersFunctioncreate_model
(opt)
codes/models/__init__.py:5
↓ 1 callersFunctionexample_dconv
()
codes/models/modules/DCNv2/test.py:171
↓ 1 callersFunctionexample_dpooling
()
codes/models/modules/DCNv2/test.py:185
↓ 1 callersFunctionexample_mdpooling
()
codes/models/modules/DCNv2/test.py:228
↓ 1 callersMethodfeed_data
(self, data)
codes/models/base_model.py:17
↓ 1 callersFunctiongaussian
(x)
codes/models/modules/loss.py:25
↓ 1 callersFunctiongenerate_mod_LR_bic
(up_scale, sourcedir, savedir)
codes/data_scripts/generate_mod_LR_bic.py:14
↓ 1 callersMethodget_current_learning_rate
(self)
codes/models/base_model.py:68
↓ 1 callersMethodget_current_log
(self)
codes/models/VideoSR_base_model.py:115
↓ 1 callersFunctionget_extensions
()
codes/models/modules/DCNv2/setup.py:18
↓ 1 callersMethodget_network_description
Get the string and total parameters of the network
codes/models/base_model.py:74
↓ 1 callersFunctioninit_dist
initialization for distributed training
codes/train.py:18
↓ 1 callersMethodinit_hidden
(self, batch_size, tensor_size)
codes/models/modules/convlstm.py:62
↓ 1 callersMethodinit_offset
(self)
codes/models/modules/DCNv2/dcn_v2.py:99
↓ 1 callersMethodinit_offset
(self)
codes/models/modules/DCNv2/dcn_v2.py:126
↓ 1 callersFunctioninitialize_weights
(net_l, scale=1)
codes/models/modules/module_util.py:7
↓ 1 callersFunctionis_image_file
(filename)
codes/data/util.py:18
↓ 1 callersMethodload
(self)
codes/models/VideoSR_base_model.py:138
↓ 1 callersMethodload_network
(self, load_path, network, strict=True)
codes/models/base_model.py:92
↓ 1 callersFunctionmain
()
codes/train.py:29
↓ 1 callersFunctionmain
()
codes/test.py:19
↓ 1 callersFunctionmain
()
codes/video_to_zsm.py:55
↓ 1 callersMethodoptimize_parameters
(self)
codes/models/base_model.py:20
↓ 1 callersMethodprint_network
(self)
codes/models/VideoSR_base_model.py:126
↓ 1 callersFunctionread_image
read one image from img_path Return img: HWC, BGR, [0,1], numpy
codes/utils/util.py:259
↓ 1 callersFunctionread_seq_imgs_by_list
read a sequence of images from the given list
codes/utils/util.py:276
↓ 1 callersMethodreset_parameters
(self)
codes/models/modules/DCNv2/dcn_v2.py:70
↓ 1 callersMethodresume_training
Resume the optimizers and schedulers for training
codes/models/base_model.py:117
↓ 1 callersMethodsave_network
(self, network, network_label, iter_label)
codes/models/base_model.py:82
↓ 1 callersMethodsave_training_state
Saves training state during training, which will be used for resuming
codes/models/base_model.py:104
↓ 1 callersMethodset_epoch
(self, epoch)
codes/data/data_sampler.py:67
↓ 1 callersFunctionsingle_forward
(model, imgs_in)
codes/video_to_zsm.py:88
↓ 1 callersMethodstart
(self)
codes/utils/util.py:226
↓ 1 callersFunctiontest_lmdb
(dataroot, dataset='vimeo7')
codes/data_scripts/create_lmdb_mp.py:127
↓ 1 callersMethodupdate
(self, msg='In progress...')
codes/utils/util.py:235
↓ 1 callersMethodupdate_learning_rate
(self, cur_iter, warmup_iter=-1)
codes/models/base_model.py:53
↓ 1 callersFunctionvimeo7
create lmdb for the Vimeo90K-7 frames dataset, each image with fixed size GT: [3, 256, 448] Only need the 4th frame currently, e.g., 00001
codes/data_scripts/create_lmdb_mp.py:28
FunctionPYBIND11_MODULE
codes/models/modules/DCNv2/src/vision.cpp:4
Method__getitem__
(self, index)
codes/data/Vimeo7_dataset.py:114
Method__init__
(self, task_num=0, bar_width=50, start=True)
codes/utils/util.py:208
Method__init__
(self, dataset, num_replicas=None, rank=None, ratio=100)
codes/data/data_sampler.py:30
Method__init__
(self, opt)
codes/data/Vimeo7_dataset.py:32
Method__init__
(self, opt)
codes/models/VideoSR_base_model.py:16
Method__init__
(self, opt)
codes/models/base_model.py:9
Method__init__
(self, optimizer, T_period, restarts=None, weights=None, eta_min=0, last_epoch=-1)
codes/models/lr_scheduler.py:35
Method__init__
(self, eps=1e-6)
codes/models/modules/loss.py:11
Method__init__
(self, nf=64)
codes/models/modules/module_util.py:40
Method__init__
Initialize ConvLSTM cell. Parameters ---------- input_size: (int, int) Height and width of input tensor
codes/models/modules/convlstm.py:8
Method__init__
(self, input_size, input_dim, hidden_dim, kernel_size, num_layers, batch_first=False, bias=Tr
codes/models/modules/convlstm.py:180
Method__init__
(self, nf=64, groups=8)
codes/models/modules/Sakuya_arch.py:19
Method__init__
(self, input_size, input_dim, hidden_dim, kernel_size, num_layers, front_RBs, groups, batch_f
codes/models/modules/Sakuya_arch.py:196
Method__init__
(self, input_size, input_dim, hidden_dim, kernel_size, num_layers, front_RBs, groups, batch_f
codes/models/modules/Sakuya_arch.py:277
Method__init__
(self, nf=64, nframes=3, groups=8, front_RBs=5, back_RBs=10)
codes/models/modules/Sakuya_arch.py:299
Method__init__
(self, in_channels, out_channels, kernel_size, stride, padding, dilation=1, deformable_groups
codes/models/modules/DCNv2/dcn_v2.py:54
Method__init__
(self, in_channels, out_channels, kernel_size, stride, padding, dilation=1, deformable_groups
codes/models/modules/DCNv2/dcn_v2.py:115
Method__init__
(self, spatial_scale, pooled_size, output_dim, no_trans, group_size=1, part_size=None, sample
codes/models/modules/DCNv2/dcn_v2.py:196
Method__init__
(self, spatial_scale, pooled_size, output_dim, no_trans, group_size=1, part_size=None, sample
codes/models/modules/DCNv2/dcn_v2.py:218
next →1–100 of 175, ranked by callers