Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JavierGurrola/RDUNet
/ types & classes
Types & classes
17 in github.com/JavierGurrola/RDUNet
⨍
Functions
52
◇
Types & classes
17
↓ 14 callers
Class
DenoisingBlock
model.py:78
↓ 3 callers
Class
DownsampleBlock
model.py:25
↓ 3 callers
Class
UpsampleBlock
model.py:35
↓ 2 callers
Class
AdditiveWhiteGaussianNoise
Additive white gaussian noise generator.
transforms.py:6
↓ 2 callers
Class
NoisyImagesDataset
data_management.py:70
↓ 2 callers
Class
RDUNet
r""" Residual-Dense U-net for image denoising.
model.py:106
↓ 1 callers
Class
DataSampler
data_management.py:38
↓ 1 callers
Class
EpochLogger
r""" Keeps a log of metrics in the current epoch.
train.py:12
↓ 1 callers
Class
FileLogger
Keeps a log of the whole training and validation process. The results are recorded in a CSV files. Args: file_path (string): pat
train.py:52
↓ 1 callers
Class
InputBlock
model.py:50
↓ 1 callers
Class
OutputBlock
model.py:64
↓ 1 callers
Class
PSNR
r""" Evaluates the PSNR metric in a tensor. It can return a result with different reduction methods. Args: data_range (int, float
metrics.py:5
↓ 1 callers
Class
RandomHorizontalFlip
transforms.py:77
↓ 1 callers
Class
RandomRot90
transforms.py:95
↓ 1 callers
Class
RandomVerticalFlip
transforms.py:59
↓ 1 callers
Class
SSIM
r""" Evaluates the SSIM metric in a tensor. It can return a result with different reduction methods. Args: channels (int): Number
metrics.py:36
↓ 1 callers
Class
ToTensor
Convert data sample to pytorch tensor
transforms.py:47