MCPcopy Create free account

hub / github.com/JavierGurrola/RDUNet / functions

Functions52 in github.com/JavierGurrola/RDUNet

↓ 5 callersMethod__init__
(self, **kwargs)
model.py:110
↓ 2 callersFunctionmod_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 callersFunctionbuild_ensemble
Create image ensemble to estimate denoised image. :param image: numpy array Noisy image. :param normalize: bool Normalize
utils.py:85
↓ 1 callersFunctioncreate_patches
(image, patch_size, step)
data_management.py:30
↓ 1 callersFunctionfit_model
Training of the denoiser model. :param model: torch Module Neural network to fit. :param data_loaders: dict Dictionary wi
train.py:104
↓ 1 callersMethodget_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 callersMethodload_dataset
(self, files)
data_management.py:92
↓ 1 callersFunctionload_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 callersFunctionmain
()
main_train.py:17
↓ 1 callersFunctionmod_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 callersFunctionpredict
(model, noisy_dataset, gt_dataset, device, padding, n_channels, results_path)
main_test.py:14
↓ 1 callersFunctionpredict_ensemble
Predict batch of images from an ensemble. :param model: torch Module Trained model to estimate denoised images. :param ensemble:
utils.py:153
↓ 1 callersFunctionseparate_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 callersFunctionset_seed
Sets all random seeds. :param seed: int Seed value. :return: None
utils.py:73
↓ 1 callersMethodupdate_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
Functiondata_augmentation
(image)
data_management.py:9
Methodforward
(self, x)
model.py:31
Methodforward
(self, x)
model.py:44
Methodforward
(self, x)
model.py:59
Methodforward
(self, x)
model.py:73
Methodforward
(self, x)
model.py:91
Methodforward
(self, inputs)
model.py:158
Functioninit_weights
(init_type='xavier')
model.py:6
Functioninitializer
(m)
model.py:14
Methodnum_samples
(self)
data_management.py:48