supervisely 6.73.325__py3-none-any.whl → 6.73.326__py3-none-any.whl

supervisely/convert/volume/nii/nii_planes_volume_converter.py

@@ -8,7 +8,7 @@ from supervisely.convert.volume.nii import nii_volume_helper as helper
 from supervisely.convert.volume.nii.nii_volume_converter import NiiConverter
 from supervisely.convert.volume.volume_converter import VolumeConverter
 from supervisely.geometry.mask_3d import Mask3D
-from supervisely.io.fs import get_file_name
+from supervisely.io.fs import get_file_ext, get_file_name
 from supervisely.volume.volume import is_nifti_file
 from supervisely.volume_annotation.volume_annotation import VolumeAnnotation
 from supervisely.volume_annotation.volume_object import VolumeObject
@@ -34,14 +34,46 @@ class NiiPlaneStructuredConverter(NiiConverter, VolumeConverter):
         ├── 🩻 sag_inference_1.nii class 1
         ├── 🩻 sag_inference_2.nii class 2
         └── 🩻 sag_inference_3.nii class 3
+
+    Additionally, if a TXT file with class color map is present, it will be used to
+    create the classes with names and colors corresponding to the pixel values in the NIfTI files.
+    The TXT file should be structured as follows:
+
+    ```txt
+    1 Femur 255 0 0
+    2 Femoral cartilage 0 255 0
+    3 Tibia 0 0 255
+    4 Tibia cartilage 255 255 0
+    5 Patella 0 255 255
+    6 Patellar cartilage 255 0 255
+    7 Miniscus 175 175 175
+    ```
+    where   1, 2, ... are the pixel values in the NIfTI files
+            Femur, Femoral cartilage, ... are the names of the classes
+            255, 0, 0, ... are the RGB colors of the classes
+    The class name will be used to create the corresponding ObjClass in Supervisely.
     """
 
+    class Item(VolumeConverter.BaseItem):
+        def __init__(self, *args, **kwargs):
+            super().__init__(*args, **kwargs)
+            self._is_semantic = False
+
+        @property
+        def is_semantic(self) -> bool:
+            return self._is_semantic
+
+        @is_semantic.setter
+        def is_semantic(self, value: bool):
+            self._is_semantic = value
+
     def validate_format(self) -> bool:
         # create Items
         converted_dir_name = "converted"
 
         volumes_dict = defaultdict(list)
         ann_dict = defaultdict(list)
+        cls_color_map = None
 
         for root, _, files in os.walk(self._input_data):
             if converted_dir_name in root:
@@ -56,17 +88,24 @@ class NiiPlaneStructuredConverter(NiiConverter, VolumeConverter):
                     if prefix not in helper.PlanePrefix.values():
                         continue
                     name = full_name.split("_")[1]
-                    idx = 1 if len(name.split("_")) < 3 else int(name.split("_")[2])
                     if name in helper.LABEL_NAME or name[:-1] in helper.LABEL_NAME:
                         ann_dict[prefix].append(path)
                     else:
                         volumes_dict[prefix].append(path)
+                ext = get_file_ext(path)
+                if ext == ".txt":
+                    cls_color_map = helper.read_cls_color_map(path)
+                    if cls_color_map is None:
+                        logger.warning(f"Failed to read class color map from {path}.")
 
         self._items = []
         for prefix, paths in volumes_dict.items():
             if len(paths) == 1:
                 item = self.Item(item_path=paths[0])
-                item.ann_data = ann_dict.get(prefix)
+                item.ann_data = ann_dict.get(prefix, [])
+                item.is_semantic = len(item.ann_data) == 1
+                if cls_color_map is not None:
+                    item.custom_data["cls_color_map"] = cls_color_map
                 self._items.append(item)
             elif len(paths) > 1:
                 logger.info(
@@ -79,6 +118,9 @@ class NiiPlaneStructuredConverter(NiiConverter, VolumeConverter):
                         if Path(ann_path).parent == Path(path).parent:
                             possible_ann_paths.append(ann_path)
                     item.ann_data = possible_ann_paths
+                    item.is_semantic = len(possible_ann_paths) == 1
+                    if cls_color_map is not None:
+                        item.custom_data["cls_color_map"] = cls_color_map
                     self._items.append(item)
         self._meta = ProjectMeta()
         return self.items_count > 0
@@ -96,12 +138,18 @@ class NiiPlaneStructuredConverter(NiiConverter, VolumeConverter):
             objs = []
             spatial_figures = []
             for idx, ann_path in enumerate(item.ann_data, start=1):
-                for mask, _ in helper.get_annotation_from_nii(ann_path):
-                    class_name = f"Segment_{idx}"
+                for mask, pixel_id in helper.get_annotation_from_nii(ann_path):
+                    class_id = pixel_id if item.is_semantic else idx
+                    class_name = f"Segment_{class_id}"
+                    color = None
+                    if item.custom_data.get("cls_color_map") is not None:
+                        class_info = item.custom_data["cls_color_map"].get(class_id)
+                        if class_info is not None:
+                            class_name, color = class_info
                     class_name = renamed_classes.get(class_name, class_name)
                     obj_class = meta.get_obj_class(class_name)
                     if obj_class is None:
-                        obj_class = ObjClass(class_name, Mask3D)
+                        obj_class = ObjClass(class_name, Mask3D, color)
                         meta = meta.add_obj_class(obj_class)
                         self._meta_changed = True
                         self._meta = meta

supervisely/convert/volume/nii/nii_volume_converter.py

@@ -169,11 +169,11 @@ class NiiConverter(VolumeConverter):
             item_paths = []
 
             for item in batch:
-                if self._upload_as_links:
-                    remote_path = self.remote_files_map.get(item.path)
-                    if remote_path is not None:
-                        item.custom_data = {"remote_path": remote_path}
-                # nii_path = item.path
+                # if self._upload_as_links:
+                #     remote_path = self.remote_files_map.get(item.path)
+                #     if remote_path is not None:
+                #         item.custom_data = {"remote_path": remote_path}
+
                 item.path = helper.nifti_to_nrrd(item.path, converted_dir)
                 ext = get_file_ext(item.path)
                 if ext.lower() != ext:
@@ -195,8 +195,8 @@ class NiiConverter(VolumeConverter):
                     leave=True if progress_cb is None else False,
                     position=1,
                 )
-                if item.custom_data is not None:
-                    volume_meta.update(item.custom_data)
+                # if item.custom_data is not None:
+                #     volume_meta.update(item.custom_data)
                 api.volume.upload_np(dataset_id, item.name, volume_np, volume_meta, progress_nrrd)
                 info = api.volume.get_info_by_name(dataset_id, item.name)
                 item.volume_meta = info.meta

supervisely/convert/volume/nii/nii_volume_helper.py

@@ -7,6 +7,7 @@ import numpy as np
 from supervisely.collection.str_enum import StrEnum
 from supervisely.geometry.mask_3d import Mask3D
 from supervisely.io.fs import ensure_base_path, get_file_ext, get_file_name
+from supervisely.sly_logger import logger
 from supervisely.volume.volume import convert_3d_nifti_to_nrrd
 
 VOLUME_NAME = "anatomic"
@@ -21,6 +22,54 @@ class PlanePrefix(str, StrEnum):
     AXIAL = "axl"
 
 
+def read_cls_color_map(path: str) -> dict:
+    """Read class color map from TXT file.
+
+    ```txt
+    1 Femur 255 0 0
+    2 Femoral cartilage 0 255 0
+    3 Tibia 0 0 255
+    4 Tibia cartilage 255 255 0
+    5 Patella 0 255 255
+    6 Patellar cartilage 255 0 255
+    7 Miniscus 175 175 175
+    ```
+    """
+
+    cls_color_map = {}
+    if not os.path.exists(path):
+        return None
+    try:
+        with open(path, "r") as file:
+            for line in file:
+                parts = line.strip().split()
+                color = None
+                try:
+                    color = list(map(int, parts[-3:]))
+                except:
+                    pass
+                cls_id = int(parts[0])
+                if not color:
+                    cls_name = " ".join(parts[1:])
+                else:
+                    cls_name = " ".join(parts[1:-3])
+                if cls_id in cls_color_map:
+                    logger.warning(f"Duplicate class ID {cls_id} found in color map.")
+                if cls_name in cls_color_map:
+                    logger.warning(f"Duplicate class name {cls_name} found in color map.")
+                if len(color) != 3:
+                    logger.warning(f"Invalid color format for class {cls_name}. Expected 3 values.")
+                if any(c < 0 or c > 255 for c in color):
+                    logger.warning(
+                        f"Invalid color value for class {cls_name}. Expected values between 0 and 255."
+                    )
+                cls_color_map[cls_id] = (cls_name, color)
+    except Exception as e:
+        logger.warning(f"Failed to read class color map from {path}: {e}")
+        return None
+    return cls_color_map
+
+
 def nifti_to_nrrd(nii_file_path: str, converted_dir: str) -> str:
     """Convert NIfTI 3D volume file to NRRD 3D volume file."""
 

supervisely/volume/volume.py

@@ -816,30 +816,28 @@ def convert_3d_nifti_to_nrrd(path: str) -> Tuple[np.ndarray, dict]:
         path = "/home/admin/work/volumes/vol_01.nii"
         data, header = sly.volume.convert_nifti_to_nrrd(path)
     """
