INNER CODE UNIT · Python
unary_from_labels
lucasb-eyer/pydensecrf · pydensecrf/utils.py:6
def unary_from_labels(labels, n_labels, gt_prob, zero_unsure=True):
"""
Simple classifier that is 50% certain that the annotation is correct.
(same as in the inference example).
Parameters
----------
labels: numpy.array
The label-map, i.e. an array of your data's shape where each unique
value corresponds to a label.
n_labels: int
The total number of labels there are.
If `zero_unsure` is True (the default), this number should not include
`0` in counting the labels, since `0` is not a label!
gt_prob: float
The certainty of the ground-truth (must be within (0,1)).
zero_unsure: bool