synapse-sdk 1.0.0a78__tar.gz → 1.0.0a79__tar.gz

{synapse_sdk-1.0.0a78 → synapse_sdk-1.0.0a79}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a78
+Version: 1.0.0a79
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a78 → synapse_sdk-1.0.0a79}/synapse_sdk/utils/converters/__init__.py

@@ -2,16 +2,25 @@ import json
 import os
 import shutil
 import uuid
+from typing import IO
 
 
 class BaseConverter:
     """Base class for shared logic between converters."""
 
-    def __init__(self, root_dir: str, is_categorized_dataset: bool = False) -> None:
+    def __init__(
+        self, root_dir: str = None, is_categorized_dataset: bool = False, is_single_conversion: bool = False
+    ) -> None:
         self.root_dir: str = root_dir
         self.is_categorized_dataset: bool = is_categorized_dataset
+        self.is_single_conversion: bool = is_single_conversion
         self.converted_data = None
 
+        # Set directories if single_conversion is False.
+        if not is_single_conversion:
+            if not root_dir:
+                raise ValueError('root_dir must be specified for conversion')
+
     @staticmethod
     def ensure_dir(path: str) -> None:
         """Ensure that the directory exists, creating it if necessary."""
@@ -55,6 +64,24 @@ class BaseConverter:
 
         return splits
 
+    def convert_single_file(self, data, original_file: IO, **kwargs):
+        """Convert a single data object and corresponding original file.
+
+        This method should be implemented by subclasses for single file conversion.
+        Only available when is_single_conversion=True.
+
+        Args:
+            data: The data object to convert (dict for JSON data, etc.)
+            original_file_path: File object for the corresponding original file
+            **kwargs: Additional parameters specific to each converter
+
+        Returns:
+            Converted data in the target format
+        """
+        if not self.is_single_conversion:
+            raise RuntimeError('convert_single_file is only available when is_single_conversion=True')
+        raise NotImplementedError('Subclasses must implement convert_single_file method')
+
     def _set_directories(self, split=None):
         """Set `self.json_dir` and `self.original_file_dir` based on the dataset split."""
         if split:
