Functions — github.com/RylonW/DocNLC

CodeHub MCPcopy Create free account

hub / github.com/RylonW/DocNLC / functions

Functions426 in github.com/RylonW/DocNLC

⨍ Functions 426◇ Types & classes 48

↓ 1 callersFunctionbgr2ycbcr

bgr version of rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]

data/util.py:255

↓ 1 callersFunctionblur

generate_dataset.py:130

↓ 1 callersFunctionbuild_model

Build model depending on its size Args: setting (str): model size (small/base/large) image_size (int, int): ihabe height and

models/multitask_ViT_model.py:96

↓ 1 callersFunctioncalculate_psnr

metrics/calculate_PSNR_SSIM.py:69

↓ 1 callersFunctioncalculate_ssim

calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]

metrics/calculate_PSNR_SSIM.py:104

↓ 1 callersFunctiongaussian

(window_size, sigma)

models/loss_new.py:19

↓ 1 callersFunctiongenerate_mutitask_corresponding_txt

Target line form : path_GT|path_back|path_nosie|path_watermark|path_blur|path_shadow

generate_dataset.py:323

↓ 1 callersMethodget_current_visuals

models/base_model.py:22

↓ 1 callersMethodget_grad_outputs

models/loss.py:60

↓ 1 callersFunctionget_img_path_randomly

generate_dataset.py:156

↓ 1 callersFunctionget_pad_layer

models/archs/arch_util.py:61

↓ 1 callersFunctionget_pad_layer_1d

models/archs/arch_util.py:116

↓ 1 callersFunctionget_patch

(input, target, patch_size, scale = 1, ix=-1, iy=-1)

data/SIEN_dataset.py:18

↓ 1 callersFunctionget_patch

(input, target, patch_size, scale = 1, ix=-1, iy=-1)

data/multitask_dataset.py:18

↓ 1 callersMethodget_target_label

(self, input, target_is_real)

models/loss.py:40

↓ 1 callersMethodget_weight

(self, input, reverse)

models/archs/arch_util.py:157

↓ 1 callersFunctiongkern

(kernlen=13, nsig=1.6)

utils/util.py:162

↓ 1 callersFunctionimresize

(img, scale, antialiasing=True)

data/util.py:385

↓ 1 callersFunctionis_image_file

data/util.py:18

↓ 1 callersMethodlatent_contrast

