prediction

Classes

Prediction

Functions

from_pytorch_tensor(tensor[, target])
default_prediction_function(model)

Module Contents

class prediction.Prediction(pred=None, conf=None, box=None, target=None, target_confidence=None)

classification: int | None = None

confidence: float | None = None

bounding_box: numpy.typing.NDArray | None = None

target: int | None = None

target_confidence: float | None = None

__repr__()

Return type:

str

get_class()

is_empty()

is_passing()

prediction.from_pytorch_tensor(tensor, target=None)

Return type:

List[Prediction]

prediction.default_prediction_function(model)