elektronn3.models.tiramisu_2d module

adopted from https://github.com/bfortuner/pytorch_tiramisu/blob/master/models/ LICENSE https://github.com/bfortuner/pytorch_tiramisu/blob/master/LICENSE

class elektronn3.models.tiramisu_2d.Bottleneck(*args: Any, **kwargs: Any)[source]

Bases: Sequential

forward(x)[source]
class elektronn3.models.tiramisu_2d.DenseBlock(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
class elektronn3.models.tiramisu_2d.DenseLayer(*args: Any, **kwargs: Any)[source]

Bases: Sequential

forward(x)[source]
class elektronn3.models.tiramisu_2d.FCDenseNet(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
elektronn3.models.tiramisu_2d.FCDenseNet103(n_classes)[source]
elektronn3.models.tiramisu_2d.FCDenseNet57(n_classes)[source]
elektronn3.models.tiramisu_2d.FCDenseNet67(n_classes)[source]
class elektronn3.models.tiramisu_2d.TransitionDown(*args: Any, **kwargs: Any)[source]

Bases: Sequential

forward(x)[source]
class elektronn3.models.tiramisu_2d.TransitionUp(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, skip)[source]
elektronn3.models.tiramisu_2d.center_crop(layer, max_height, max_width)[source]