+    import SimpleITK as sitk
 
-    import nibabel as nib  # pylint: disable=import-error
+    nifti_image = sitk.ReadImage(path)
+    nifti_image = _sitk_image_orient_ras(nifti_image)
+    data = sitk.GetArrayFromImage(nifti_image)
+    data = np.transpose(data, (2, 1, 0))
+
+    direction = np.array(nifti_image.GetDirection()).reshape(3, 3)
+    spacing = np.array(nifti_image.GetSpacing())
+    origin = np.array(nifti_image.GetOrigin())
+
+    space_directions = (direction.T * spacing[:, None]).tolist()
 
-    orientation_map = {
-        ('R', 'A', 'S'): "right-anterior-superior",
-        ('L', 'P', 'S'): "left-posterior-superior",
-        ('R', 'P', 'I'): "right-posterior-inferior",
-        ('L', 'A', 'I'): "left-anterior-inferior"
-    }
-    nifti = nib.load(path)
-    reordered_to_ras_nifti = nib.as_closest_canonical(nifti)
-    data = reordered_to_ras_nifti.get_fdata()
-    affine = reordered_to_ras_nifti.affine
-    orientation = nib.aff2axcodes(affine)
-    space_directions = affine[:3, :3].tolist()
-    space_origin = affine[:3, 3].tolist()
     header = {
-        "space": orientation_map.get(orientation, "unknown"),
-        "space directions": space_directions,
-        "space origin": space_origin,
-        "sizes": data.shape,
-        "type": str(data.dtype),
-        "dimension": len(data.shape),
-    }
+            "dimension": 3,
+            "space": "right-anterior-superior",
+            "sizes": list(data.shape),
+            "space directions": space_directions,
+            "endian": "little",
+            "encoding": "gzip",
+            "space origin": origin
+        }
     return data, header
 
 

