label-studio-sdk 1.0.11__tar.gz → 1.0.12__tar.gz

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: label-studio-sdk
-Version: 1.0.11
+Version: 1.0.12
 Summary: 
 Requires-Python: >=3.9,<4
 Classifier: Intended Audience :: Developers
@@ -28,7 +28,7 @@ Requires-Dist: jsf (>=0.11.2,<0.12.0)
 Requires-Dist: jsonschema (>=4.23.0)
 Requires-Dist: lxml (>=4.2.5)
 Requires-Dist: nltk (>=3.9.1,<4.0.0)
-Requires-Dist: numpy (>=1.26.4,<2.0.0)
+Requires-Dist: numpy (>=1.26.4,<3.0.0)
 Requires-Dist: pandas (>=0.24.0)
 Requires-Dist: pydantic (>=1.9.2)
 Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/pyproject.toml

@@ -3,7 +3,7 @@ name = "label-studio-sdk"
 
 [tool.poetry]
 name = "label-studio-sdk"
-version = "1.0.11"
+version = "1.0.12"
 description = ""
 readme = "README.md"
 authors = []
@@ -44,7 +44,7 @@ jsf = "^0.11.2"
 jsonschema = ">=4.23.0"
 lxml = ">=4.2.5"
 nltk = "^3.9.1"
-numpy = ">=1.26.4,<2.0.0"
+numpy = ">=1.26.4,<3.0.0"
 pandas = ">=0.24.0"
 pydantic = ">= 1.9.2"
 pydantic-core = "^2.18.2"

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/__init__.py

@@ -3,6 +3,7 @@
 from .types import (
     AccessTokenResponse,
     Annotation,
+    AnnotationCompletedBy,
     AnnotationFilterOptions,
     AnnotationLastAction,
     AnnotationsDmField,
@@ -85,6 +86,7 @@ from .types import (
     RedisImportStorageStatus,
     RefinedPromptResponse,
     RefinedPromptResponseRefinementStatus,
+    RotateTokenResponse,
     S3ExportStorage,
     S3ExportStorageStatus,
     S3ImportStorage,
@@ -200,6 +202,7 @@ __all__ = [
     "ActionsCreateRequestSelectedItemsExcluded",
     "ActionsCreateRequestSelectedItemsIncluded",
     "Annotation",
+    "AnnotationCompletedBy",
     "AnnotationFilterOptions",
     "AnnotationLastAction",
     "AnnotationsCreateBulkRequestSelectedItems",
@@ -307,6 +310,7 @@ __all__ = [
     "RedisImportStorageStatus",
     "RefinedPromptResponse",
     "RefinedPromptResponseRefinementStatus",
+    "RotateTokenResponse",
     "S3ExportStorage",
     "S3ExportStorageStatus",
     "S3ImportStorage",

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/converter/converter.py

@@ -59,6 +59,7 @@ class Format(Enum):
     YOLO_WITH_IMAGES = 14
     COCO_WITH_IMAGES = 15
     YOLO_OBB_WITH_IMAGES = 16
+    BRUSH_TO_COCO = 17 
 
     def __str__(self):
         return self.name
@@ -167,6 +168,13 @@ class Converter(object):
             "format expected by NVIDIA NeMo models.",
             "link": "https://labelstud.io/guide/export.html#ASR-MANIFEST",
             "tags": ["speech recognition"],
+        },   
+        Format.BRUSH_TO_COCO: {
+
+            "title": "Brush labels to COCO",
+            "description": "Export your brush labels as COCO format for segmentation tasks. Converts RLE encoded masks to COCO polygons.",
+            "link": "https://labelstud.io/guide/export.html#COCO",
+            "tags": ["image segmentation", "brush annotations"],
         },
     }
 
@@ -293,6 +301,19 @@ class Converter(object):
                 upload_dir=self.upload_dir,
                 download_resources=self.download_resources,
             )
+        elif format == Format.BRUSH_TO_COCO:
+            items = (
+                self.iter_from_dir(input_data)
+                if is_dir
+                else self.iter_from_json_file(input_data)
+            )
+            from label_studio_sdk.converter.exports.brush_to_coco import convert_to_coco
+            image_dir = kwargs.get("image_dir")
+            convert_to_coco(
+                items, 
+                output_data, 
+                output_image_dir=image_dir
+            )
 
     def _get_data_keys_and_output_tags(self, output_tags=None):
         data_keys = set()
@@ -376,6 +397,7 @@ class Converter(object):
         ):
             all_formats.remove(Format.BRUSH_TO_NUMPY.name)
             all_formats.remove(Format.BRUSH_TO_PNG.name)
