Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Liyong8490/DnCNN-pytorch
/ functions
Functions
30 in github.com/Liyong8490/DnCNN-pytorch
⨍
Functions
30
◇
Types & classes
8
↓ 5 callers
Method
__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=None, groups=1, bias=False)
models_DnCNN.py:5
↓ 1 callers
Method
_make_layers
(self, block, num_chnl, kernel_size, padding, num_of_layers, groups=1)
models_DnCNN.py:80
↓ 1 callers
Method
_make_layers
(self, block, kernel_size, num_chnl, num_of_layers, padding=1, groups=1, bias=False)
models_DnCNN.py:121
↓ 1 callers
Function
adjust_learning_rate
Sets the learning rate to the initial LR decayed by 10 every 10 epochs
train_DnCNN.py:151
↓ 1 callers
Function
c_psnr
(im1, im2)
utils.py:64
↓ 1 callers
Function
c_ssim
(im1, im2)
utils.py:71
↓ 1 callers
Function
data_aug
(image, mode=0)
utils.py:37
↓ 1 callers
Function
data_augmentation
(label, mode=0)
generate_data.py:9
↓ 1 callers
Function
dataparallel
(model, ngpus, gpu0=0)
train_DnCNN.py:170
↓ 1 callers
Function
denoising
(path, imgName, model)
ValidateResult_DnCNN.py:34
↓ 1 callers
Function
main
()
train_DnCNN.py:38
↓ 1 callers
Function
save_checkpoint
(model, epoch)
train_DnCNN.py:159
↓ 1 callers
Function
train
(training_data_loader, optimizer, model, criterion, epoch)
train_DnCNN.py:96
Function
Im2Patch
(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
Method
forward
(self, x)
models_DnCNN.py:16
Method
forward
(self, x)
models_DnCNN.py:32
Method
forward
(self, x)
models_DnCNN.py:49
Method
forward
(self, x)
models_DnCNN.py:66
Method
forward
(self, x)
models_DnCNN.py:86
Method
forward
(self, x)
models_DnCNN.py:127
Method
forward
(self, res, label)
train_DnCNN.py:146