MCPcopy Create free account

hub / github.com/RylonW/DocNLC / functions

Functions426 in github.com/RylonW/DocNLC

⨍ Functions 426◇ Types & classes 48
↓ 29 callersMethodlrelu
(self, x)
models/archs/EnhanceN_arch.py:167
↓ 18 callersMethodload_network
(self, load_path, network, strict=True)
models/base_model.py:84
↓ 18 callersMethodlrelu
(self, x)
models/multitask_docnc_model.py:190
↓ 18 callersMethodlrelu
(self, x)
models/multitask_DeGAN_model.py:129
↓ 18 callersMethodlrelu
(self, x)
models/multitask_Barlow_model_new.py:186
↓ 18 callersMethodlrelu
(self, x)
models/multitask_UNet_model.py:129
↓ 18 callersMethodsave_network
(self, network, network_label, iter_label)
models/base_model.py:74
↓ 13 callersMethodsave
(self, label)
models/base_model.py:31
↓ 10 callersMethodcor_mat
(self, z1, z2)
models/multitask_Barlow_model_new.py:258
↓ 9 callersMethodget_network_description
Get the string and total parameters of the network
models/base_model.py:68
↓ 9 callersFunctionpsnr_np
(enhanced, image_dslr)
metrics/calculate_PSNR_SSIM.py:149
↓ 8 callersMethodload
(self)
models/base_model.py:34
↓ 6 callersFunctioncreate_dataloader
(dataset, dataset_opt, opt=None, sampler=None)
data/__init__.py:7
↓ 6 callersFunctioncreate_dataset
(opt,dataset_opt)
data/__init__.py:25
↓ 6 callersFunctionget_image_ldr
(img)
data/multitask_dataset.py:82
↓ 5 callersMethodfeed_data
(self, data)
models/base_model.py:16
↓ 4 callersMethod__init__
(self, in_channels, channels)
models/multitask_Barlow_model_new.py:42
↓ 4 callersMethod__init__
(self, input_dim=3, output_dim=3, num_filter=64, frame_size=(256, 256), bidirectional=False, norm='instance')
models/multitask_BCDU_model.py:127
↓ 4 callersFunctioncalculate_weights_indices
(in_length, out_length, scale, kernel, kernel_width, antialiasing)
data/util.py:330
↓ 4 callersFunctionget_weight_init_fn
get weight_initialization function according to activation_fn Notes ------------------------------------- if activation_fn requires argume
models/archs/layer.py:122
↓ 4 callersFunctionsingle_forward
PyTorch model forward (single test), it is just a simple warpper Args: model (PyTorch model) inp (Tensor): inputs defined by the m
utils/util.py:186
↓ 3 callersMethod__init__
(self, id, requires_grad=False)
models/loss_new.py:96
↓ 3 callersMethod_createKernel
[summary] Finds a kernel (psf) of given intensity. [description] use displayKernel to actually see the kernel. Keywor
utils/motionblur.py:233
↓ 3 callersFunctionconv
pytorch torch.nn.Conv2d wrapper Notes --------------------------------------------------------------------- Arguments: activation_
models/archs/layer.py:147
↓ 3 callersFunctioncreate_model
(opt)
models/__init__.py:5
↓ 3 callersFunctioncreate_window
(window_size, channel)
models/loss_new.py:24
↓ 3 callersMethodload
(self)
models/multitask_ViT_model.py:529
↓ 3 callersMethodload
(self)
models/multitask_docnc_model.py:610
↓ 3 callersMethodload
(self)
models/multitask_DeGAN_model.py:526
↓ 3 callersMethodload
(self)
models/multitask_Barlow_model_new.py:716
↓ 3 callersMethodload
(self)
models/multitask_DIAE_model.py:648
↓ 3 callersMethodload
(self)
models/multitask_SIEN_model.py:424
↓ 3 callersMethodload
(self)
models/multitask_BCDU_model.py:592
↓ 3 callersMethodload
(self)
models/SIEN_model.py:429
↓ 3 callersMethodload
(self)
models/multitask_UNet_model.py:523
↓ 3 callersFunctionnoisy
(image, sigma = 0.05)
generate_dataset.py:14
↓ 3 callersMethodresume_training
Resume the optimizers and schedulers for training
models/base_model.py:111
↓ 3 callersMethodsave_best
(self,name)
models/SIEN_model.py:451
↓ 3 callersMethodsave_training_state
Save training state during training, which will be used for resuming
models/base_model.py:100
↓ 3 callersFunctionssim
(img1, img2)
utils/util.py:247
↓ 3 callersFunctionssim
(img1, img2)
metrics/calculate_PSNR_SSIM.py:81
↓ 2 callersMethod__init__
(self, gan_type, real_label_val=1.0, fake_label_val=0.0)
models/loss.py:20
↓ 2 callersMethod__init__
(self)
models/multitask_docnc_model.py:83
↓ 2 callersMethod__init__
(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)
models/archs/arch_util.py:20
↓ 2 callersMethod__init__
(self, input_chans, num_features, filter_size, num_layers=1)
models/archs/layer.py:69
↓ 2 callersFunction_augment
(img)
data/util.py:172
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
models/loss_new.py:31
↓ 2 callersMethodapplyTo
[summary] Applies kernel to one of the following: 1. Path to image file 2. Pillow image object 3. (H,W,3)-shaped nump
utils/motionblur.py:323
↓ 2 callersFunctionbgr2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
metrics/calculate_PSNR_SSIM.py:125
↓ 2 callersMethodcompute_M
(self, step)
models/SIEN_model.py:383
↓ 2 callersFunctioncubic
(x)
data/util.py:321
↓ 2 callersMethodfeed_val_data
(self, data, need_GT=True)
models/SIEN_model.py:207
↓ 2 callersMethodget_current_learning_rate
(self)
models/base_model.py:65
↓ 2 callersMethodget_current_log
(self)
models/SIEN_model.py:406
↓ 2 callersMethodget_gradient
(self, src)
models/loss_new.py:180
↓ 2 callersMethodget_gradient_hist
(self, gradient_x, gradient_y)
models/loss_new.py:189
↓ 2 callersFunctionget_image_ldr
(img)
data/SIEN_dataset.py:78
↓ 2 callersMethodget_response
(self, gradient, mean)
models/loss_new.py:174
↓ 2 callersFunctionget_timestamp
()
utils/util.py:43
↓ 2 callersMethodget_val_current_visuals
(self, need_GT=True)
models/SIEN_model.py:364
↓ 2 callersMethodmerge_image2
(self,splitted_images, h,w)
models/SIEN_model.py:228
↓ 2 callersFunctionmkdir
(path)
utils/util.py:47
↓ 2 callersMethodoptimize_parameters
(self)
models/base_model.py:19
↓ 2 callersFunctionpair
(t)
models/multitask_DIAE_model.py:37
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_ViT_model.py:548
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_docnc_model.py:629
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_DeGAN_model.py:545
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_Barlow_model_new.py:735
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_DIAE_model.py:667
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_SIEN_model.py:443
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_BCDU_model.py:611
↓ 2 callersMethodsave
(self, iter_label)
models/SIEN_model.py:448
↓ 2 callersMethodsave
(self, iter_label)
models/multitask_UNet_model.py:542
↓ 2 callersMethodsave_M
(self,name)
models/SIEN_model.py:441
↓ 2 callersMethodsplit2
(self,dataset,size,h,w)
models/SIEN_model.py:217
↓ 2 callersMethodupdate_learning_rate
(self, cur_iter, warmup_iter=-1)
models/base_model.py:51
↓ 2 callersMethodval
(self)
models/SIEN_model.py:238
↓ 1 callersFunctionBGR2RGB_toTensor
(inputs, target)
data/SIEN_dataset.py:126
↓ 1 callersFunctionBGR2RGB_toTensor
(back, blur, noise, shadow, watermark, target)
data/multitask_dataset.py:126
↓ 1 callersFunctionOrderedYaml
yaml orderedDict support
utils/util.py:23
↓ 1 callersMethod__init__
( self, *, encoder, decoder_dim, decoder_depth = 1, decoder_he
models/multitask_ViT_model.py:42
↓ 1 callersMethod__init__
(self)
models/multitask_DeGAN_model.py:27
↓ 1 callersMethod__init__
( self, *, encoder, image_size, patch_size, dim, decod
models/multitask_DIAE_model.py:41
↓ 1 callersMethod__init__
(self, optimizer, milestones, restarts=None, weights=None, gamma=0.1, clear_state=False, last
models/lr_scheduler.py:9
↓ 1 callersMethod__init__
(self)
models/multitask_UNet_model.py:27
↓ 1 callersMethod__init__
(self, in_channels, channels)
models/archs/EnhanceN_arch.py:23
↓ 1 callersMethod__init__
(self, feature_layer=34, use_bn=False, use_input_norm=True, device=torch.device('cpu'))
models/archs/discriminator_vgg_arch.py:63
↓ 1 callersFunction_augment
(inputs,target)
data/SIEN_dataset.py:54
↓ 1 callersFunction_augment
(inputs,target)
data/multitask_dataset.py:55
↓ 1 callersFunction_augment_flow
(flow)
data/util.py:199
↓ 1 callersMethod_createPath
[summary] creates a motion blur path with the given intensity. [description] Proceede in 5 steps 1. Get a random numbe
utils/motionblur.py:111
↓ 1 callersMethod_get_init_lr
Get the initial lr, which is set by the scheduler
models/base_model.py:44
↓ 1 callersMethod_get_max_bar_width
(self)
utils/util.py:304
↓ 1 callersFunction_get_paths_from_images
get image path list from image folder
data/util.py:22
↓ 1 callersFunction_get_paths_from_lmdb
get image path list from lmdb meta info
data/util.py:35
↓ 1 callersFunction_read_img_lmdb
read image from lmdb with key (w/ and w/o fixed size) size: (C, H, W) tuple
data/util.py:64
↓ 1 callersMethod_set_lr
Set learning rate for warmup lr_groups_l: list for lr_groups. each for a optimizer
models/base_model.py:37
↓ 1 callersFunctionalpha_mix
(fore, back)
generate_dataset.py:28
↓ 1 callersFunctionaugment
(inputs, target, hflip, rot)
data/SIEN_dataset.py:49
↓ 1 callersFunctionaugment
(inputs, target, hflip, rot)
data/multitask_dataset.py:50
next →1–100 of 426, ranked by callers