supervisely 6.73.323__tar.gz → 6.73.325__tar.gz

{supervisely-6.73.323/supervisely.egg-info → supervisely-6.73.325}/PKG-INFO

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

{supervisely-6.73.323 → supervisely-6.73.325}/supervisely/_utils.py

@@ -471,9 +471,20 @@ def get_or_create_event_loop() -> asyncio.AbstractEventLoop:
             return loop
 
 
-def sync_call(coro):
+def run_coroutine(coroutine):
     """
-    This function is used to run asynchronous functions in synchronous context.
+    Runs an asynchronous coroutine in a synchronous context and waits for its result.
+
+    This function checks if an event loop is already running:
+    - If a loop is running, it schedules the coroutine using `asyncio.run_coroutine_threadsafe()`
+      and waits for the result.
+    - If no loop is running, it creates one and executes the coroutine with `run_until_complete()`.
+
+    This ensures compatibility with both synchronous and asynchronous environments
+    without creating unnecessary event loops.
+
+    ⚠️ Note: This method is preferable when working with `asyncio` objects like `Semaphore`,
+    since it avoids issues with mismatched event loops.
 
     :param coro: Asynchronous function.
     :type coro: Coroutine
@@ -484,13 +495,13 @@ def sync_call(coro):
 
     .. code-block:: python
 
-            from supervisely.utils import sync_call
+            from supervisely._utils import run_coroutine
 
             async def async_function():
                 await asyncio.sleep(1)
                 return "Hello, World!"
             coro = async_function()
-            result = sync_call(coro)
+            result = run_coroutine(coro)
             print(result)
             # Output: Hello, World!
     """
@@ -498,10 +509,10 @@ def sync_call(coro):
     loop = get_or_create_event_loop()
 
     if loop.is_running():
-        future = asyncio.run_coroutine_threadsafe(coro, loop=loop)
+        future = asyncio.run_coroutine_threadsafe(coroutine, loop=loop)
         return future.result()
     else:
-        return loop.run_until_complete(coro)
+        return loop.run_until_complete(coroutine)
 
 
 def get_filename_from_headers(url):

{supervisely-6.73.323 → supervisely-6.73.325}/supervisely/api/file_api.py

@@ -23,8 +23,9 @@ from typing_extensions import Literal
 
 import supervisely.io.env as env
 import supervisely.io.fs as sly_fs
-from supervisely._utils import batched, rand_str
+from supervisely._utils import batched, rand_str, run_coroutine
 from supervisely.api.module_api import ApiField, ModuleApiBase
