From b6547ba58e2d84736b3a0e413b4511c8a5eefc9f Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 5 Aug 2024 13:03:05 +0000 Subject: [PATCH] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20form?= =?UTF-8?q?at=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/detection/core.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 2964840b..ee889c88 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -443,13 +443,13 @@ class Detections: [Transformer](https://github.com/huggingface/transformers) inference result. Args: - transformers_results (Union[dict, torch.Tensor]): Inference results from - your Transformers model. This can be either a dictionary containing - valuable outputs like `scores`, `labels`, `boxes`, `masks`, - `segments_info`, and `segmentation`, or a `torch.Tensor` holding a + transformers_results (Union[dict, torch.Tensor]): Inference results from + your Transformers model. This can be either a dictionary containing + valuable outputs like `scores`, `labels`, `boxes`, `masks`, + `segments_info`, and `segmentation`, or a `torch.Tensor` holding a segmentation map where values represent class IDs. - id2label (Optional[Dict[int, str]]): A map from index to label. Typically - part of `transformers` model configuration. If provided, each detection + id2label (Optional[Dict[int, str]]): A map from index to label. Typically + part of `transformers` model configuration. If provided, each detection will include the associated class name. Returns: