PyPI - lightly-studio - Versions diffs - 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl - Mend

lightly-studio 0.3.2py3-none-any.whl → 0.3.4py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of lightly-studio might be problematic. Click here for more details.

Files changed (163) hide show

lightly_studio/dist_lightly_studio_view_app/_app/immutable/workers/search.worker-vNSty3B0.js ADDED Viewed

@@ -0,0 +1 @@

lightly_studio/dist_lightly_studio_view_app/_app/version.json CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":"~~1758802389068~~"}
1	+ {"version":"1760445896529"}

lightly_studio/dist_lightly_studio_view_app/index.html CHANGED Viewed

@@ -5,35 +5,36 @@
         <link rel="icon" href="/favicon.png" />
         <meta name="viewport" content="width=device-width, initial-scale=1" />
-		<link rel="modulepreload" href="/_app/immutable/entry/start.CcsRl3cZ.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/2O287xak.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/DcGCxgpH.js">
+		<link rel="modulepreload" href="/_app/immutable/entry/start.SNvc2nrm.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/DQ8aZ1o-.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/Frwd2CjB.js">
 		<link rel="modulepreload" href="/_app/immutable/chunks/DIeogL5L.js">
 		<link rel="modulepreload" href="/_app/immutable/chunks/CYgJF_JY.js">
-		<link rel="modulepreload" href="/_app/immutable/entry/app.BI-EA5gL.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/D8GZDMNN.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/DqUGznj_.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/BqBqV92V.js">
+		<link rel="modulepreload" href="/_app/immutable/entry/app.BPc0HQPq.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/C1FmrZbK.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/Dj4O-5se.js">
 		<link rel="modulepreload" href="/_app/immutable/chunks/CWj6FrbW.js">
 		<link rel="modulepreload" href="/_app/immutable/chunks/69_IOA4Y.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/Ccq4ZD0B.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/KpAtIldw.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/C98Hk3r5.js">
