Merge pull request #5 from supervisely-ecosystem/float

segmentation + bbox to float
supervisely-ecosystem · Aug 25, 2021 · d9faba8 · d9faba8
2 parents e6e1617 + 2f525f1
commit d9faba8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/functions.py b/src/functions.py
@@ -29,12 +29,12 @@ def get_project_contributors():
 
 
 def coco_segmentation(segmentation):  # works only with external vertices for now
-    segmentation = [coord for sublist in segmentation for coord in sublist]
+    segmentation = [float(coord) for sublist in segmentation for coord in sublist]
     return segmentation
 
 
 def coco_bbox(bbox):
-    bbox = [coord for sublist in bbox for coord in sublist]
+    bbox = [float(coord) for sublist in bbox for coord in sublist]
     x, y, max_x, max_y = bbox
     width = max_x - x
     height = max_y - y