+from supervisely.api.remote_storage_api import RemoteStorageApi
 from supervisely.io.fs import (
     ensure_base_path,
     get_file_ext,
@@ -1420,7 +1421,8 @@ class FileApi(ModuleApiBase):
             api.file.upload_directory(9, local_path, path_to_dir)
         """
         if not remote_dir.startswith("/"):
-            remote_dir = "/" + remote_dir
+            if not RemoteStorageApi.is_bucket_url(remote_dir):
+                remote_dir = "/" + remote_dir
 
         if self.dir_exists(team_id, remote_dir):
             if change_name_if_conflict is True:
@@ -2302,3 +2304,89 @@ class FileApi(ModuleApiBase):
             else:
                 raise e
         return res_remote_dir
+
+    def upload_directory_fast(
+        self,
+        team_id: int,
+        local_dir: str,
+        remote_dir: str,
+        change_name_if_conflict: Optional[bool] = True,
+        progress_cb: Optional[Union[tqdm, Callable]] = None,
+        replace_if_conflict: Optional[bool] = False,
+        enable_fallback: Optional[bool] = True,
+    ) -> str:
+        """
+        Upload Directory to Team Files from local path in fast mode.
+        Files are uploaded asynchronously. If an error occurs, the method will fallback to synchronous upload.
+
+        :param team_id: Team ID in Supervisely.
+        :type team_id: int
+        :param local_dir: Path to local Directory.
+        :type local_dir: str
+        :param remote_dir: Path to Directory in Team Files.
+        :type remote_dir: str
+        :param change_name_if_conflict: Checks if given name already exists and adds suffix to the end of the name.
+        :type change_name_if_conflict: bool, optional
+        :param progress_cb: Function for tracking download progress in bytes.
+        :type progress_cb: Progress, optional
+        :param replace_if_conflict: If True, replace existing dir.
+        :type replace_if_conflict: bool, optional
+        :param enable_fallback: If True, the method will fallback to synchronous upload if an error occurs.
+        :type enable_fallback: bool, optional
+        :return: Path to Directory in Team Files
+        :rtype: :class:`str`
+        """
+        coroutine = self.upload_directory_async(
+            team_id=team_id,
+            local_dir=local_dir,
+            remote_dir=remote_dir,
+            change_name_if_conflict=change_name_if_conflict,
+            progress_size_cb=progress_cb,
+            replace_if_conflict=replace_if_conflict,
+            enable_fallback=enable_fallback,
+        )
+        return run_coroutine(coroutine)
+
+    def upload_bulk_fast(
+        self,
+        team_id: int,
+        src_paths: List[str],
+        dst_paths: List[str],
+        semaphore: Optional[asyncio.Semaphore] = None,
+        progress_cb: Optional[Union[tqdm, Callable]] = None,
+        progress_cb_type: Literal["number", "size"] = "size",
+        enable_fallback: Optional[bool] = True,
+    ) -> None:
+        """
+        Upload multiple files from local paths to Team Files in fast mode.
+        Files are uploaded asynchronously. If an error occurs, the method will fallback to synchronous upload.
+
+        :param team_id: Team ID in Supervisely.
+        :type team_id: int
+        :param src_paths: List of local paths to files.
+        :type src_paths: List[str]
+        :param dst_paths: List of paths to save files in Team Files.
+        :type dst_paths: List[str]
+        :param semaphore: Semaphore for limiting the number of simultaneous uploads.
+        :type semaphore: asyncio.Semaphore, optional
+        :param progress_cb: Function for tracking download progress.
+        :type progress_cb: tqdm or callable, optional
+        :param progress_cb_type: Type of progress callback. Can be "number" or "size". Default is "size".
+                                "size" is used to track the number of transferred bytes.
+                                "number" is used to track the number of transferred files.
+        :type progress_cb_type: Literal["number", "size"], optional 
+        :param enable_fallback: If True, the method will fallback to synchronous upload if an error occurs.
+        :type enable_fallback: bool, optional
+        :return: None
+        :rtype: :class:`NoneType`
+        """
+        coroutine = self.upload_bulk_async(
+            team_id=team_id,
+            src_paths=src_paths,
+            dst_paths=dst_paths,
+            semaphore=semaphore,
+            progress_cb=progress_cb,
+            progress_cb_type=progress_cb_type,
+            enable_fallback=enable_fallback,
+        )
+        return run_coroutine(coroutine)

supervisely-6.73.325/supervisely/app/fastapi/utils.py

@@ -0,0 +1,28 @@
+import asyncio
+import concurrent.futures
+
+
+def run_sync(coroutine):
+    """
+    Runs an asynchronous coroutine in a separate thread and waits for its result.
+    It is useful for running async functions in a synchronous
+    environment.
+
+    This method creates a new thread using ThreadPoolExecutor and executes the coroutine
+    inside a new event loop.
+
+    ⚠️ Note: This function creates a new event loop every time it is called,
+    which can cause issues when using objects tied to a specific loop (e.g., asyncio.Semaphore).
+
+    :param coroutine: coroutine to run
+    :return: result of coroutine
+    """
+    try:
+        with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
+            result = executor.submit(
+                lambda coroutine_to_exec: asyncio.run(coroutine_to_exec), coroutine
+            ).result()
+        return result
+    except RuntimeError as ex:
+        print(repr(ex))
+        return None

{supervisely-6.73.323 → supervisely-6.73.325}/supervisely/convert/converter.py

@@ -57,12 +57,12 @@ class ImportManager:
         self._labeling_interface = labeling_interface
         self._upload_as_links = upload_as_links
         self._remote_files_map = {}
+        self._modality = project_type
 
         self._input_data = self._prepare_input_data(input_data)
         self._unpack_archives(self._input_data)
         remove_junk_from_dir(self._input_data)
 
-        self._modality = project_type
         self._converter = self.get_converter()
         if isinstance(self._converter, (HighColorDepthImageConverter, CSVConverter)):
             self._converter.team_id = self._team_id
@@ -112,17 +112,27 @@ class ImportManager:
             logger.info(f"Input data is a local file: {input_data}. Will use its directory")
             return os.path.dirname(input_data)
         elif self._api.storage.exists(self._team_id, input_data):
-            if self._upload_as_links:
+            if self._upload_as_links and str(self._modality) in [
+                ProjectType.IMAGES.value,
+                ProjectType.VIDEOS.value,
+            ]:
                 logger.info(f"Input data is a remote file: {input_data}. Scanning...")
-                return self._scan_remote_files(input_data)
+                return self._reproduce_remote_files(input_data)
             else:
+                if self._upload_as_links and str(self._modality) == ProjectType.VOLUMES.value:
+                    self._scan_remote_files(input_data)
                 logger.info(f"Input data is a remote file: {input_data}. Downloading...")
                 return self._download_input_data(input_data)
         elif self._api.storage.dir_exists(self._team_id, input_data):
-            if self._upload_as_links:
+            if self._upload_as_links and str(self._modality) in [
+                ProjectType.IMAGES.value,
+                ProjectType.VIDEOS.value,
+            ]:
                 logger.info(f"Input data is a remote directory: {input_data}. Scanning...")
-                return self._scan_remote_files(input_data, is_dir=True)
+                return self._reproduce_remote_files(input_data, is_dir=True)
             else:
+                if self._upload_as_links and str(self._modality) == ProjectType.VOLUMES.value:
+                    self._scan_remote_files(input_data, is_dir=True)
                 logger.info(f"Input data is a remote directory: {input_data}. Downloading...")
                 return self._download_input_data(input_data, is_dir=True)
         else:
@@ -160,7 +170,35 @@ class ImportManager:
         return local_path
 
     def _scan_remote_files(self, remote_path, is_dir=False):
-        """Scan remote directory and create dummy structure locally"""
+        """
+        Scan remote directory. Collect local-remote paths mapping
+        Will be used to save relations between uploaded files and remote files (for volumes).
+        """
+
+        dir_path = remote_path.rstrip("/") if is_dir else os.path.dirname(remote_path)
+        dir_name = os.path.basename(dir_path)
+
+        local_path = os.path.join(get_data_dir(), dir_name)
+
+        if is_dir:
+            files = self._api.storage.list(self._team_id, remote_path, include_folders=False)
+        else:
+            files = [self._api.storage.get_info_by_path(self._team_id, remote_path)]
+
+        unique_directories = set()
+        for file in files:
+            new_path = file.path.replace(dir_path, local_path)
+            self._remote_files_map[new_path] = file.path
+            unique_directories.add(str(Path(file.path).parent))
+
+        logger.info(f"Scanned remote directories:\n   - " + "\n   - ".join(unique_directories))
+        return local_path
+
+    def _reproduce_remote_files(self, remote_path, is_dir=False):
+        """
+        Scan remote directory and create dummy structure locally.
+        Will be used to detect annotation format (by dataset structure) remotely.
+        """
 
         dir_path = remote_path.rstrip("/") if is_dir else os.path.dirname(remote_path)
         dir_name = os.path.basename(dir_path)

{supervisely-6.73.323 → supervisely-6.73.325}/supervisely/convert/volume/__init__.py

@@ -2,3 +2,6 @@
 from supervisely.convert.volume.sly.sly_volume_converter import SLYVolumeConverter
 from supervisely.convert.volume.dicom.dicom_converter import DICOMConverter
 from supervisely.convert.volume.nii.nii_volume_converter import NiiConverter
+from supervisely.convert.volume.nii.nii_planes_volume_converter import (
+    NiiPlaneStructuredConverter,
+)

supervisely-6.73.325/supervisely/convert/volume/nii/nii_planes_volume_converter.py

@@ -0,0 +1,114 @@
+import os
+from collections import defaultdict
+from pathlib import Path
+
+from supervisely import ProjectMeta, logger
+from supervisely.annotation.obj_class import ObjClass
+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.volume.volume import is_nifti_file
+from supervisely.volume_annotation.volume_annotation import VolumeAnnotation
+from supervisely.volume_annotation.volume_object import VolumeObject
+
+
+class NiiPlaneStructuredConverter(NiiConverter, VolumeConverter):
+    """Convert NIfTI 3D volume file to Supervisely format.
+    The NIfTI file should be structured as follows:
+    - <prefix>_anatomic_<idx>.nii (or .nii.gz)
+    - <prefix>_inference_<idx>.nii (or .nii.gz)
+        where   <prefix> is one of the following: cor, sag, axl
+                <idx> is the index of the volume (to match volumes with annotations)
+
+    Supports .nii and .nii.gz files.
+
+    Example:
+        📂 .
+        ├── 🩻 axl_anatomic_1.nii
+        ├── 🩻 axl_inference_1.nii class 1 (may contain multiple instances of the same class)
+        ├── 🩻 cor_anatomic_1.nii
+        ├── 🩻 cor_inference_1.nii class 1
+        ├── 🩻 sag_anatomic_1.nii
+        ├── 🩻 sag_inference_1.nii class 1
+        ├── 🩻 sag_inference_2.nii class 2
+        └── 🩻 sag_inference_3.nii class 3
+    """
+
+    def validate_format(self) -> bool:
+        # create Items
+        converted_dir_name = "converted"
+
+        volumes_dict = defaultdict(list)
+        ann_dict = defaultdict(list)
+
+        for root, _, files in os.walk(self._input_data):
+            if converted_dir_name in root:
+                continue
+            for file in files:
+                path = os.path.join(root, file)
+                if is_nifti_file(path):
+                    full_name = get_file_name(path)
+                    if full_name.endswith(".nii"):
+                        full_name = get_file_name(full_name)
+                    prefix = full_name.split("_")[0]
+                    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)
+
+        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)
+                self._items.append(item)
+            elif len(paths) > 1:
+                logger.info(
+                    f"Found {len(paths)} volumes with prefix {prefix}. Will try to match them by directories."
+                )
+                for path in paths:
+                    item = self.Item(item_path=path)
+                    possible_ann_paths = []
+                    for ann_path in ann_dict.get(prefix):
+                        if Path(ann_path).parent == Path(path).parent:
+                            possible_ann_paths.append(ann_path)
+                    item.ann_data = possible_ann_paths
+                    self._items.append(item)
+        self._meta = ProjectMeta()
+        return self.items_count > 0
+
+    def to_supervisely(
+        self,
+        item: VolumeConverter.Item,
+        meta: ProjectMeta = None,
+        renamed_classes: dict = None,
+        renamed_tags: dict = None,
+    ) -> VolumeAnnotation:
+        """Convert to Supervisely format."""
+
+        try:
+            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}"
+                    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)
+                        meta = meta.add_obj_class(obj_class)
+                        self._meta_changed = True
+                        self._meta = meta
+                    obj = VolumeObject(obj_class, mask_3d=mask)
+                    spatial_figures.append(obj.figure)
+                    objs.append(obj)
+            return VolumeAnnotation(item.volume_meta, objects=objs, spatial_figures=spatial_figures)
+        except Exception as e:
+            logger.warning(f"Failed to convert {item.path} to Supervisely format: {e}")
+            return item.create_empty_annotation()

supervisely-6.73.325/supervisely/convert/volume/nii/nii_volume_converter.py

@@ -0,0 +1,219 @@
+import os
+from pathlib import Path
+
+from supervisely import ProjectMeta, generate_free_name, logger
+from supervisely._utils import batched, is_development
+from supervisely.annotation.obj_class import ObjClass
+from supervisely.api.api import Api
+from supervisely.convert.base_converter import AvailableVolumeConverters
+from supervisely.convert.volume.nii import nii_volume_helper as helper
+from supervisely.convert.volume.volume_converter import VolumeConverter
+from supervisely.geometry.mask_3d import Mask3D
+from supervisely.io.fs import (
+    get_file_ext,
+    get_file_name,
+    get_file_name_with_ext,
+    list_files,
+)
+from supervisely.task.progress import tqdm_sly
+from supervisely.volume.volume import is_nifti_file, read_nrrd_serie_volume_np
+from supervisely.volume_annotation.volume_annotation import VolumeAnnotation
+from supervisely.volume_annotation.volume_object import VolumeObject
+
+
+class NiiConverter(VolumeConverter):
+    """
+    Convert NIfTI 3D volume file to Supervisely format.
+    Supports .nii and .nii.gz files.
+
+    The NIfTI file should be structured as follows:
+    - <volume_name>.nii
+    - <volume_name>/
+        - <cls_name_1>.nii
+        - <cls_name_2>.nii
+        - ...
+    - ...
+
+    where   <volume_name> is the name of the volume
+            If the volume has annotations, they should be in the corresponding directory
+                with the same name as the volume (without extension)
+            <cls_name> is the name of the annotation class
+                <cls_name>.nii:
+                    - represent objects of the single class
+                    - should be unique for the current volume (e.g. tumor.nii.gz, lung.nii.gz)
+                    - can contain multiple objects of the class (each object should be represented by a different value in the mask)
+
+    Example:
+    📂 .
+    ├── 📂 CTChest
+    │   ├── 🩻 lung.nii.gz
+    │   └── 🩻 tumor.nii.gz
+    ├── 🩻 CTChest.nii.gz
+    └── 🩻 Spine.nii.gz
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._supports_links = True
+        self._meta_changed = False
+
+    def __str__(self) -> str:
+        return AvailableVolumeConverters.NII
+
+    def validate_format(self) -> bool:
+        # create Items
+        converted_dir_name = "converted"
+        # nrrds_dict = {}
+        nifti_dict = {}
+        nifti_dirs = {}
+
+        planes_detected = {p: False for p in helper.PlanePrefix.values()}
+
+        for root, _, files in os.walk(self._input_data):
+            dir_name = os.path.basename(root)
+            nifti_dirs[dir_name] = root
+            if converted_dir_name in root:
+                continue
+            for file in files:
+                path = os.path.join(root, file)
+                if is_nifti_file(path):  # is nifti
+                    name = get_file_name(path)
+                    if name.endswith(".nii"):
+                        name = get_file_name(name)
+                    nifti_dict[name] = path
+                    for prefix in planes_detected.keys():
+                        if name.startswith(prefix):
+                            planes_detected[prefix] = True
+
+        if any(planes_detected.values()):
+            return False
+
+        self._items = []
+        skip_files = []
+        for name, nrrd_path in nifti_dict.items():
+            if name in nifti_dirs:
+                item = self.Item(item_path=nrrd_path)
+                ann_dir = nifti_dirs[name]
+                item.ann_data = list_files(ann_dir, [".nii", ".nii.gz", ".gz"], None, True)
+                self._items.append(item)
+                skip_files.extend(item.ann_data)
+                skip_files.append(nrrd_path)
+
+        for name, nrrd_path in nifti_dict.items():
+            if nrrd_path in skip_files:
+                continue
+            item = self.Item(item_path=nrrd_path)
+            self._items.append(item)
+
+        self._meta = ProjectMeta()
+        return self.items_count > 0
+
+    def to_supervisely(
+        self,
+        item: VolumeConverter.Item,
+        meta: ProjectMeta = None,
+        renamed_classes: dict = None,
+        renamed_tags: dict = None,
+    ) -> VolumeAnnotation:
+        """Convert to Supervisely format."""
+
+        try:
+            objs = []
+            spatial_figures = []
+            for ann_path in item.ann_data:
+                ann_name = get_file_name(ann_path)
+                if ann_name.endswith(".nii"):
+                    ann_name = get_file_name(ann_name)
+
+                ann_name = renamed_classes.get(ann_name, ann_name)
+                for mask, _ in helper.get_annotation_from_nii(ann_path):
+                    obj_class = meta.get_obj_class(ann_name)
+                    if obj_class is None:
+                        obj_class = ObjClass(ann_name, Mask3D)
+                        meta = meta.add_obj_class(obj_class)
+                        self._meta_changed = True
+                        self._meta = meta
+                    obj = VolumeObject(obj_class, mask_3d=mask)
+                    spatial_figures.append(obj.figure)
+                    objs.append(obj)
+            return VolumeAnnotation(item.volume_meta, objects=objs, spatial_figures=spatial_figures)
+        except Exception as e:
+            logger.warning(f"Failed to convert {item.path} to Supervisely format: {e}")
+            return item.create_empty_annotation()
+
+    def upload_dataset(
+        self,
+        api: Api,
+        dataset_id: int,
+        batch_size: int = 1,
+        log_progress=True,
+    ):
+        """Upload converted data to Supervisely"""
+
+        meta, renamed_classes, _ = self.merge_metas_with_conflicts(api, dataset_id)
+
+        existing_names = set([vol.name for vol in api.volume.get_list(dataset_id)])
+
+        if log_progress:
+            progress, progress_cb = self.get_progress(
+                self.items_count, "Converting and uploading volumes..."
+            )
+        else:
+            progress_cb = None
+
+        converted_dir_name = "converted"
+        converted_dir = os.path.join(self._input_data, converted_dir_name)
+
+        for batch in batched(self._items, batch_size=batch_size):
+            item_names = []
+            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
+                item.path = helper.nifti_to_nrrd(item.path, converted_dir)
+                ext = get_file_ext(item.path)
+                if ext.lower() != ext:
+                    new_volume_path = Path(item.path).with_suffix(ext.lower()).as_posix()
+                    os.rename(item.path, new_volume_path)
+                    item.path = new_volume_path
+                item.name = get_file_name_with_ext(item.path)
+                item.name = generate_free_name(
+                    existing_names, item.name, with_ext=True, extend_used_names=True
+                )
+                item_names.append(item.name)
+                item_paths.append(item.path)
+
+                # upload volume
+                volume_np, volume_meta = read_nrrd_serie_volume_np(item.path)
+                progress_nrrd = tqdm_sly(
+                    desc=f"Uploading volume '{item.name}'",
+                    total=sum(volume_np.shape),
+                    leave=True if progress_cb is None else False,
+                    position=1,
+                )
+                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
+
+                # create and upload annotation
+                if item.ann_data is not None:
+                    ann = self.to_supervisely(item, meta, renamed_classes, None)
+
+                    if self._meta_changed:
+                        meta, renamed_classes, _ = self.merge_metas_with_conflicts(api, dataset_id)
+
+                    api.volume.annotation.append(info.id, ann)
+
+            if log_progress:
+                progress_cb(len(batch))
+
+        if log_progress:
+            if is_development():
+                progress.close()
+        logger.info(f"Dataset ID:{dataset_id} has been successfully uploaded.")

{supervisely-6.73.323 → supervisely-6.73.325}/supervisely/convert/volume/nii/nii_volume_helper.py

@@ -4,10 +4,22 @@ from typing import Generator
 import nrrd
 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.volume.volume import convert_3d_nifti_to_nrrd
 
+VOLUME_NAME = "anatomic"
+LABEL_NAME = ["inference", "label", "annotation", "mask", "segmentation"]
+
+
+class PlanePrefix(str, StrEnum):
+    """Prefix for plane names."""
+
+    CORONAL = "cor"
+    SAGITTAL = "sag"
+    AXIAL = "axl"
+
 
 def nifti_to_nrrd(nii_file_path: str, converted_dir: str) -> str:
     """Convert NIfTI 3D volume file to NRRD 3D volume file."""