skself.partial_annotations.lazy_model#

Classes#

`LazyLossWrapper`
`LazyMetricWrapper`
`LazySegmentationModel`

Functions#

pop_channel(tensor, channel_to_remove)

Module Contents#

class skself.partial_annotations.lazy_model.LazyLossWrapper(base_loss, mask_index=-1)[source]#

Bases: tensorflow.keras.losses.Loss

call(y_true, y_pred)[source]#

base_loss[source]#

mask_index = -1[source]#

class skself.partial_annotations.lazy_model.LazyMetricWrapper(metric_fn, mask_index=-1, name=None, **kwargs)[source]#

Bases: tensorflow.keras.metrics.MeanMetricWrapper

update_state(y_true, y_pred, sample_weight=None)[source]#

mask_index = -1[source]#

class skself.partial_annotations.lazy_model.LazySegmentationModel(base_model, ignore_channel_index=-1, **kwargs)[source]#

Bases: tensorflow.keras.Model

call(*args, **kwargs)[source]#

compile(optimizer='rmsprop', loss=None, loss_weights=None, metrics=None, weighted_metrics=None, run_eagerly=False, steps_per_execution=1, jit_compile='auto', **kwargs)[source]#

evaluate(*args, **kwargs)[source]#

fit(*args, **kwargs)[source]#

base_unet[source]#

mask_index = -1[source]#

skself.partial_annotations.lazy_model.pop_channel(tensor, channel_to_remove)[source]#