diff --git a/supervision/detection/lmm.py b/supervision/detection/lmm.py index 41a7a71a..9cd9e99a 100644 --- a/supervision/detection/lmm.py +++ b/supervision/detection/lmm.py @@ -143,7 +143,7 @@ def from_florence_2( for polygons_of_same_class in result["polygons"]: for polygon in polygons_of_same_class: polygon = np.reshape(polygon, (-1, 2)).astype(np.int32) - mask = polygon_to_mask(polygon, resolution_wh) + mask = polygon_to_mask(polygon, resolution_wh).astype(bool) masks_list.append(mask) xyxy = polygon_to_xyxy(polygon) xyxy_list.append(xyxy)