3D_Res_UNet.py 🚧

Source: LION/models/CNNs/UNets/3D_Res_UNet.py

Warning

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

class ResidualUnit(channels, relu_type='ReLU', kernel_size=3)

No docstring is available.

ResidualUnit.forward(self, x)

No docstring is available.

class Down

Downscaling with maxpool

Down.forward(self, x)

No docstring is available.

class Up(channels, stride=2, relu_type='ReLU')

Upscaling with transpose conv

Up.forward(self, x)

No docstring is available.

class ResUNet3D(model_parameters: LIONParameter = None)

No docstring is available.

ResUNet3D.default_parameters()

No docstring is available.

ResUNet3D.forward(self, x)

No docstring is available.