Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JavierGurrola/RDUNet
/ functions
Functions
52 in github.com/JavierGurrola/RDUNet
⨍
Functions
52
◇
Types & classes
17
↓ 5 callers
Method
__init__
(self, **kwargs)
model.py:110
↓ 2 callers
Function
mod_crop
Crops image according to mod to restore spatial dimensions adequately in the decoding sections of the model. :param image: numpy array
utils.py:30
↓ 1 callers
Function
build_ensemble
Create image ensemble to estimate denoised image. :param image: numpy array Noisy image. :param normalize: bool Normalize
utils.py:85
↓ 1 callers
Function
create_patches
(image, patch_size, step)
data_management.py:30
↓ 1 callers
Function
fit_model
Training of the denoiser model. :param model: torch Module Neural network to fit. :param data_loaders: dict Dictionary wi
train.py:104
↓ 1 callers
Method
get_log
Returns the average of the monitored metrics in the current moment, given the number of evaluated samples. :param n_samples:
train.py:33
↓ 1 callers
Method
load_dataset
(self, files)
data_management.py:92
↓ 1 callers
Function
load_image
Load image and change it color space from RGB to Grayscale if necessary. :param image_path: str Path of the image. :param channel
utils.py:7
↓ 1 callers
Function
main
()
main_train.py:17
↓ 1 callers
Function
mod_pad
Pads image according to mod to restore spatial dimensions adequately in the decoding sections of the model. :param image: numpy array
utils.py:49
↓ 1 callers
Function
predict
(model, noisy_dataset, gt_dataset, device, padding, n_channels, results_path)
main_test.py:14
↓ 1 callers
Function
predict_ensemble
Predict batch of images from an ensemble. :param model: torch Module Trained model to estimate denoised images. :param ensemble:
utils.py:153
↓ 1 callers
Function
separate_ensemble
Apply inverse transforms to predicted image ensemble and average them. :param ensemble: list Predicted images, ensemble[0] is the ori
utils.py:116
↓ 1 callers
Function
set_seed
Sets all random seeds. :param seed: int Seed value. :return: None
utils.py:73
↓ 1 callers
Method
update_log
Update the metrics in the current epoch, this method is called at every step of the epoch. :param metrics: dict Metrics t
train.py:21
Method
__call__
Updates the CSV record file. :param epoch_log: dict Log of the current epoch. :return: None
train.py:72
Method
__call__
Generates additive white gaussian noise, and it is applied to the clean image. :param sample: :return:
transforms.py:16
Method
__call__
(self, sample)
transforms.py:49
Method
__call__
(self, sample)
transforms.py:64
Method
__call__
(self, sample)
transforms.py:82
Method
__call__
(self, sample)
transforms.py:100
Method
__call__
(self, outputs, targets)
metrics.py:23
Method
__call__
(self, outputs, targets)
metrics.py:54
Method
__getitem__
(self, idx)
data_management.py:83
Method
__init__
(self)
train.py:16
Method
__init__
Creates the csv record file. :param f
train.py:60
Method
__init__
(self, noise_level, fix_sigma=False, clip=False)
transforms.py:8
Method
__init__
(self, p=0.5)
transforms.py:61
Method
__init__
(self, p=0.5)
transforms.py:79
Method
__init__
(self, p=0.5)
transforms.py:97
Method
__init__
(self, data_source, num_samples=None)
data_management.py:40
Method
__init__
(self, files, channels, patch_size, transform=None, noise_transform=None)
data_management.py:71
Method
__init__
(self, data_range, reduction='none', eps=1e-8)
metrics.py:18
Method
__init__
(self, channels, data_range, reduction='none')
metrics.py:49
Method
__init__
(self, in_channels, out_channels)
model.py:26
Method
__init__
(self, in_channels, cat_channels, out_channels)
model.py:36
Method
__init__
(self, in_channels, out_channels)
model.py:51
Method
__init__
(self, in_channels, out_channels)
model.py:65
Method
__init__
(self, in_channels, inner_channels, out_channels)
model.py:79
Method
__iter__
(self)
data_management.py:53
Method
__len__
(self)
data_management.py:66
Method
__len__
(self)
data_management.py:80
Function
data_augmentation
(image)
data_management.py:9
Method
forward
(self, x)
model.py:31
Method
forward
(self, x)
model.py:44
Method
forward
(self, x)
model.py:59
Method
forward
(self, x)
model.py:73
Method
forward
(self, x)
model.py:91
Method
forward
(self, inputs)
model.py:158
Function
init_weights
(init_type='xavier')
model.py:6
Function
initializer
(m)
model.py:14
Method
num_samples
(self)
data_management.py:48