cFBPConvNet.py 🚧

Source: LION/models/post_processing/cFBPConvNet.py

Warning

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

class InitialVelocity(channels: int)

Initial velocity for second order ODE

InitialVelocity.forward(self, x0)

No docstring is available.

class ConvSODEFunc(channels, relu_type='ReLU', relu_last=True, kernel_size=3)

Second order ODE block

ConvSODEFunc.forward(self, t, x)

No docstring is available.

class ODEBlock(odefunc: nn.Module, tol: float = 0.001, adjoint: bool = False)

No docstring is available.

ODEBlock.forward(self, x, eval_times=None, solver='rk4')

No docstring is available.

class ConvODEFunc(channels, relu_type='ReLU', relu_last=True, kernel_size=3)

No docstring is available.

ConvODEFunc.forward(self, t, 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 cFBPConvNet(geometry_parameters: ct.Geometry, model_parameters: LIONParameter = None)

No docstring is available.

cFBPConvNet.default_parameters()

No docstring is available.

cFBPConvNet.cite(cite_format='MLA')

No docstring is available.

cFBPConvNet.forward(self, x)

No docstring is available.