@@ -106,8 +133,10 @@ class FromDMConverter(BaseConverter):
         converter.save_to_folder('/my/target/output')  # Writes files/folders to output location
     """
 
-    def __init__(self, root_dir: str, is_categorized_dataset: bool = False) -> None:
-        super().__init__(root_dir, is_categorized_dataset)
+    def __init__(
+        self, root_dir: str = None, is_categorized_dataset: bool = False, is_single_conversion: bool = False
+    ) -> None:
+        super().__init__(root_dir, is_categorized_dataset, is_single_conversion)
         self.version: str = '1.0'
 
     def convert(self):

{synapse_sdk-1.0.0a78 → synapse_sdk-1.0.0a79}/synapse_sdk/utils/converters/yolo/from_dm.py

@@ -2,7 +2,7 @@ import json
 import os
 import shutil
 from glob import glob
-from typing import Any, Dict, List, Optional, Union
+from typing import IO, Any, Dict, List, Optional, Union
 
 from PIL import Image
 
@@ -14,8 +14,8 @@ class FromDMToYOLOConverter(FromDMConverter):
 
     IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.bmp']
 
-    def __init__(self, root_dir: str, is_categorized_dataset: bool = False):
-        super().__init__(root_dir, is_categorized_dataset)
+    def __init__(self, root_dir: str = None, is_categorized_dataset: bool = False, is_single_conversion: bool = False):
+        super().__init__(root_dir, is_categorized_dataset, is_single_conversion)
         self.class_names: List[str] = []
         self.class_map: Dict[str, int] = {}
         self.dataset_yaml_content: str = ''
@@ -300,3 +300,85 @@ class FromDMToYOLOConverter(FromDMConverter):
             for c in self.class_names:
                 f.write(f'{c}\n')
         print(f'YOLO data exported to {output_dir}')
+
+    def convert_single_file(
+        self, data: Dict[str, Any], original_file: IO, class_names: Optional[List[str]] = None
+    ) -> Dict[str, Any]:
+        """Convert a single DM data dict and corresponding image file object to YOLO format.
+
+        Args:
+            data: DM format data dictionary (JSON content)
+            original_file: File object for the corresponding original image
+            class_names: Optional list of class names. If not provided, classes will be extracted from data.
+
+        Returns:
+            Dictionary containing YOLO format data for the single file
+        """
+        if not self.is_single_conversion:
+            raise RuntimeError('convert_single_file is only available when is_single_conversion=True')
+
+        if class_names is None:
+            classes = set()
+            for img_ann in data['images']:
+                for k in ['bounding_box', 'polygon', 'keypoint']:
+                    if k in img_ann:
+                        for ann in img_ann[k]:
+                            classes.add(ann['classification'])
+            class_names = sorted(list(classes))
+
+        class_map = {name: idx for idx, name in enumerate(class_names)}
+        # You need to update get_image_size to accept a file object
+        width, height = self.get_image_size(original_file)
+
+        img_ann = data['images'][0]
+        label_lines = []
+
+        # bbox
+        if 'bounding_box' in img_ann:
+            for box in img_ann['bounding_box']:
+                if box['classification'] not in class_map:
+                    continue
+                cidx = class_map[box['classification']]
+                x, y, w, h = box['data']
+                cx = x + w / 2
+                cy = y + h / 2
+                cx /= width
+                cy /= height
+                w /= width
+                h /= height
+                label_lines.append(f'{cidx} {cx:.6f} {cy:.6f} {w:.6f} {h:.6f}')
+
+        # polygon
+        if 'polygon' in img_ann:
+            for poly in img_ann['polygon']:
+                if poly['classification'] not in class_map:
+                    continue
+                cidx = class_map[poly['classification']]
+                poly_str = self.polygon_to_yolo_string(poly['data'], width, height)
+                if poly_str:
+                    label_lines.append(f'{cidx} {poly_str}')
+
+        # keypoint
+        if 'keypoint' in img_ann:
+            for kp in img_ann['keypoint']:
+                if kp['classification'] not in class_map:
+                    continue
+                cidx = class_map[kp['classification']]
+                if 'bounding_box' in kp:
+                    x, y, w, h = kp['bounding_box']
+                    cx = x + w / 2
+                    cy = y + h / 2
+                    cx /= width
+                    cy /= height
+                    w /= width
+                    h /= height
+                else:
+                    cx, cy, w, h = 0.5, 0.5, 1.0, 1.0
+                kp_str = self.keypoints_to_yolo_string(kp['data'], width, height)
+                label_lines.append(f'{cidx} {cx:.6f} {cy:.6f} {w:.6f} {h:.6f} {kp_str}')
+
+        return {
+            'label_lines': label_lines,
+            'class_names': class_names,
+            'class_map': class_map,
+        }

{synapse_sdk-1.0.0a78 → synapse_sdk-1.0.0a79}/synapse_sdk/utils/converters/yolo/to_dm.py

@@ -12,6 +12,9 @@ class YOLOToDMConverter(ToDMConverter):
 
     IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.bmp']
 
+    def __init__(self, root_dir: str = None, is_categorized_dataset: bool = False, is_single_conversion: bool = False):
+        super().__init__(root_dir, is_categorized_dataset, is_single_conversion)
+
     def convert(self):
         """Convert YOLO dataset to DM format."""
         if self.is_categorized_dataset:
@@ -183,3 +186,69 @@ class YOLOToDMConverter(ToDMConverter):
             dm_json = {'images': [dm_img]}
             result[os.path.basename(img_path)] = (dm_json, img_path)
         return result
+
+    def convert_single_file(self, data: List[str], original_file_path: str, class_names: List[str]) -> Dict[str, Any]:
+        """Convert a single YOLO label data and corresponding image to DM format.
+
+        Args:
+            data: List of YOLO label lines (strings from .txt file content)
+            original_file_path: Path to the corresponding original file
+            class_names: List of class names corresponding to indices in the label data
+
+        Returns:
+            Dictionary containing DM format data for the single file
+        """
+        if not self.is_single_conversion:
+            raise RuntimeError('convert_single_file is only available when is_single_conversion=True')
+
+        if not os.path.exists(original_file_path):
+            raise FileNotFoundError(f'Original file not found: {original_file_path}')
+
+        img_size = self._get_image_size(original_file_path)
+
+        # data is already a list of label lines
+        label_lines = [line.strip() for line in data if line.strip()]
+
+        # Prepare DM annotation structure
+        dm_img = {
+            'bounding_box': [],
+            'polygon': [],
+            'keypoint': [],
+            'relation': [],
+            'group': [],
+        }
+
+        for line in label_lines:
+            ann = self._parse_yolo_line(line, class_names, img_size)
+            if ann is None:
+                continue
+
+            if ann['type'] == 'bounding_box':
+                dm_img['bounding_box'].append({
+                    'id': self._generate_unique_id(),
+                    'classification': ann['classification'],
+                    'attrs': [],
+                    'data': ann['data'],
+                })
+            elif ann['type'] == 'polygon':
+                dm_img['polygon'].append({
+                    'id': self._generate_unique_id(),
+                    'classification': ann['classification'],
+                    'attrs': [],
+                    'data': ann['data'],
+                })
+            elif ann['type'] == 'keypoint':
+                dm_img['keypoint'].append({
+                    'id': self._generate_unique_id(),
+                    'classification': ann['classification'],
+                    'attrs': [],
+                    'data': ann['data'],
+                    'bounding_box': ann['bounding_box'],
+                })
+
+        dm_json = {'images': [dm_img]}
+        return {
+            'dm_json': dm_json,
+            'image_path': original_file_path,
+            'image_name': os.path.basename(original_file_path),
+        }

{synapse_sdk-1.0.0a78 → synapse_sdk-1.0.0a79}/synapse_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a78
+Version: 1.0.0a79
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT