From ec75a86a8322743b17c39b7d4e6dccb5ab4bf707 Mon Sep 17 00:00:00 2001 From: LinasKo Date: Thu, 20 Jun 2024 21:26:54 +0300 Subject: [PATCH] Fix, from_florence_2: polygon_to_mask expects np.int32 --- supervision/detection/lmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/detection/lmm.py b/supervision/detection/lmm.py index e6555706..c450403b 100644 --- a/supervision/detection/lmm.py +++ b/supervision/detection/lmm.py @@ -143,7 +143,7 @@ def from_florence_2( masks_list = [] for polygons_of_same_class in result["polygons"]: for polygon in polygons_of_same_class: - polygon = np.reshape(polygon, (-1, 2)) + polygon = np.reshape(polygon, (-1, 2)).astype(np.int32) mask = polygon_to_mask(polygon, resolution_wh) masks_list.append(mask) xyxy = polygon_to_xyxy(polygon)