con_loss = (self.cor_mat(instance, ins1) + self.cor_mat(instance, ins2) + self.cor_mat(instance, ins3)

models/multitask_Barlow_model_new.py:273

↓ 1 callersMethodload_Pre

models/multitask_ViT_model.py:535

↓ 1 callersMethodload_Pre

models/multitask_docnc_model.py:616

↓ 1 callersMethodload_Pre

models/multitask_DeGAN_model.py:532

↓ 1 callersMethodload_Pre

models/multitask_Barlow_model_new.py:722

↓ 1 callersMethodload_Pre

models/multitask_DIAE_model.py:654

↓ 1 callersMethodload_Pre

models/multitask_SIEN_model.py:430

↓ 1 callersMethodload_Pre

models/multitask_BCDU_model.py:598

↓ 1 callersMethodload_Pre

models/SIEN_model.py:435

↓ 1 callersMethodload_Pre

models/multitask_UNet_model.py:529

↓ 1 callersFunctionload_image_train2

data/SIEN_dataset.py:107

↓ 1 callersFunctionload_image_train2

data/multitask_dataset.py:109

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

pre_train.py:20

↓ 1 callersFunctionmain

metrics/calculate_PSNR_SSIM.py:13

↓ 1 callersFunctionmean_channels

models/multitask_docnc_model.py:26

↓ 1 callersFunctionmean_channels

models/multitask_Barlow_model_new.py:27

↓ 1 callersFunctionmean_channels

models/archs/EnhanceN_arch.py:8

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_ViT_model.py:384

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_docnc_model.py:465

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_DeGAN_model.py:381

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_Barlow_model_new.py:566

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_DIAE_model.py:503

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_SIEN_model.py:279

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_BCDU_model.py:447

↓ 1 callersMethodmerge_image2

(self,splitted_images, h,w)

models/multitask_UNet_model.py:378

↓ 1 callersMethodoff_diagonal

models/multitask_Barlow_model_new.py:252

↓ 1 callersFunctionpolar2z

[summary] Takes a list of radii and angles (radians) and converts them into a corresponding list of complex numbers x + yi. [descripti

utils/motionblur.py:35

↓ 1 callersMethodprint_network

models/multitask_ViT_model.py:518

↓ 1 callersMethodprint_network

models/multitask_docnc_model.py:599

↓ 1 callersMethodprint_network

models/multitask_DeGAN_model.py:515

↓ 1 callersMethodprint_network

models/multitask_Barlow_model_new.py:705

↓ 1 callersMethodprint_network

models/multitask_DIAE_model.py:637

↓ 1 callersMethodprint_network

models/multitask_SIEN_model.py:413

↓ 1 callersMethodprint_network

models/multitask_BCDU_model.py:581

↓ 1 callersMethodprint_network

models/SIEN_model.py:418

↓ 1 callersMethodprint_network

models/multitask_UNet_model.py:512

↓ 1 callersFunctionread_img

read image by cv2 or from lmdb return: Numpy float32, HWC, BGR, [0,1]

data/util.py:75

↓ 1 callersMethodset_params_lr_zero

models/multitask_ViT_model.py:284

↓ 1 callersMethodset_params_lr_zero

models/multitask_docnc_model.py:342

↓ 1 callersMethodset_params_lr_zero

models/multitask_DeGAN_model.py:284

↓ 1 callersMethodset_params_lr_zero

models/multitask_Barlow_model_new.py:448

↓ 1 callersMethodset_params_lr_zero

models/multitask_DIAE_model.py:403

↓ 1 callersMethodset_params_lr_zero

models/multitask_SIEN_model.py:174

↓ 1 callersMethodset_params_lr_zero

models/multitask_BCDU_model.py:347

↓ 1 callersMethodset_params_lr_zero

models/SIEN_model.py:158

↓ 1 callersMethodset_params_lr_zero

models/multitask_UNet_model.py:281

↓ 1 callersFunctionshadow

(src, shawdow_src)

generate_dataset.py:111

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_ViT_model.py:373

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_docnc_model.py:454

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_DeGAN_model.py:370

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_Barlow_model_new.py:555

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_DIAE_model.py:492

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_SIEN_model.py:268

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_BCDU_model.py:436

↓ 1 callersMethodsplit2

(self,dataset,size,h,w)

models/multitask_UNet_model.py:367

↓ 1 callersMethodstart

utils/util.py:313

↓ 1 callersMethodtest

models/SIEN_model.py:301

↓ 1 callersMethodthreshold_control

models/multitask_docnc_model.py:346

↓ 1 callersFunctiontransform

data/SIEN_dataset.py:121

↓ 1 callersFunctiontransform

data/multitask_dataset.py:121

↓ 1 callersFunctionwatermark

generate_dataset.py:60

FunctionDUF_downsample

Downsamping with Gaussian kernel used in the DUF official code Args: x (Tensor, [B, T, C, H, W]): frames to be downsampled. scale

utils/util.py:152

models/multitask_ViT_model.py:475

models/multitask_docnc_model.py:556

models/multitask_DeGAN_model.py:472

models/multitask_Barlow_model_new.py:662

models/multitask_DIAE_model.py:594

models/multitask_SIEN_model.py:370

models/multitask_BCDU_model.py:538

models/SIEN_model.py:374

models/multitask_UNet_model.py:469

Method__getitem__

data/SIEN_dataset.py:156

Method__getitem__

data/multitask_dataset.py:165

(self, size: tuple = (100, 100), intensity: float=0)

utils/motionblur.py:78

(self, task_num=0, bar_width=50, start=True)

utils/util.py:296

(self, upscale_factor, data_augmentation, group_file, patch_size, black_edges_crop, hflip, rot, transform=tran

data/SIEN_dataset.py:142

(self, upscale_factor, data_augmentation, group_file, patch_size, black_edges_crop, hflip, rot, transform=tran

data/multitask_dataset.py:151

(self, eps=1e-6)

models/loss.py:8

(self, device=torch.device('cpu'))

models/loss.py:55

← previous next →101–200 of 426, ranked by callers