📚 Docs changelog.md update

docs/changelog.md

@@ -1,3 +1,8 @@
+### 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.1"
 
 from supervision.annotation.voc import detections_to_voc_xml
 from supervision.detection.annotate import BoxAnnotator, MaskAnnotator

supervision/detection/core.py

@@ -319,6 +319,9 @@ 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 +344,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