dacapo.experiments.tasks.predictors.affinities_predictor
Module Contents
Classes
Helper class that provides a standard way to create an ABC using |
- class dacapo.experiments.tasks.predictors.affinities_predictor.AffinitiesPredictor(neighborhood: List[funlib.geometry.Coordinate], lsds: bool = True, num_voxels: int = 20, downsample_lsds: int = 1, grow_boundary_iterations: int = 0, affs_weight_clipmin: float = 0.05, affs_weight_clipmax: float = 0.95, lsd_weight_clipmin: float = 0.05, lsd_weight_clipmax: float = 0.95, background_as_object: bool = False)
Helper class that provides a standard way to create an ABC using inheritance.
- property dims
- property num_channels
- property output_array_type
- extractor(voxel_size)
- sigma(voxel_size)
- lsd_pad(voxel_size)
- create_model(architecture)
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.
- create_target(gt)
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.
- create_weight(gt, target, mask, moving_class_counts=None)
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.