{supervisely-6.73.325.dist-info → supervisely-6.73.326.dist-info}/METADATA

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

{supervisely-6.73.325.dist-info → supervisely-6.73.326.dist-info}/RECORD

@@ -664,9 +664,9 @@ supervisely/convert/volume/dicom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 supervisely/convert/volume/dicom/dicom_converter.py,sha256=__QP8fMAaq_BdWFYh1_nAYT2gpY1WwZzdlDj39YwHhw,3195
 supervisely/convert/volume/dicom/dicom_helper.py,sha256=1EXmxl5Z8Xi3ZkZnfJ4EbiPCVyITSXUc0Cn_oo02pPE,1284
 supervisely/convert/volume/nii/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-supervisely/convert/volume/nii/nii_planes_volume_converter.py,sha256=SV1OAoel92qOvWTfcVqzee9LQypAQGQ-hJEMJ9j7nLg,4995
-supervisely/convert/volume/nii/nii_volume_converter.py,sha256=nVHKK7wVhMvBiX1awMoXOwt2f5MHujqVmof0v_zW5E4,8502
-supervisely/convert/volume/nii/nii_volume_helper.py,sha256=wy8GqjgnFJaEgKKSOgKw_QPETdiYYY5c9BBfyJE3Hqo,1417
+supervisely/convert/volume/nii/nii_planes_volume_converter.py,sha256=9TtN_AgCQgv16Olip6inFanCA5JlEEJ7JQf-0XjIw_Q,7091
+supervisely/convert/volume/nii/nii_volume_converter.py,sha256=IZ6DJeLLbLAW-kifOJ_9ddV3h7gL3AswM2TTbXB9Os0,8476
+supervisely/convert/volume/nii/nii_volume_helper.py,sha256=RvYab6Z530Qw-qTAsZ3WM8WZKqhijia9OC-g4_zOSEs,3142
 supervisely/convert/volume/sly/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/convert/volume/sly/sly_volume_converter.py,sha256=XmSuxnRqxchG87b244f3h0UHvOt6IkajMquL1drWlCM,5595
 supervisely/convert/volume/sly/sly_volume_helper.py,sha256=gUY0GW3zDMlO2y-zQQG36uoXMrKkKz4-ErM1CDxFCxE,5620
