function

benchmarks.hota.models.segments_to_prediction

def segments_to_prediction(segmentation: torch.Tensor, segments_info: list[dict], thing_ids: Iterable[int]) -> FramePrediction

Convert HF panoptic output into a thing-only ``FramePrediction``.

Source: unitrack/benchmarks/hota/models.py:30