MCPcopy Create free account

hub / github.com/Liyong8490/DnCNN-pytorch / functions

Functions30 in github.com/Liyong8490/DnCNN-pytorch

↓ 5 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=None, groups=1, bias=False)
models_DnCNN.py:5
↓ 1 callersMethod_make_layers
(self, block, num_chnl, kernel_size, padding, num_of_layers, groups=1)
models_DnCNN.py:80
↓ 1 callersMethod_make_layers
(self, block, kernel_size, num_chnl, num_of_layers, padding=1, groups=1, bias=False)
models_DnCNN.py:121
↓ 1 callersFunctionadjust_learning_rate
Sets the learning rate to the initial LR decayed by 10 every 10 epochs
train_DnCNN.py:151
↓ 1 callersFunctionc_psnr
(im1, im2)
utils.py:64
↓ 1 callersFunctionc_ssim
(im1, im2)
utils.py:71
↓ 1 callersFunctiondata_aug
(image, mode=0)
utils.py:37
↓ 1 callersFunctiondata_augmentation
(label, mode=0)
generate_data.py:9
↓ 1 callersFunctiondataparallel
(model, ngpus, gpu0=0)
train_DnCNN.py:170
↓ 1 callersFunctiondenoising
(path, imgName, model)
ValidateResult_DnCNN.py:34
↓ 1 callersFunctionmain
()
train_DnCNN.py:38
↓ 1 callersFunctionsave_checkpoint
(model, epoch)
train_DnCNN.py:159
↓ 1 callersFunctiontrain
(training_data_loader, optimizer, model, criterion, epoch)
train_DnCNN.py:96
FunctionIm2Patch
(img, win, stride=1)
generate_data.py:36
Method__getitem__
(self, index)
utils.py:29
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=None, groups=1, bias=True)
models_DnCNN.py:20
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=None, groups=1, bias=False)
models_DnCNN.py:37
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=None, groups=1, bias=False)
models_DnCNN.py:53
Method__init__
(self, input_chnl, output_chnl=None,inner_chnl=64, padding=1, num_of_layers=15, groups=1)
models_DnCNN.py:70
Method__init__
(self, input_chnl, groups=1)
models_DnCNN.py:90
Method__init__
(self, file_path, sigma)
utils.py:15
Method__init__
(self, N)
train_DnCNN.py:141
Method__len__
(self)
utils.py:33
Methodforward
(self, x)
models_DnCNN.py:16
Methodforward
(self, x)
models_DnCNN.py:32
Methodforward
(self, x)
models_DnCNN.py:49
Methodforward
(self, x)
models_DnCNN.py:66
Methodforward
(self, x)
models_DnCNN.py:86
Methodforward
(self, x)
models_DnCNN.py:127
Methodforward
(self, res, label)
train_DnCNN.py:146