Accept any geometry class

supervisely-ecosystem · Sep 27, 2024 · ea9d325 · ea9d325
1 parent 2607ed5
commit ea9d325
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/functions.py b/src/functions.py
@@ -1,12 +1,11 @@
+import src.globals as g
 import supervisely as sly
 from supervisely.nn import TaskType
 from supervisely.nn.inference import SessionJSON
 
-import src.globals as g
-
 geometry_to_task_type = {
-    TaskType.OBJECT_DETECTION: [sly.Rectangle],
-    TaskType.INSTANCE_SEGMENTATION: [sly.Bitmap, sly.Polygon],
+    TaskType.OBJECT_DETECTION: [sly.Rectangle, sly.AnyGeometry],
+    TaskType.INSTANCE_SEGMENTATION: [sly.Bitmap, sly.Polygon, sly.AnyGeometry],
 }