supervisely 6.73.386__py3-none-any.whl → 6.73.388__py3-none-any.whl

supervisely/convert/image/pascal_voc/pascal_voc_converter.py

@@ -160,7 +160,6 @@ class PascalVOCConverter(ImageConverter):
             self._items.append(item)
         return detected_ann_cnt
 
-
     def _scan_for_item_segm_paths(self, item: Item, item_name_noext: str) -> Item:
         if self._segm_dir is not None:
             segm_path = os.path.join(self._segm_dir, f"{item_name_noext}.png")
@@ -170,11 +169,11 @@ class PascalVOCConverter(ImageConverter):
             inst_path = os.path.join(self._inst_dir, f"{item_name_noext}.png")
             if file_exists(inst_path):
                 item.inst_path = inst_path
-        
+
         return item
 
     def _scan_for_item_ann_path_and_update_meta(
-            self, item: Item, ann_path: Optional[str], existing_cls_names: Set[str]
+        self, item: Item, ann_path: Optional[str], existing_cls_names: Set[str]
     ) -> Item:
         if ann_path is None:
             return item
@@ -186,7 +185,6 @@ class PascalVOCConverter(ImageConverter):
         item.ann_data = ann_path
         return item
 
-
     def to_supervisely(
         self,
         item: Item,
@@ -200,7 +198,12 @@ class PascalVOCConverter(ImageConverter):
         try:
             item.set_shape()
             return pascal_voc_helper.get_ann(
-                item, self.color2class_name, meta, self._bbox_classes_map, renamed_classes
+                item,
+                self.color2class_name,
+                meta,
+                self._bbox_classes_map,
+                renamed_classes,
+                renamed_tags,
             )
         except Exception as e:
             logger.warn(f"Failed to convert annotation: {repr(e)}")

supervisely/convert/image/pascal_voc/pascal_voc_helper.py

@@ -12,6 +12,8 @@ from supervisely.annotation.annotation import Annotation
 from supervisely.annotation.label import Label
 from supervisely.annotation.obj_class import ObjClass
 from supervisely.annotation.obj_class_collection import ObjClassCollection
+from supervisely.annotation.tag import Tag, TagValueType
+from supervisely.annotation.tag_meta import TagMeta
 from supervisely.convert.image.image_helper import validate_image_bounds
 from supervisely.geometry.bitmap import Bitmap
 from supervisely.geometry.polygon import Polygon
@@ -33,6 +35,7 @@ VALID_IMG_EXT = {".jpe", ".jpeg", ".jpg"}
 TRAIN_TAG_NAME = "train"
 VAL_TAG_NAME = "val"
 TRAINVAL_TAG_NAME = "trainval"
+DEFAULT_OBJECT_FIELDS = {"name", "class", "bndbox"}
 
 default_classes_colors = {
     "neutral": (224, 224, 192),
@@ -118,6 +121,7 @@ def get_ann(
     meta: ProjectMeta,
     bbox_classes_map: dict,
     renamed_classes=None,
+    renamed_tags=None,
 ) -> Annotation:
     segm_path, inst_path = item.segm_path, item.inst_path
     height, width = item.shape
@@ -126,7 +130,7 @@ def get_ann(
 
     if item.ann_data is not None:
         bbox_labels = xml_to_sly_labels(
-            item.ann_data, meta, bbox_classes_map, img_rect, renamed_classes
+            item.ann_data, meta, bbox_classes_map, img_rect, renamed_classes, renamed_tags
         )
         ann = ann.add_labels(bbox_labels)
 
@@ -188,6 +192,7 @@ def xml_to_sly_labels(
     bbox_classes_map: dict,
     img_rect: Rectangle,
     renamed_classes=None,
+    renamed_tags=None,
 ) -> List[Label]:
     import xml.etree.ElementTree as ET
 
@@ -196,20 +201,40 @@ def xml_to_sly_labels(
         tree = ET.parse(in_file)
         root = tree.getroot()
 
-        for obj in root.iter("object"):
-            cls_name = obj.find("name").text
-            cls_name = bbox_classes_map.get(cls_name, cls_name)
-            if renamed_classes and cls_name in renamed_classes:
-                cls_name = renamed_classes[cls_name]
-            obj_cls = meta.obj_classes.get(cls_name)
-            if obj_cls is None:
-                logger.warn(f"Class {cls_name} is not found in meta. Skipping.")
-                continue
-            xmlbox = obj.find("bndbox")
-            bbox_coords = [float(xmlbox.find(x).text) for x in ("ymin", "xmin", "ymax", "xmax")]
-            bbox = Rectangle(*bbox_coords)
-            label = Label(bbox, obj_cls)
-            labels.append(label)
+    for obj in root.iter("object"):
+        geometry = None
+        obj_cls = None
+        tags = []
+
+        for element in obj:
+            field_name, value = element.tag, element.text
+            if field_name in ["name", "class"]:
+                cls_name = bbox_classes_map.get(value, value)
+                if renamed_classes and cls_name in renamed_classes:
+                    cls_name = renamed_classes[cls_name]
+                obj_cls = meta.obj_classes.get(cls_name)
+                if obj_cls is None:
+                    logger.warn(f"Class {cls_name} is not found in meta. Skipping.")
+                    continue
+            elif field_name == "bndbox":
+                bbox_coords = [
+                    float(element.find(x).text) for x in ("ymin", "xmin", "ymax", "xmax")
+                ]
+                geometry = Rectangle(*bbox_coords)
+            elif field_name not in DEFAULT_OBJECT_FIELDS:
+                tag_name = field_name
+                if renamed_tags and tag_name in renamed_tags:
+                    tag_name = renamed_tags[tag_name]
+                tag_meta = meta.get_tag_meta(tag_name)
+                if tag_meta is None:
+                    logger.warn(f"Tag meta for '{field_name}' is not found in meta. Skipping.")
+                    continue
+                if not isinstance(value, str):
+                    value = str(value)
+                tags.append(Tag(tag_meta, value))
+        if geometry is None or obj_cls is None:
+            continue
+        labels.append(Label(geometry, obj_cls, tags))
     labels = validate_image_bounds(labels, img_rect)
 
     return labels
@@ -227,32 +252,40 @@ def update_meta_from_xml(
         tree = ET.parse(in_file)
         root = tree.getroot()
 
-        for obj in root.iter("object"):
-            class_name = obj.find("name").text
-            original_class_name = class_name
-            obj_cls = meta.obj_classes.get(class_name)
-            if obj_cls is None:
-                obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
-                meta = meta.add_obj_class(obj_cls)
-                existing_cls_names.add(class_name)
-                continue
-            elif obj_cls.geometry_type == Rectangle:
-                continue
-            class_name = class_name + "_bbox"
-            obj_cls = meta.obj_classes.get(class_name)
-            if obj_cls is None:
-                obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
-                meta = meta.add_obj_class(obj_cls)
-                existing_cls_names.add(class_name)
-            elif obj_cls.geometry_type == Rectangle:
-                pass
-            else:
-                class_name = generate_free_name(
-                    existing_cls_names, class_name, extend_used_names=True
-                )
-                obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
-                meta = meta.add_obj_class(obj_cls)
-            bbox_classes_map[original_class_name] = class_name
+    for obj in root.iter("object"):
+        for element in obj:
+            field_name = element.tag
+            if field_name in ["name", "class"]:
+                class_name = element.text
+                original_class_name = class_name
+                obj_cls = meta.obj_classes.get(class_name)
+                if obj_cls is None:
+                    obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
+                    meta = meta.add_obj_class(obj_cls)
+                    existing_cls_names.add(class_name)
+                    continue
+                elif obj_cls.geometry_type == Rectangle:
+                    continue
+                class_name = class_name + "_bbox"
+                obj_cls = meta.obj_classes.get(class_name)
+                if obj_cls is None:
+                    obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
+                    meta = meta.add_obj_class(obj_cls)
+                    existing_cls_names.add(class_name)
+                elif obj_cls.geometry_type == Rectangle:
+                    pass
+                else:
+                    class_name = generate_free_name(
+                        existing_cls_names, class_name, extend_used_names=True
+                    )
+                    obj_cls = ObjClass(name=class_name, geometry_type=Rectangle)
+                    meta = meta.add_obj_class(obj_cls)
+                bbox_classes_map[original_class_name] = class_name
+            elif field_name not in DEFAULT_OBJECT_FIELDS:
+                tag_meta = meta.get_tag_meta(field_name)
+                if tag_meta is None:
+                    tag_meta = TagMeta(field_name, TagValueType.ANY_STRING)
+                    meta = meta.add_tag_meta(tag_meta)
 
     return meta
 

supervisely/nn/benchmark/object_detection/visualizer.py

@@ -705,8 +705,16 @@ class ObjectDetectionVisualizer(BaseVisualizer):
         return False
 
     def _get_sample_data_for_gallery(self):
-        # get sample images with annotations for visualization (Prediction project)
-        pred_ds = random.choice(self.eval_result.pred_dataset_infos)
-        self.eval_result.sample_images = self.api.image.get_list(pred_ds.id, limit=9)
-        image_ids = [x.id for x in self.eval_result.sample_images]
-        self.eval_result.sample_anns = self.api.annotation.download_batch(pred_ds.id, image_ids)
+        """Get sample images with annotations for visualization (preview gallery)"""
+        sample_images = []
+        limit = 9
+        for ds_info in self.eval_result.pred_dataset_infos:
+            images = self.api.image.get_list(
+                ds_info.id, limit=limit, force_metadata_for_links=False
+            )
+            sample_images.extend(images)
+        if len(sample_images) > limit:
+            sample_images = random.sample(sample_images, limit)
+        self.eval_result.sample_images = sample_images
+        ids = [img.id for img in sample_images]
+        self.eval_result.sample_anns = self.api.annotation.download_batch(ds_info.id, ids)

supervisely/nn/benchmark/semantic_segmentation/visualizer.py

@@ -280,15 +280,19 @@ class SemanticSegmentationVisualizer(BaseVisualizer):
                     raise RuntimeError(f"Match data was not created properly. {e}")
 
     def _get_sample_data_for_gallery(self):
-        # get sample images with annotations for visualization
-        pred_ds = random.choice(self.eval_result.pred_dataset_infos)
-        imgs = self.api.image.get_list(pred_ds.id, limit=9, force_metadata_for_links=False)
-        anns = self.api.annotation.download_batch(
-            pred_ds.id, [x.id for x in imgs], force_metadata_for_links=False
-        )
-
-        self.eval_result.sample_images = imgs
-        self.eval_result.sample_anns = anns
+        """Get sample images with annotations for visualization (preview gallery)"""
+        sample_images = []
+        limit = 9
+        for ds_info in self.eval_result.pred_dataset_infos:
+            images = self.api.image.get_list(
+                ds_info.id, limit=limit, force_metadata_for_links=False
+            )
+            sample_images.extend(images)
+        if len(sample_images) > limit:
+            sample_images = random.sample(sample_images, limit)
+        self.eval_result.sample_images = sample_images
+        ids = [img.id for img in sample_images]
+        self.eval_result.sample_anns = self.api.annotation.download_batch(ds_info.id, ids)
 
     def _create_clickable_label(self):
         return MarkdownWidget(name="clickable_label", title="", text=self.vis_texts.clickable_label)

{supervisely-6.73.386.dist-info → supervisely-6.73.388.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.386
+Version: 6.73.388
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely

{supervisely-6.73.386.dist-info → supervisely-6.73.388.dist-info}/RECORD

@@ -610,8 +610,8 @@ supervisely/convert/image/multi_view/multi_view.py,sha256=V-6oFN6oDre7UhejfyDkGK
 supervisely/convert/image/multispectral/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/convert/image/multispectral/multispectral_converter.py,sha256=T3etYVNI0AUUrQsQhxw_r85NthXrqhqmdZQfz8kUY0g,5194
 supervisely/convert/image/pascal_voc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/convert/image/pascal_voc/pascal_voc_converter.py,sha256=h5Q3qW4riUCXPo5535wuKGurlLvPfKbWGML0RGnMxyg,7648
-supervisely/convert/image/pascal_voc/pascal_voc_helper.py,sha256=oM-HKUePUwB27aQjL-EvyeS7K-GX0X31rms2yOTTnGo,27872
+supervisely/convert/image/pascal_voc/pascal_voc_converter.py,sha256=zpn_s_WT1Z72WztXlPEJIBfO7-DbHCnjsOZFGOoYMCA,7729
+supervisely/convert/image/pascal_voc/pascal_voc_helper.py,sha256=vu6L2Gm1IU5aDPGibngVJvdje0ndS8NW-0WRhmr6vPE,29415
 supervisely/convert/image/pdf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/convert/image/pdf/pdf_converter.py,sha256=LKvVng9jPp0cSIjYEjKLOb48wtdOdB7LXS2gjmOdZhE,2442
 supervisely/convert/image/pdf/pdf_helper.py,sha256=IDwLEvsVy8lu-KC1lXvSRkZZ9BCC6ylebnNEtLQU5L4,1288
@@ -809,7 +809,7 @@ supervisely/nn/benchmark/object_detection/evaluation_params.yaml,sha256=fEYA-Exm
 supervisely/nn/benchmark/object_detection/evaluator.py,sha256=s-hPBm5BmoCgwoozVyDacum4kVLNtYK6I6NCt_L_LSA,7278
 supervisely/nn/benchmark/object_detection/metric_provider.py,sha256=_fxS24Q8KhE2n2HknMnt86xVPYEDFw5atxsvXRnCl4w,23855
 supervisely/nn/benchmark/object_detection/text_templates.py,sha256=4BgTIX1Co4WK9_VSUa1qWCmh5OJzo3_opVU6LOjKSjc,25842
-supervisely/nn/benchmark/object_detection/visualizer.py,sha256=0eJ-ATDVMyjMRInoEK604vL3vJPN5Ge1HjarS11CS0A,32365
+supervisely/nn/benchmark/object_detection/visualizer.py,sha256=4BOkSYl8rlvOKO87ZlmZuZCsAM9QTqDqAgNMoiwE8eI,32622
 supervisely/nn/benchmark/object_detection/vis_metrics/__init__.py,sha256=AXCLHEySEdR-B-5sfDoWBmmOLBVlyW2U_xr8Ta42sQI,2096
 supervisely/nn/benchmark/object_detection/vis_metrics/confidence_distribution.py,sha256=J6l9Vc8TGsvkTyH5u7Ry4P2jgJC2GYmcgOeMALftZBw,4254
 supervisely/nn/benchmark/object_detection/vis_metrics/confidence_score.py,sha256=pmxnF_UJk0WhqEdL7O_yIjIBtxPipLQZVJwCr6XB3zc,4751
@@ -838,7 +838,7 @@ supervisely/nn/benchmark/semantic_segmentation/evaluation_params.yaml,sha256=47D
 supervisely/nn/benchmark/semantic_segmentation/evaluator.py,sha256=XafPMpGL6v0ZQ-m7DkEjoY7W6fGCJNKolql5BA3M8V0,7261
 supervisely/nn/benchmark/semantic_segmentation/metric_provider.py,sha256=478l7w2n-yueBMVABsakfIQEo3MksbCYmKNrwMFOl6w,6546
 supervisely/nn/benchmark/semantic_segmentation/text_templates.py,sha256=7yRRD2FAdJHGSRqBVIjNjzCduKzaepA1OWtggi7B0Dg,8580
-supervisely/nn/benchmark/semantic_segmentation/visualizer.py,sha256=T7WTjnyFLL3BAyH7yQIqI6R5VNj1M3guamwhDglOegE,13293
+supervisely/nn/benchmark/semantic_segmentation/visualizer.py,sha256=ny7S9cYdYeRL5je5eGpJOneVSz7MxZdykImPdNQ8uPY,13500
 supervisely/nn/benchmark/semantic_segmentation/vis_metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/nn/benchmark/semantic_segmentation/vis_metrics/acknowledgement.py,sha256=Lm82x8AIMKv1WqmqA0W9fugSlQ_JrP9dwYYYReZmhvI,440
 supervisely/nn/benchmark/semantic_segmentation/vis_metrics/classwise_error_analysis.py,sha256=0bmL43a4cqw3grFoG68NN8Y1fkRpHBIRJptcxMor-78,1884
@@ -1103,9 +1103,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.386.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.386.dist-info/METADATA,sha256=pAcUu38hzyDhqOlzLArYsO0AzNHXluQ4MbM0UtAm1nk,35154
-supervisely-6.73.386.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.386.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.386.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.386.dist-info/RECORD,,
+supervisely-6.73.388.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.388.dist-info/METADATA,sha256=Jj-5_iGTZP0t8mfQSkiS3ADiyJg1YKqY0ZnVpmkarZQ,35154
+supervisely-6.73.388.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.388.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.388.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.388.dist-info/RECORD,,