-		<link rel="modulepreload" href="/_app/immutable/chunks/Crk-jcvV.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/B7302SU7.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/H60ATh8g.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/Dug7Bq1S.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/CiaNZCBa.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/-DXuGN29.js">
+		<link rel="modulepreload" href="/_app/immutable/chunks/Dafy4oEQ.js">
     </head>
     <body data-sveltekit-preload-data="hover" class="flex h-screen w-screen flex-col dark:bg-black">
         <div style="display: contents">
 			<script>
 				{
-					__sveltekit_1hp03lk = {
+					__sveltekit_7kclq1 = {
 						base: ""
 					};
 					const element = document.currentScript.parentElement;
 					Promise.all([
-						import("/_app/immutable/entry/start.CcsRl3cZ.js"),
-						import("/_app/immutable/entry/app.BI-EA5gL.js")
+						import("/_app/immutable/entry/start.SNvc2nrm.js"),
+						import("/_app/immutable/entry/app.BPc0HQPq.js")
 					]).then(([kit, app]) => {
 						kit.start(app, element);
 					});

lightly_studio/examples/example.py CHANGED Viewed

@@ -5,11 +5,15 @@ from pathlib import Path
 from environs import Env
 import lightly_studio as ls
+from lightly_studio import db_manager
 # Read environment variables
 env = Env()
 env.read_env()
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
 # Define the path to the dataset directory
 dataset_path = Path(env.path("DATASET_PATH", "/path/to/your/dataset"))
 dataset_path = dataset_path.parent if dataset_path.is_file() else dataset_path

lightly_studio/examples/example_coco.py CHANGED Viewed

@@ -1,6 +1,10 @@
 """Example of how to add samples in coco format to a dataset."""
 import lightly_studio as ls
+from lightly_studio import db_manager
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
 # Create a DatasetLoader from a path
 dataset = ls.Dataset.create()

lightly_studio/examples/example_coco_caption.py ADDED Viewed

@@ -0,0 +1,24 @@
+"""Example of how to add samples in coco caption format to a dataset."""
+import lightly_studio as ls
+from lightly_studio import db_manager
+from lightly_studio.resolvers import caption_resolver
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
+# Create a DatasetLoader from a path
+dataset = ls.Dataset.create()
+dataset.add_samples_from_coco_caption(
+    annotations_json="/path/to/your/dataset/annotations.json",
+    images_path="/path/to/your/dataset",
+)
+# Display some details about the captions
+captions_result = caption_resolver.get_all(session=dataset.session, dataset_id=dataset.dataset_id)
+print(captions_result.total_count)
+for caption in captions_result.captions[:10]:
+    print(caption)
+ls.start_gui()

lightly_studio/examples/example_metadata.py CHANGED Viewed

@@ -83,7 +83,7 @@ def add_bulk_metadata(session: Session, sample_ids: list[UUID]) -> None:
     # Bulk insert metadata
     start_time = time.time()
-    metadata_resolver.bulk_set_metadata(session, sample_metadata)
+    metadata_resolver.bulk_update_metadata(session, sample_metadata)
     elapsed_time = time.time() - start_time
     print(f"✅ Added metadata to {len(sample_ids)} samples in {elapsed_time:.2f}s")
@@ -293,6 +293,9 @@ def demonstrate_dictionary_like_access(samples: list[Sample]) -> None:
 def main() -> None:
     """Main function to demonstrate  metadata functionality."""
     try:
+        # Cleanup an existing database
+        db_manager.connect(cleanup_existing=True)
         # Load existing dataset
         dataset, samples = load_existing_dataset()

lightly_studio/examples/example_selection.py CHANGED Viewed

@@ -5,11 +5,15 @@ from pathlib import Path
 from environs import Env
 import lightly_studio as ls
+from lightly_studio import db_manager
 # Read environment variables
 env = Env()
 env.read_env()
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
 # Define the path to the dataset directory
 dataset_path = Path(env.path("DATASET_PATH", "/path/to/your/dataset"))
 dataset_path = dataset_path.parent if dataset_path.is_file() else dataset_path

lightly_studio/examples/example_split_work.py CHANGED Viewed

@@ -5,11 +5,15 @@ import math
 from environs import Env
 import lightly_studio as ls
+from lightly_studio import db_manager
 # Read environment variables
 env = Env()
 env.read_env()
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
 # Create a Dataset instance
 dataset = ls.Dataset.create()

lightly_studio/examples/example_yolo.py CHANGED Viewed

@@ -5,11 +5,15 @@ from pathlib import Path
 from environs import Env
 import lightly_studio as ls
+from lightly_studio import db_manager
 # Read environment variables
 env = Env()
 env.read_env()
+# Cleanup an existing database
+db_manager.connect(cleanup_existing=True)
 # Define the path to the dataset directory
 dataset_path = Path(env.path("DATASET_PATH", "/path/to/your/dataset"))

lightly_studio/export/export_dataset.py ADDED Viewed

@@ -0,0 +1,73 @@
+"""Exports datasets from Lightly Studio into various formats."""
+from __future__ import annotations
+from pathlib import Path
+from typing import Iterable
+from labelformat.formats import COCOObjectDetectionOutput
+from sqlmodel import Session
+from lightly_studio.core.sample import Sample
+from lightly_studio.export.lightly_studio_label_input import LightlyStudioObjectDetectionInput
+from lightly_studio.type_definitions import PathLike
+DEFAULT_EXPORT_FILENAME = "coco_export.json"
+class DatasetExport:
+    """Provides methods to export a dataset or a subset of it.
+    This class is typically not instantiated directly but returned by `Dataset.export()`.
+    It allows exporting data in various formats.
+    """
+    def __init__(self, session: Session, samples: Iterable[Sample]):
+        """Initializes the DatasetExport object.
+        Args:
+            session: The database session.
+            samples: Samples to export.
+        """
+        self.session = session
+        self.samples = samples
+    def to_coco_object_detections(self, output_json: PathLike | None = None) -> None:
+        """Exports object detection annotations to a COCO format JSON file.
+        Args:
+            output_json: The path to the output COCO JSON file. If not provided,
+                defaults to "coco_export.json" in the current working directory.
+        Raises:
+            ValueError: If the annotation task with the given name does not exist.
+        """
+        if output_json is None:
+            output_json = DEFAULT_EXPORT_FILENAME
+        to_coco_object_detections(
+            session=self.session,
+            samples=self.samples,
+            output_json=Path(output_json),
+        )
+def to_coco_object_detections(
+    session: Session,
+    samples: Iterable[Sample],
+    output_json: Path,
+) -> None:
+    """Exports object detection annotations to a COCO format JSON file.
+    This function is for internal use. Use `Dataset.query().export().to_coco_object_detections()`
+    instead.
+    Args:
+        session: The database session.
+        samples: The samples to export.
+        output_json: The path to save the output JSON file.
+    """
+    export_input = LightlyStudioObjectDetectionInput(
+        session=session,
+        samples=samples,
+    )
+    COCOObjectDetectionOutput(output_file=output_json).save(label_input=export_input)

lightly_studio/export/lightly_studio_label_input.py ADDED Viewed

@@ -0,0 +1,120 @@
+"""Converts annotations from Lightly Studio to Labelformat format."""
+from __future__ import annotations
+from argparse import ArgumentParser
+from typing import Iterable
+from uuid import UUID
+from labelformat.model.bounding_box import BoundingBox
+from labelformat.model.category import Category
+from labelformat.model.image import Image
+from labelformat.model.object_detection import (
+    ImageObjectDetection,
+    ObjectDetectionInput,
+    SingleObjectDetection,
+)
+from sqlmodel import Session
+from lightly_studio.core.sample import Sample
+from lightly_studio.models.annotation.annotation_base import AnnotationBaseTable, AnnotationType
+from lightly_studio.resolvers import annotation_label_resolver
+class LightlyStudioObjectDetectionInput(ObjectDetectionInput):
+    """Labelformat adapter backed by dataset samples and annotations."""
+    def __init__(self, session: Session, samples: Iterable[Sample]) -> None:
+        """Initializes the LightlyStudioObjectDetectionInput.
+        Args:
+            session: The SQLModel session to use for database access. Used only in the
+                constructor to fetch the labels for the given annotation task.
+            samples: Dataset samples.
+        """
+        self._samples = list(samples)
+        self._label_id_to_category = _build_label_id_to_category(session=session)
+    @staticmethod
+    def add_cli_arguments(parser: ArgumentParser) -> None:
+        """Adds CLI arguments."""
+        # Add CLI arguments implementation is not needed for this class. We need it only
+        # to satisfy the interface.
+        raise NotImplementedError()
+    def get_categories(self) -> Iterable[Category]:
+        """Returns the categories for export."""
+        return self._label_id_to_category.values()
+    def get_images(self) -> Iterable[Image]:
+        """Returns the images for export."""
+        for idx, sample in enumerate(self._samples):
+            yield _sample_to_image(sample=sample, image_id=idx)
+    def get_labels(self) -> Iterable[ImageObjectDetection]:
+        """Returns the labels for export."""
+        for idx, sample in enumerate(self._samples):
+            yield _sample_to_image_obj_det(
+                sample=sample,
+                image_id=idx,
+                label_id_to_category=self._label_id_to_category,
+            )
+def _build_label_id_to_category(session: Session) -> dict[UUID, Category]:
+    labels = annotation_label_resolver.get_all_sorted_alphabetically(
+        session=session,
+    )
+    # TODO(Horatiu, 09/2025): We should get only labels that are attached to Object Detection
+    # annotations.
+    return {
+        label.annotation_label_id: Category(id=idx, name=label.annotation_label_name)
+        for idx, label in enumerate(labels)
+    }
+def _sample_to_image(sample: Sample, image_id: int) -> Image:
+    return Image(
+        id=image_id,
+        filename=sample.file_path_abs,
+        width=sample.width,
+        height=sample.height,
+    )
+def _sample_to_image_obj_det(
+    sample: Sample,
+    image_id: int,
+    label_id_to_category: dict[UUID, Category],
+) -> ImageObjectDetection:
+    # TODO(Michal, 09/2025): We can optimise in the future to filter annotations in a DB query.
+    objects = [
+        _annotation_to_single_obj_det(
+            annotation=annotation,
+            label_id_to_category=label_id_to_category,
+        )
+        for annotation in sample.inner.annotations
+        if annotation.annotation_type == AnnotationType.OBJECT_DETECTION
+    ]
+    return ImageObjectDetection(
+        image=_sample_to_image(sample=sample, image_id=image_id),
+        objects=objects,
+    )
+def _annotation_to_single_obj_det(
+    annotation: AnnotationBaseTable, label_id_to_category: dict[UUID, Category]
+) -> SingleObjectDetection:
+    assert annotation.object_detection_details is not None
+    box = BoundingBox(
+        xmin=annotation.object_detection_details.x,
+        ymin=annotation.object_detection_details.y,
+        xmax=annotation.object_detection_details.x + annotation.object_detection_details.width,
+        ymax=annotation.object_detection_details.y + annotation.object_detection_details.height,
+    )
+    category = label_id_to_category[annotation.annotation_label.annotation_label_id]
+    return SingleObjectDetection(
+        category=category,
+        box=box,
+        confidence=annotation.confidence,
+    )

lightly_studio/few_shot_classifier/classifier_manager.py CHANGED Viewed

@@ -22,20 +22,16 @@ from lightly_studio.few_shot_classifier.random_forest_classifier import (
 )
 from lightly_studio.models.annotation.annotation_base import (
     AnnotationCreate,
+    AnnotationType,
 )
 from lightly_studio.models.annotation_label import (
     AnnotationLabelCreate,
 )
-from lightly_studio.models.annotation_task import (
-    AnnotationTaskTable,
-    AnnotationType,
-)
 from lightly_studio.models.classifier import EmbeddingClassifier
 from lightly_studio.models.sample import SampleTable
 from lightly_studio.resolvers import (
     annotation_label_resolver,
     annotation_resolver,
-    annotation_task_resolver,
     embedding_model_resolver,
     sample_embedding_resolver,
     sample_resolver,
@@ -92,7 +88,6 @@ class ClassifierEntry:
     # will be false.
     is_active: bool = False
-    annotation_task_id: UUID | None = None
     annotation_label_ids: list[UUID] | None = None
@@ -469,16 +464,14 @@ class ClassifierManager:
         embeddings = [se.embedding for se in sample_embeddings]
         predictions = classifier.few_shot_classifier.predict(embeddings)
         if len(predictions):
-            _create_annotation_task_and_labels_for_classifier(
+            _create_annotation_labels_for_classifier(
                 classifier=classifier,
                 session=session,
                 dataset_id=dataset_id,
             )
         else:
             raise ValueError(f"Predict returned empty list for classifier:'{classifier_id}'")
-        # Check if annotation task or labels are available
-        if not classifier.annotation_task_id:
-            raise ValueError(f"Classifier with ID '{classifier_id}' has no annotation task.")
+        # Check if annotation labels are available
         if not classifier.annotation_label_ids:
             raise ValueError(f"Classifier with ID '{classifier_id}' has no annotation labels")
@@ -490,7 +483,6 @@ class ClassifierManager:
             classification_annotations.append(
                 AnnotationCreate(
                     sample_id=sample_embedding.sample_id,
-                    annotation_task_id=classifier.annotation_task_id,
                     dataset_id=dataset_id,
                     annotation_label_id=classifier.annotation_label_ids[max_index],
                     annotation_type=AnnotationType.CLASSIFICATION,
@@ -500,7 +492,6 @@ class ClassifierManager:
         #  Clear previous annotations by this classifier
         annotation_resolver.delete_annotations(
             session=session,
-            annotation_task_ids=[classifier.annotation_task_id],
             annotation_label_ids=classifier.annotation_label_ids,
         )
         annotation_resolver.create_many(session=session, annotations=classification_annotations)
@@ -597,30 +588,18 @@ class ClassifierManager:
         return self._classifiers[classifier_id]
-def _create_annotation_task_and_labels_for_classifier(
+def _create_annotation_labels_for_classifier(
     session: Session,
     dataset_id: UUID,
     classifier: ClassifierEntry,
 ) -> None:
-    """Create annotation task and labels for the classifier.
+    """Create annotation labels for the classifier.
     Args:
         session: Database session.
         dataset_id: The dataset ID to which the samples belong.
         classifier: The classifier object to update.
     """
-    # Check if the annotation task exists and if not create it.
-    if classifier.annotation_task_id is None:
-        annotation_task = annotation_task_resolver.create(
-            session=session,
-            annotation_task=AnnotationTaskTable(
-                name=FSC_ANNOTATION_TASK_PREFIX + classifier.few_shot_classifier.name,
-                annotation_type=AnnotationType.CLASSIFICATION,
-                is_prediction=True,
-            ),
-        )
-        classifier.annotation_task_id = annotation_task.annotation_task_id
     # Check if the annotation label with the classifier name and class
     # names exists and if not create it.
     if classifier.annotation_label_ids is None:

lightly-studio 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl

Potentially problematic release.

lightly-studio 0.3.2py3-none-any.whl → 0.3.4py3-none-any.whl