elektronn3.models.unet3d_lite module

exception elektronn3.models.unet3d_lite.PoolingError

Bases: Exception

class elektronn3.models.unet3d_lite.UNet3dLite(*args: Any, **kwargs: Any)

Bases: Module

(WIP) Re-implementation of the unet3d_lite model from ELEKTRONN2

See https://github.com/ELEKTRONN/ELEKTRONN2/blob/master/examples/unet3d_lite.py

Pay attention to shapes: Only spatial input shape (22, 140, 140) is supported.

fov=[12, 88, 88], offsets=[6, 44, 44], strides=[1 1 1], spatial shape=[10, 52, 52]

This model is directly compatible with torch.jit.script.

static autocrop(from_down, from_up)

Return type

Tuple[Tensor, Tensor]

static down(x, ks=(1, 2, 2))

forward(inp)