drunet.py 🚧

Source: LION/models/CNNs/drunet.py

Warning

This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.

class ResBlock(n_channels=64, kernel_size=(3, 3), bias_free=True, act=torch.nn.functional.leaky_relu)

No docstring is available.

ResBlock.forward(self, x)

No docstring is available.

downsample_strideconv(in_channels=1, out_channels=64, kernel_size=2, stride=2, bias_free=True)

No docstring is available.

upsample_convtranspose(in_channels=64, out_channels=1, kernel_size=2, stride=2, bias_free=True)

No docstring is available.

class DRUNet(model_parameters: LIONParameter = None)

No docstring is available.

DRUNet.default_parameters()

No docstring is available.

DRUNet.cite(cite_format='MLA')

No docstring is available.

DRUNet.forward(self, x0, noise_level: float | None = None)

No docstring is available.