@@ -1060,7 +1060,7 @@ supervisely/volume/__init__.py,sha256=EBZBY_5mzabXzMUQh5akusIGd16XnX9n8J0jIi_JmW
 supervisely/volume/nrrd_encoder.py,sha256=1lqwwyqxEvctw1ysQ70x4xPSV1uy1g5YcH5CURwL7-c,4084
 supervisely/volume/nrrd_loader.py,sha256=_yqahKcqSRxunHZ5LtnUWIRA7UvIhPKOhAUwYijSGY4,9065
 supervisely/volume/stl_converter.py,sha256=WIMQgHO_u4JT58QdcMXcb_euF1BFhM7D52IVX_0QTxE,6285
-supervisely/volume/volume.py,sha256=7ebCIICqZMwRP3ruRy3PtSeiUpBpSlyRH20VJybBQbI,25828
+supervisely/volume/volume.py,sha256=bUPrDQAr4ZIkSQMzpSWXjsHRqcXUq2Z2H6Fe1uLdYmw,25687
 supervisely/volume_annotation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 supervisely/volume_annotation/constants.py,sha256=BdFIh56fy7vzLIjt0gH8xP01EIU-qgQIwbSHVUcABCU,569
 supervisely/volume_annotation/plane.py,sha256=wyezAcc8tLp38O44CwWY0wjdQxf3VjRdFLWooCrk-Nw,16301
@@ -1082,9 +1082,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.325.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.325.dist-info/METADATA,sha256=wRj0QjkNc2OGdL0AtTslp2JAlgDO8-1f3XeeUfoZfOY,33596
-supervisely-6.73.325.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-supervisely-6.73.325.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.325.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.325.dist-info/RECORD,,
+supervisely-6.73.326.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.326.dist-info/METADATA,sha256=0hQAKuBU9KX8mYNuop6LSvLzz0RQmCGZSdFIqqgmJUE,33596
+supervisely-6.73.326.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+supervisely-6.73.326.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.326.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.326.dist-info/RECORD,,