From ebfaee22eb6dabb1c9dd4998bcb93c396c4400ac Mon Sep 17 00:00:00 2001 From: hd Date: Mon, 3 Jul 2023 17:30:53 +0200 Subject: [PATCH] updated codebase --- supervision/detection/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 0d47af29..78ce1be2 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -229,7 +229,7 @@ class Detections: ) @classmethod - def from_mmdet(cls, mmdet_results) -> Detections: + def from_mmdetection(cls, mmdet_results) -> Detections: """ Creates a Detections instance from a [mmdetection](https://github.com/open-mmlab/mmdetection) inference result. Also supported for [mmyolo](https://github.com/open-mmlab/mmyolo) @@ -258,7 +258,7 @@ class Detections: return cls( xyxy=mmdet_results.pred_instances.bboxes.xyxy.cpu().numpy(), confidence=mmdet_results.pred_instances.scores.cpu().numpy(), - class_id=mmdet_results.pred_instances.labels.cpu().numpy().astype(int) + class_id=mmdet_results.pred_instances.labels.cpu().numpy().astype(int), ) @classmethod