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)