supervisely 6.73.456__py3-none-any.whl → 6.73.458__py3-none-any.whl

supervisely/nn/inference/predict_app/gui/classes_selector.py

@@ -1,10 +1,48 @@
-from typing import Dict, Any
-from supervisely.app.widgets import Button, Card, ClassesTable, Container, Text
-
-
+from typing import Any, Dict
+
+import pandas as pd
+
+from supervisely.app.widgets import (
+    Button,
+    Card,
+    ClassesTable,
+    Container,
+    FastTable,
+    Text,
+)
+from supervisely.geometry.any_geometry import AnyGeometry
+from supervisely.geometry.bitmap import Bitmap
+from supervisely.geometry.cuboid import Cuboid
+from supervisely.geometry.cuboid_3d import Cuboid3d
+from supervisely.geometry.geometry import Geometry
+from supervisely.geometry.graph import GraphNodes
+from supervisely.geometry.multichannel_bitmap import MultichannelBitmap
+from supervisely.geometry.point import Point
+from supervisely.geometry.point_3d import Point3d
+from supervisely.geometry.pointcloud import Pointcloud
+from supervisely.geometry.polygon import Polygon
+from supervisely.geometry.polyline import Polyline
+from supervisely.geometry.rectangle import Rectangle
+from supervisely.project.project_meta import ProjectMeta
+
+type_to_zmdi_icon = {
+    AnyGeometry: "zmdi zmdi-shape",
+    Rectangle: "zmdi zmdi-crop-din",  # "zmdi zmdi-square-o"
+    # sly.Polygon: "icons8-polygon",  # "zmdi zmdi-edit"
+    Polygon: "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAABmJLR0QA/wD/AP+gvaeTAAAB6klEQVRYhe2Wuy8EURTGf+u5VESNXq2yhYZCoeBv8RcI1i6NVUpsoVCKkHjUGlFTiYb1mFmh2MiKjVXMudmb3cPOzB0VXzKZm5k53/nmvO6Ff4RHD5AD7gFP1l3Kd11AHvCBEpAVW2esAvWmK6t8l1O+W0lCQEnIJoAZxUnzNQNkZF36jrQjgoA+uaciCgc9VaExBOyh/6WWAi1VhbjOJ4FbIXkBtgkK0BNHnYqNKUIPeBPbKyDdzpld5T6wD9SE4AwYjfEDaXFeFzE/doUWuhqwiFsOCwqv2hV2lU/L+sHBscGTxdvSFVoXpAjCZdauMHVic6ndl6U1VBsJCFhTeNUU9IiIEo3qvQYGHAV0AyfC5wNLhKipXuBCjA5wT8WxcM1FMRoBymK44CjAE57hqIazwCfwQdARcXa3UXHuRXVucIjb7jYvNkdxBZg0TBFid7PQTRAtX2xOiXkuMAMqYwkIE848rZFbjyNAmw9bIeweaZ2A5TgC7PnwKkTPtN+cTOrsyN3FEWAjRTAX6sA5ek77gSL6+WHZVQDAIHAjhJtN78aAS3lXAXYIivBOnCdyOAUYB6o0xqsvziry7FLE/Cp20cNcJEjDr8MUmVOVRzkVN+Nd7vZGVXXgiwxtPiRS5WFhz4fEq/zv4AvToMn7vCn3eAAAAABJRU5ErkJggg==",
+    Bitmap: "zmdi zmdi-brush",
+    Polyline: "zmdi zmdi-gesture",
+    Point: "zmdi zmdi-dot-circle-alt",
+    Cuboid: "zmdi zmdi-ungroup",  #
+    GraphNodes: "zmdi zmdi-grain",
+    Cuboid3d: "zmdi zmdi-codepen",
+    Pointcloud: "zmdi zmdi-cloud-outline",  # "zmdi zmdi-border-clear"
+    MultichannelBitmap: "zmdi zmdi-layers",  # "zmdi zmdi-collection-item"
+    Point3d: "zmdi zmdi-filter-center-focus",  # "zmdi zmdi-select-all"
+}
 class ClassesSelector:
-    title = "Classes Selector"
-    description = "Select classes that will be used for inference"
+    title = "Classes"
+    description = "Select classes that will be used for inference. This classes are defined by the deployed model"
     lock_message = "Select previous step to unlock"
 
     def __init__(self):
@@ -24,7 +62,9 @@ class ClassesSelector:
         # -------------------------------- #
 
         # Classes
-        self.classes_table = ClassesTable()
+        columns = ["class", "shape"]
+        self.classes_table = FastTable(columns=columns, page_size=100, is_selectable=True)
+        # self.classes_table = ClassesTable()
         self.classes_table.hide()
         # Add widgets to display ------------ #
         self.display_widgets.extend([self.classes_table])
@@ -45,25 +85,36 @@ class ClassesSelector:
             content=self.container,
             lock_message=self.lock_message,
         )
-        self.card.lock()
         # -------------------------------- #
 
+    def lock(self):
+        self.card.lock()
+        self.classes_table.hide()
+        self.validator_text.hide()
+
+    def unlock(self):
+        self.card.unlock()
+        self.classes_table.show()
+
     @property
     def widgets_to_disable(self) -> list:
         return [self.classes_table]
 
+    def set_project_meta(self, project_meta) -> None:
+        self._update_meta(project_meta)
+
     def load_from_json(self, data: Dict[str, Any]) -> None:
         if "classes" in data:
             self.set_classes(data["classes"])
 
     def get_selected_classes(self) -> list:
-        return self.classes_table.get_selected_classes()
+        return [row.row[0] for row in self.classes_table.get_selected_rows()]
 
     def set_classes(self, classes) -> None:
-        self.classes_table.select_classes(classes)
+        self.classes_table.select_rows_by_value("class", classes)
 
     def select_all_classes(self) -> None:
-        self.classes_table.select_all()
+        self.classes_table.select_rows(list(range(len(self.classes_table._rows_total))))
 
     def get_settings(self) -> Dict[str, Any]:
         return {"classes": self.get_selected_classes()}
@@ -73,7 +124,7 @@ class ClassesSelector:
             return True
 
         self.validator_text.hide()
-        selected_classes = self.classes_table.get_selected_classes()
+        selected_classes = self.get_selected_classes()
         n_classes = len(selected_classes)
 
         if n_classes == 0:
@@ -89,3 +140,21 @@ class ClassesSelector:
         self.validator_text.set(text=". ".join(message_parts), status=status)
         self.validator_text.show()
         return is_valid
+
+    def _update_meta(self, project_meta: ProjectMeta) -> None:
+        table_data = []
+        for obj_class in project_meta.obj_classes:
+            table_line = []
+            name = obj_class.name
+            icon = type_to_zmdi_icon[AnyGeometry]
+            icon = type_to_zmdi_icon.get(obj_class.geometry_type, icon)
+            shape = obj_class.geometry_type.geometry_name().lower()
+            if shape == "graph":
+                shape = "graph (keypoints)"
+            table_line = [name, shape]
+            table_data.append(table_line)
+        self._table_data = table_data
+        if self._table_data:
+            self.classes_table.read_pandas(pd.DataFrame(self._table_data))
+        else:
+            self.classes_table.clear()