Merge remote-tracking branch 'origin/feature/make-from_ultralytics-extract-class-names-from-result' into feature/make-from_ultralytics-extract-class-names-from-result

This commit is contained in:
SkalskiP committed 2024-01-31 08:28:45 +01:00
commit fc571cc191
1 file changed
+3 -5
+3 -5
View File
@@ -5,8 +5,8 @@ from dataclasses import dataclass, field
from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
import numpy as np
from supervision.config import CLASS_NAME_DATA_FIELD, ORIENTED_BOX_COORDINATES
from supervision.config import CLASS_NAME_DATA_FIELD, ORIENTED_BOX_COORDINATES
from supervision.detection.utils import (
calculate_masks_centroids,
extract_ultralytics_masks,
@@ -219,7 +219,7 @@ class Detections:
else None,
data={
ORIENTED_BOX_COORDINATES: oriented_box_coordinates,
CLASS_NAME_DATA_FIELD: class_names
CLASS_NAME_DATA_FIELD: class_names,
},
)
@@ -233,9 +233,7 @@ class Detections:
tracker_id=ultralytics_results.boxes.id.int().cpu().numpy()
if ultralytics_results.boxes.id is not None
else None,
data={
CLASS_NAME_DATA_FIELD: class_names
}
data={CLASS_NAME_DATA_FIELD: class_names},
)
@classmethod