+            all_formats.remove(Format.BRUSH_TO_COCO.name) 
         if not (
             ("Audio" in input_tag_types or "AudioPlus" in input_tag_types)
             and "TextArea" in output_tag_types

label_studio_sdk-1.0.12/src/label_studio_sdk/converter/exports/brush_to_coco.py

@@ -0,0 +1,332 @@
+"""
+Convert brush annotations and polygon annotations to COCO format.
+This module handles RLE encoded brush masks and converts them to COCO segmentation format.
+"""
+import logging
+import random
+import io
+import os
+import json
+import numpy as np
+import cv2
+from copy import deepcopy
+from datetime import datetime
+
+from label_studio_sdk.converter.utils import ensure_dir, get_annotator
+import label_studio_sdk.converter.brush as brush_module
+
+logger = logging.getLogger(__name__)
+
+
+def generate_random_color():
+    """Generates a random hex color for category visualization"""
+    return '#{:02x}{:02x}{:02x}'.format(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
+
+
+def flatten(item):
+    """Flatten nested lists for segmentation format"""
+    if isinstance(item, list):
+        for subitem in item:
+            yield from flatten(subitem)
+    else:
+        yield item
+
+
+def generate_contour_from_rle(rle, original_width, original_height):
+    """Convert RLE encoded masks to COCO segmentation format
+    
+    Args:
+        rle: RLE encoded mask data
+        original_width: Width of the original image
+        original_height: Height of the original image
+        
+    Returns:
+        tuple: (segmentation, bbox_list, area_list) in COCO format
+    """
+    # Ensure rle is bytes, not string
+    if isinstance(rle, str):
+        rle = rle.encode('utf-8')
+        
+    # Decode RLE
+    try:
+        rle_binary = brush_module.decode_rle(rle)
+    except Exception as e:
+        logger.warning(f"Failed to decode RLE: {e}")
+        return [], [], []
+    
+    # Reshape to image dimensions with 4 channels (RGBA)
+    try:
+        reshaped_image = np.reshape(rle_binary, [original_height, original_width, 4])
+    except Exception as e:
+        logger.warning(f"Failed to reshape RLE data: {e}")
+        return [], [], []
+    
+    # Use only the alpha channel for contour detection
+    alpha_channel = np.array(reshaped_image[:, :, 3]).astype('uint8')
+    
+    # Find contours
+    contours, _ = cv2.findContours(
+        alpha_channel, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
+    )
+
+    segmentation = []
+    bbox_list = []
+    area_list = []
+    for contour in contours:
+        # OpenCV contours are in format [[[x1,y1]], [[x2,y2]], ...], 
+        # we need to convert to a flat list [x1,y1,x2,y2,...]
+        contour_points = [point[0] for point in contour]
+        
+        # Flatten to a single list
+        contour_flat = []
+        for point in contour_points:
+            contour_flat.extend(point.tolist())
+        
+        # Check if we have at least 3 points (6 coordinates) for a valid polygon
+        if len(contour_points) >= 3:
+            segmentation.append(contour_flat)
+
+            x, y, w, h = cv2.boundingRect(contour)
+            bbox_list.append([x, y, w, h])   # Format: [top-left x, top-left y, width, height]
+
+            # Compute the area of the contour
+            area = cv2.contourArea(contour)
+            area_list.append(int(area))
+
+    return segmentation, bbox_list, area_list
+
+
+def generate_contour_from_polygon(points, original_width, original_height):
+    """Convert polygon annotations to COCO segmentation format
+    
+    Args:
+        points: List of [x, y] points in percentage format
+        original_width: Width of the original image
+        original_height: Height of the original image
+        
+    Returns:
+        tuple: (segmentation, bbox, area) in COCO format
+    """
+    # Convert from percentage to absolute coordinates
+    modified_points = [
+        [round(x * original_width / 100), round(y * original_height / 100)]
+        for x, y in points
+    ]
+
+    # Initialize lists
+    segmentation = [list(flatten(modified_points))]
+
+    # Convert points to a NumPy array and reshape for OpenCV
+    contour = np.array(modified_points, dtype=np.float32).reshape((-1, 1, 2))
+
+    # Calculate bounding box
+    bbox = cv2.boundingRect(contour)
+    bbox = [int(coordinate) for coordinate in bbox]
+
+    # Compute the area of the contour
+    area = cv2.contourArea(contour)
+
+    return segmentation, bbox, int(area)
+
+
+def convert_to_coco(items, output_dir, output_image_dir=None):
+    """Convert Label Studio annotations to COCO format
+
+    Args:
+        items: Iterable of annotation items
+        output_dir: Directory where to save the resulting COCO JSON
+        output_image_dir: Optional directory for images
+
+    Returns:
+        Path to the generated COCO JSON file
+    """
+    ensure_dir(output_dir)
+    output_file = os.path.join(output_dir, "result_coco.json")
+    
+    if output_image_dir is not None:
+        ensure_dir(output_image_dir)
+    else:
+        output_image_dir = os.path.join(output_dir, "images")
+        os.makedirs(output_image_dir, exist_ok=True)
+
+    # Initialize COCO JSON structure
+    coco_data = {
+        'images': [],
+        'annotations': [],
+        'categories': [],
+        'info': {
+            'year': datetime.now().year,
+            'version': '1.0',
+            'description': 'Converted from Label Studio',
+            'contributor': 'Label Studio Converter',
+            'date_created': str(datetime.now()),
+        }
+    }
+
+    category_map = {}  # To track unique categories
+    annotation_id = 0
+
+    # Process each item
+    for image_id, item in enumerate(items):
+        # Extract image information
+        image_path = None
+        for key, value in item['input'].items():
+            if isinstance(value, str) and any(value.lower().endswith(ext) for ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']):
+                image_path = value
+                break
+        
+        if not image_path:
+            logger.warning(f"No image path found for item {item.get('id')}. Skipping.")
+            continue
+
+        image_info = {
+            'id': image_id,
+            'file_name': os.path.basename(image_path),
+            'path': image_path,
+            'width': None,  # Will be filled from annotations
+            'height': None  # Will be filled from annotations
+        }
+
+        # Skip tasks without annotations
+        if not item['output']:
+            logger.warning(f"No annotations found for item {item.get('id')}")
+            continue
+
+        valid_annotations = False
+
+        # Process all annotations
+        for output_key, annotations in item['output'].items():
+            for annotation in annotations:
+                # Get image dimensions from the first annotation
+                if image_info['width'] is None and 'original_width' in annotation:
+                    image_info['width'] = annotation['original_width']
+                    image_info['height'] = annotation['original_height']
+                
+                # Identify annotation type and extract labels
+                annotation_type = annotation.get('type', '').lower()
+                labels = []
+                
+                if 'brushlabels' in annotation:
+                    labels = annotation['brushlabels']
+                    type_key = 'brushlabels'
+                elif 'polygonlabels' in annotation:
+                    labels = annotation['polygonlabels']
+                    type_key = 'polygonlabels'
+                elif 'rectanglelabels' in annotation:
+                    labels = annotation['rectanglelabels']
+                    type_key = 'rectanglelabels'
+                elif 'labels' in annotation:
+                    labels = annotation['labels']
+                    type_key = 'labels'
+                else:
+                    logger.debug(f"Unsupported annotation type: {annotation_type}")
+                    continue
+                
+                # Process each label
+                for label in labels:
+                    # Add new categories to the map
+                    if label not in category_map:
+                        category_id = len(category_map) + 1
+                        category_map[label] = category_id
+                        coco_data['categories'].append({
+                            'id': category_id,
+                            'name': label,
+                            'supercategory': '',
+                            'color': generate_random_color(),
+                            'metadata': {}
+                        })
+                    
+                    category_id = category_map[label]
+                    
+                    # Convert annotation based on type
+                    if 'rle' in annotation and type_key == 'brushlabels':
+
+                        # check required keys exist
+                        if not all(k in annotation for k in ['rle', 'original_width', 'original_height']):
+                            logger.warning(f"Missing required keys for RLE annotation. Skipping.")
+                            continue
+
+                        # Process brush annotation (RLE encoded mask)
+                        segmentations, bboxes, areas = generate_contour_from_rle(
+                            annotation['rle'], 
+                            annotation['original_width'], 
+                            annotation['original_height']
+                        )
+                        
+                        # Create separate annotation for each contour
+                        for i, (segmentation, bbox, area) in enumerate(zip(segmentations, bboxes, areas)):
+                            coco_annotation = {
+                                'id': annotation_id,
+                                'image_id': image_id,
+                                'category_id': category_id,
+                                'segmentation': [segmentation],  # COCO format requires double nesting
+                                'area': area,
+                                'bbox': bbox,
+                                'iscrowd': 0,
+                                'annotator': get_annotator(item, "unknown")
+                            }
+                            coco_data['annotations'].append(coco_annotation)
+                            annotation_id += 1
+                            valid_annotations = True
+                    
+                    elif 'points' in annotation and type_key == 'polygonlabels':
+                        # check required keys exist
+                        if not all(k in annotation for k in ['points', 'original_width', 'original_height']):
+                            logger.warning(f"Missing required keys for polygon annotation. Skipping.")
+                            continue
+                        # Process polygon annotation
+                        segmentation, bbox, area = generate_contour_from_polygon(
+                            annotation['points'], 
+                            annotation['original_width'], 
+                            annotation['original_height']
+                        )
+                        
+                        coco_annotation = {
+                            'id': annotation_id,
+                            'image_id': image_id,
+                            'category_id': category_id,
+                            'segmentation': segmentation,
+                            'area': area,
+                            'bbox': bbox,
+                            'iscrowd': 0,
+                            'annotator': get_annotator(item, "unknown")
+                        }
+                        coco_data['annotations'].append(coco_annotation)
+                        annotation_id += 1
+                        valid_annotations = True
+                    
+                    elif annotation_type == 'rectanglelabels' or type_key == 'labels':
+                        # check required keys exist
+                        if not all(k in annotation for k in ['x', 'y', 'width', 'height', 'original_width', 'original_height']):
+                            logger.warning(f"Missing required keys for rectangle annotation. Skipping.")
+                            continue
+
+                        # Convert from percentage to absolute coordinates
+                        x = annotation['x'] * annotation['original_width'] / 100
+                        y = annotation['y'] * annotation['original_height'] / 100
+                        w = annotation['width'] * annotation['original_width'] / 100
+                        h = annotation['height'] * annotation['original_height'] / 100
+                        
+                        coco_annotation = {
+                            'id': annotation_id,
+                            'image_id': image_id,
+                            'category_id': category_id,
+                            'segmentation': [],  # Empty for bounding boxes
+                            'area': w * h,
+                            'bbox': [x, y, w, h],
+                            'iscrowd': 0,
+                            'annotator': get_annotator(item, "unknown")
+                        }
+                        coco_data['annotations'].append(coco_annotation)
+                        annotation_id += 1
+                        valid_annotations = True
+        
+        # Add image to the dataset if it has valid dimensions
+        if image_info['width'] is not None and image_info['height'] is not None and valid_annotations:
+            coco_data['images'].append(image_info)
+
+    # Write COCO JSON to file
+    with io.open(output_file, 'w', encoding='utf8') as f:
+        json.dump(coco_data, f, indent=2, ensure_ascii=False)
+    
+    return output_file

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/converter/main.py

@@ -79,6 +79,12 @@ def get_export_args(parser):
         default=True,
         help="Set this flag if your annotations are in one JSON file instead of multiple JSON files from directory",
     )
+    parser.add_argument(
+        "--brush-to-coco",
+        dest="brush_to_coco",
+        action="store_true",
+        help="If set, brush annotations will be converted to COCO format"
+    )
 
 
 def get_all_args():
@@ -158,6 +164,14 @@ def export(args):
         c.convert_to_yolo(
             args.input, args.output, is_dir=not args.heartex_format, is_obb=True
         )
+    elif args.format == Format.BRUSH_TO_COCO:
+        c.convert(
+            args.input,
+            args.output,
+            Format.BRUSH_TO_COCO,
+            output_image_dir=args.image_dir,
+            is_dir=not args.heartex_format,
+        )
     else:
         raise FormatNotSupportedError()
 

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/jwt_settings/client.py

@@ -19,7 +19,7 @@ class JwtSettingsClient:
 
     def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> JwtSettingsResponse:
         """
-        Retrieve JWT settings for the currently active organization.
+        Retrieve JWT settings for the currently-active organization.
 
         Parameters
         ----------
@@ -134,7 +134,7 @@ class AsyncJwtSettingsClient:
 
     async def get(self, *, request_options: typing.Optional[RequestOptions] = None) -> JwtSettingsResponse:
         """
-        Retrieve JWT settings for the currently active organization.
+        Retrieve JWT settings for the currently-active organization.
 
         Parameters
         ----------

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/tokens/client.py

@@ -10,6 +10,8 @@ from ..core.api_error import ApiError
 from ..types.api_token_response import ApiTokenResponse
 from ..types.access_token_response import AccessTokenResponse
 from ..errors.unauthorized_error import UnauthorizedError
+from ..types.rotate_token_response import RotateTokenResponse
+from ..errors.bad_request_error import BadRequestError
 from ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -225,6 +227,70 @@ class TokensClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def rotate(self, *, refresh: str, request_options: typing.Optional[RequestOptions] = None) -> RotateTokenResponse:
+        """
+        Blacklist existing refresh token, and get a new refresh token.
+
+        Parameters
+        ----------
+        refresh : str
+            JWT refresh token
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        RotateTokenResponse
+            Refresh token successfully rotated
+
+        Examples
+        --------
+        from label_studio_sdk import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.tokens.rotate(
+            refresh="refresh",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/token/rotate",
+            method="POST",
+            json={
+                "refresh": refresh,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    RotateTokenResponse,
+                    parse_obj_as(
+                        type_=RotateTokenResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
 
 class AsyncTokensClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -468,3 +534,77 @@ class AsyncTokensClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def rotate(
+        self, *, refresh: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> RotateTokenResponse:
+        """
+        Blacklist existing refresh token, and get a new refresh token.
+
+        Parameters
+        ----------
+        refresh : str
+            JWT refresh token
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        RotateTokenResponse
+            Refresh token successfully rotated
+
+        Examples
+        --------
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tokens.rotate(
+                refresh="refresh",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/token/rotate",
+            method="POST",
+            json={
+                "refresh": refresh,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    RotateTokenResponse,
+                    parse_obj_as(
+                        type_=RotateTokenResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

{label_studio_sdk-1.0.11 → label_studio_sdk-1.0.12}/src/label_studio_sdk/types/__init__.py

@@ -2,6 +2,7 @@
 
 from .access_token_response import AccessTokenResponse
 from .annotation import Annotation
+from .annotation_completed_by import AnnotationCompletedBy
 from .annotation_filter_options import AnnotationFilterOptions
 from .annotation_last_action import AnnotationLastAction
 from .annotations_dm_field import AnnotationsDmField
@@ -84,6 +85,7 @@ from .redis_import_storage import RedisImportStorage
 from .redis_import_storage_status import RedisImportStorageStatus
 from .refined_prompt_response import RefinedPromptResponse
 from .refined_prompt_response_refinement_status import RefinedPromptResponseRefinementStatus
+from .rotate_token_response import RotateTokenResponse
 from .s3export_storage import S3ExportStorage
 from .s3export_storage_status import S3ExportStorageStatus
 from .s3import_storage import S3ImportStorage
@@ -108,6 +110,7 @@ from .workspace import Workspace
 __all__ = [
     "AccessTokenResponse",
     "Annotation",
+    "AnnotationCompletedBy",
     "AnnotationFilterOptions",
     "AnnotationLastAction",
     "AnnotationsDmField",
@@ -190,6 +193,7 @@ __all__ = [
     "RedisImportStorageStatus",
     "RefinedPromptResponse",
     "RefinedPromptResponseRefinementStatus",
+    "RotateTokenResponse",
     "S3ExportStorage",
     "S3ExportStorageStatus",
     "S3ImportStorage",