dacapo.experiments.tasks.predictors.predictor

Module Contents

Classes

Predictor

Helper class that provides a standard way to create an ABC using

class dacapo.experiments.tasks.predictors.predictor.Predictor

Helper class that provides a standard way to create an ABC using inheritance.

abstract property output_array_type
abstract create_model(architecture: dacapo.experiments.architectures.architecture.Architecture) dacapo.experiments.model.Model

Given a training architecture, create a model for this predictor. This is usually done by appending extra layers to the output of the architecture to get the output tensor of the architecture into the right shape for this predictor.

abstract create_target(gt: dacapo.experiments.datasplits.datasets.arrays.Array) dacapo.experiments.datasplits.datasets.arrays.Array

Create the target array for training, given a ground-truth array.

In general, the target is different from the ground-truth.

The target is the array that is passed to the loss, and hence directly compared to the prediction (i.e., the output of the model). Depending on the predictor, the target can therefore be different from the ground-truth (e.g., an instance segmentation ground-truth would have to be converted into boundaries, if the model is predicting boundaries).

By default, it is assumed that the spatial dimensions of ground-truth and target are the same.

If your predictor needs more ground-truth context to create a target (e.g., because it predicts the distance to a boundary, up to a certain threshold), you can request a larger ground-truth region. See method gt_region_for_roi.

abstract create_weight(gt: dacapo.experiments.datasplits.datasets.arrays.Array, target: dacapo.experiments.datasplits.datasets.arrays.Array, mask: dacapo.experiments.datasplits.datasets.arrays.Array, moving_class_counts: Any) Tuple[dacapo.experiments.datasplits.datasets.arrays.Array, Any]

Create the weight array for training, given a ground-truth and associated target array.

gt_region_for_roi(target_spec)

Report how much spatial context this predictor needs to generate a target for the given ROI. By default, uses the same ROI.

Overwrite this method to request ground-truth in a larger ROI, as needed.

padding(gt_voxel_size: funlib.geometry.Coordinate) funlib.geometry.Coordinate