Merge pull request #64 from roboflow/feature/small_housekeeping_before_release

🧹 Small housekeeping before release

docs/changelog.md

@@ -1,3 +1,12 @@
+### 0.5.2 <small>April 13, 2023</small>
+
+- Fixed [[#63](https://github.com/roboflow/supervision/pull/63)]: `LineZone.trigger` function expects 4 values instead of 5.
+
+### 0.5.1 <small>April 12, 2023</small>
+
+- Fixed `Detections.__getitem__` method did not return mask for selected item.
+- Fixed `Detections.area` crashed for mask detections.
+
 ### 0.5.0 <small>April 10, 2023</small>
 
 - Added [[#58](https://github.com/roboflow/supervision/pull/58)]: `Detections.mask` to enable segmentation support.

supervision/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.5.0"
+__version__ = "0.5.2"
 
 from supervision.annotation.voc import detections_to_voc_xml
 from supervision.detection.annotate import BoxAnnotator, MaskAnnotator

supervision/detection/core.py

@@ -319,6 +319,7 @@ class Detections:
         ):
             return Detections(
                 xyxy=self.xyxy[index],
+                mask=self.mask[index] if self.mask is not None else None,
                 confidence=self.confidence[index]
                 if self.confidence is not None
                 else None,
@@ -341,7 +342,7 @@ class Detections:
           np.ndarray: An array of floats containing the area of each detection in the format of `(area_1, area_2, ..., area_n)`, where n is the number of detections.
         """
         if self.mask is not None:
-            return np.ndarray([np.sum(mask) for mask in self.mask])
+            return np.array([np.sum(mask) for mask in self.mask])
         else:
             return self.box_area
 

supervision/detection/line_counter.py

@@ -35,7 +35,7 @@ class LineZone:
             detections (Detections): The detections for which to update the counts.
 
         """
-        for xyxy, confidence, class_id, tracker_id in detections:
+        for xyxy, _, confidence, class_id, tracker_id in detections:
             # handle detections with no tracker_id
             if tracker_id is None:
                 continue