dtlpy 1.115.44__py3-none-any.whl → 1.116.6__py3-none-any.whl

dtlpy/examples/add_labels.py

@@ -1,135 +1,135 @@
-def main(project_name, dataset_name):
-    # Imports the SDK package
-    import dtlpy as dl
-
-    """
-    Label dictionary format:
-    
-    {
-        'label_name': 'dog',
-        'displayLabel': 'Dog',
-        'attributes': ['list of attributes'],
-        'color': (34, 6, 231),
-        'children': ['list of label dictionaries']
-    }
-
-    """
-
-    # prep
-    project = dl.projects.get(project_name=project_name)
-    dataset = project.datasets.get(dataset_name=dataset_name)
-
-    #########################
-    # View dataset's labels #
-    #########################
-    # as objects
-    labels = dataset.labels
-
-    # as instance map
-    labels = dataset.instance_map
-
-    ###############################
-    # add label to dataset entity #
-    ###############################
-    dataset.add_label(label_name='Horse', color=(2, 43, 123))
-
-    #############################
-    # add label with sub-labels #
-    #############################
-    dataset.add_label(label_name='CEO', color=(2, 43, 123),
-                      children=[{'label_name': 'Manager',
-                                 'children': [{'label_name': 'Employee'}]}])
-
-    ################################
-    # add labels to dataset entity #
-    ################################
-    labels = [
-        {'label_name': 'Dog',
-         'color': (34, 6, 231),
-         'children': [{'label_name': 'Puppy',
-                       'color': (24, 16, 130)}]},
-        {'label_name': 'Cat',
-         'color': (24, 25, 31),
-         'children': [{'label_name': 'Kitten',
-                       'color': (124, 116, 140)}]}
-    ]
-    dataset.add_labels(label_list=labels)
-
-    #################
-    # Delete Labels #
-    #################
-    dataset.delete_labels(label_names=['Cat', 'Dog'])
-
-    ########################################
-    # Copy dataset labels to a new dataset #
-    ########################################
-    new_dataset = project.datasets.create(dataset_name='new_dataset_with_labels',
-                                          labels=dataset.labels)
-
-    ##########################################
-    # Copy dataset ontology to a new dataset #
-    ##########################################
-    new_dataset = project.datasets.create(dataset_name='new_dataset_with_ontology',
-                                          ontology_ids=dataset.ontology_ids)
-
-    ##############################################
-    # Copy dataset labels to an existing dataset #
-    ##############################################
-    new_dataset = project.datasets.create(dataset_name='new_dataset_without_labels')
-
-    # Get from a list or recipes
-    recipe = new_dataset.recipes.list()[0]
-    # Or get recipe by id
-    # recipe = new_dataset.recipes.get(recipe_id='recipe_id')
-
-    # Get from the list of ontologies
-    ontology = recipe.ontologies.list()[0]
-    # Or get ontology by id
-    # ontology = recipe.ontologies.get(ontology='ontology_id')
-
-    # Add the labels to the dataset
-    ontology.add_labels(label_list=dataset.labels)
-    ontology.update()
-
-    #########################################################
-    # Copy dataset ontology to an existing dataset's recipe #
-    #########################################################
-    new_dataset = project.datasets.create(dataset_name='new_dataset_without_ontology')
-    # get recipe
-    new_dataset_recipe = new_dataset.recipes.list()[0]
-    # Copy from a different dataset
-    new_dataset_recipe.ontology_ids = dataset.ontology_ids
-    # Update the new dataset
-    new_dataset_recipe.update()
-
-    ##########################
-    # update existing recipe #
-    ##########################
-    # Get recipe from list
-    recipe = dataset.recipes.list()[0]
-    # Or get specific recipe:
-    # recipe = dataset.recipes.get(recipe_id='id')
-
-    # Get ontology from list
-    ontology = recipe.ontologies.list()[0]
-    # Or get specific ontology:
-    # ontology = recipe.ontologies.get(ontology_id='id')
-
-    # Add one label
-    ontology.add_label(label_name='Lion', color=(35, 234, 123))
-
-    # Add a list of labels
-    labels = [{'label_name': 'Shark', 'color': (1, 1, 1)}, {'label_name': 'Whale', 'color': (34, 56, 7)},
-              {'label_name': 'Dolphin', 'color': (100, 14, 150)}]
-    ontology.add_labels(label_list=labels)
-
-    # After adding - update ontology
-    ontology.update()
-
-    #####################
-    # Create new recipe #
-    #####################
-    # Label list
-    labels = [{'tag': 'Donkey', 'color': (1, 1, 1)}, {'tag': 'Mammoth', 'color': (34, 56, 7)},
-              {'tag': 'Bird', 'color': (100, 14, 150)}]
-    recipe = dataset.recipes.create(recipe_name='My Recipe', labels=labels)
+def main(project_name, dataset_name):
+    # Imports the SDK package
+    import dtlpy as dl
+
+    """
+    Label dictionary format:
+    
+    {
+        'label_name': 'dog',
+        'displayLabel': 'Dog',
+        'attributes': ['list of attributes'],
+        'color': (34, 6, 231),
+        'children': ['list of label dictionaries']
+    }
+
+    """
+
+    # prep
+    project = dl.projects.get(project_name=project_name)
+    dataset = project.datasets.get(dataset_name=dataset_name)
+
+    #########################
+    # View dataset's labels #
+    #########################
+    # as objects
+    labels = dataset.labels
+
+    # as instance map
+    labels = dataset.instance_map
+
+    ###############################
+    # add label to dataset entity #
+    ###############################
+    dataset.add_label(label_name='Horse', color=(2, 43, 123))
+
+    #############################
+    # add label with sub-labels #
+    #############################
+    dataset.add_label(label_name='CEO', color=(2, 43, 123),
+                      children=[{'label_name': 'Manager',
+                                 'children': [{'label_name': 'Employee'}]}])
+
+    ################################
+    # add labels to dataset entity #
+    ################################
+    labels = [
+        {'label_name': 'Dog',
+         'color': (34, 6, 231),
+         'children': [{'label_name': 'Puppy',
+                       'color': (24, 16, 130)}]},
+        {'label_name': 'Cat',
+         'color': (24, 25, 31),
+         'children': [{'label_name': 'Kitten',
+                       'color': (124, 116, 140)}]}
+    ]
+    dataset.add_labels(label_list=labels)
+
+    #################
+    # Delete Labels #
+    #################
+    dataset.delete_labels(label_names=['Cat', 'Dog'])
+
+    ########################################
+    # Copy dataset labels to a new dataset #
+    ########################################
+    new_dataset = project.datasets.create(dataset_name='new_dataset_with_labels',
+                                          labels=dataset.labels)
+
+    ##########################################
+    # Copy dataset ontology to a new dataset #
+    ##########################################
+    new_dataset = project.datasets.create(dataset_name='new_dataset_with_ontology',
+                                          ontology_ids=dataset.ontology_ids)
+
+    ##############################################
+    # Copy dataset labels to an existing dataset #
+    ##############################################
+    new_dataset = project.datasets.create(dataset_name='new_dataset_without_labels')
+
+    # Get from a list or recipes
+    recipe = new_dataset.recipes.list()[0]
+    # Or get recipe by id
+    # recipe = new_dataset.recipes.get(recipe_id='recipe_id')
+
+    # Get from the list of ontologies
+    ontology = recipe.ontologies.list()[0]
+    # Or get ontology by id
+    # ontology = recipe.ontologies.get(ontology='ontology_id')
+
+    # Add the labels to the dataset
+    ontology.add_labels(label_list=dataset.labels)
+    ontology.update()
+
+    #########################################################
+    # Copy dataset ontology to an existing dataset's recipe #
+    #########################################################
+    new_dataset = project.datasets.create(dataset_name='new_dataset_without_ontology')
+    # get recipe
+    new_dataset_recipe = new_dataset.recipes.list()[0]
+    # Copy from a different dataset
+    new_dataset_recipe.ontology_ids = dataset.ontology_ids
+    # Update the new dataset
+    new_dataset_recipe.update()
+
+    ##########################
+    # update existing recipe #
+    ##########################
+    # Get recipe from list
+    recipe = dataset.recipes.list()[0]
+    # Or get specific recipe:
+    # recipe = dataset.recipes.get(recipe_id='id')
+
+    # Get ontology from list
+    ontology = recipe.ontologies.list()[0]
+    # Or get specific ontology:
+    # ontology = recipe.ontologies.get(ontology_id='id')
+
+    # Add one label
+    ontology.add_label(label_name='Lion', color=(35, 234, 123))
+
+    # Add a list of labels
+    labels = [{'label_name': 'Shark', 'color': (1, 1, 1)}, {'label_name': 'Whale', 'color': (34, 56, 7)},
+              {'label_name': 'Dolphin', 'color': (100, 14, 150)}]
+    ontology.add_labels(label_list=labels)
+
+    # After adding - update ontology
+    ontology.update()
+
+    #####################
+    # Create new recipe #
+    #####################
+    # Label list
+    labels = [{'tag': 'Donkey', 'color': (1, 1, 1)}, {'tag': 'Mammoth', 'color': (34, 56, 7)},
+              {'tag': 'Bird', 'color': (100, 14, 150)}]
+    recipe = dataset.recipes.create(recipe_name='My Recipe', labels=labels)

dtlpy/examples/add_metadata_to_item.py

@@ -1,21 +1,21 @@
-def main(project_name, dataset_name, item_path):
-    """
-    Add any metadata to item
-    :return:
-    """
-    # import Dataloop SDK
-    import dtlpy as dl
-
-    # get dataset
-    dataset = dl.projects.get(project_name=project_name).datasets.get(dataset_name=dataset_name)
-
-    # upload and claim item
-    item = dataset.items.upload(local_path=item_path)
-
-    # modify metadata
-    item.metadata['user'] = dict()
-    item.metadata['user']['MyKey'] = 'MyVal'
-    # update and reclaim item
-    item = item.update()
-
-    # item in platform should have section 'user' in metadata with field 'MyKey' and value 'MyVal'
+def main(project_name, dataset_name, item_path):
+    """
+    Add any metadata to item
+    :return:
+    """
+    # import Dataloop SDK
+    import dtlpy as dl
+
+    # get dataset
+    dataset = dl.projects.get(project_name=project_name).datasets.get(dataset_name=dataset_name)
+
+    # upload and claim item
+    item = dataset.items.upload(local_path=item_path)
+
+    # modify metadata
+    item.metadata['user'] = dict()
+    item.metadata['user']['MyKey'] = 'MyVal'
+    # update and reclaim item
+    item = item.update()
+
+    # item in platform should have section 'user' in metadata with field 'MyKey' and value 'MyVal'

dtlpy/examples/annotate_items_using_model.py

@@ -1,65 +1,65 @@
-def main():
-    """
-        Annotate a batch of images using a model and upload to platform
-        :return:
-        """
-    import numpy as np
-    from PIL import Image
-    from keras.applications.imagenet_utils import decode_predictions
-    from keras.applications.inception_v3 import InceptionV3, preprocess_input
-    import dtlpy as dl
-
-    ##############
-    # load model #
-    ##############
-    model = InceptionV3()
-
-    ##########################
-    # init platform instance #
-    ##########################
-    project = dl.projects.get(project_name='ImageNet')
-    dataset = project.datasets.get(dataset_name='sample')
-
-    # get pages of images from dataset
-    pages = dataset.items.list()
-    ####################
-    # start annotating #
-    ####################
-    for page in pages:
-        for item in page:
-            if item.type == 'dir':
-                continue
-            img_batch = [item.download(save_locally=False)]
-            # load images
-            img_batch = [Image.open(buf) for buf in img_batch]
-            # get original images shapes before reshaping for model
-            orig_img_shape = [img.size[::-1] for img in img_batch]
-            # reshape and load images
-            batch = np.array([np.array(img.resize((299, 299))) for img in img_batch])
-            # preprocess batch
-            batch = preprocess_input(batch)
-            # inference the model
-            predictions = model.predict(batch)
-            # get ImageNet labels
-            labels = decode_predictions(predictions, top=1)
-            # create platform annotations instance
-            builder = item.annotations.builder()
-            for i_pred, label in enumerate(labels):
-
-                # add the class labels
-                ##############################
-                # If model is classification #
-                ##############################
-                builder.add(annotation_definition=dl.Classification(label=label[0][1]))
-                #############################
-                # If model outputs polygons #
-                #############################
-                builder.add(annotation_definition=dl.Polyline(geo=pred['polygon_pts'],
-                                                              label=labels[i_pred][0][1]))
-                #########################
-                # If model outputs mask #
-                #########################
-                builder.add(annotation_definition=dl.Segmentation(geo=pred['mask'],
-                                                                  label=labels[i_pred][0][1]))
-            # upload a annotations to matching items in platform
-            builder.upload()
+def main():
+    """
+        Annotate a batch of images using a model and upload to platform
+        :return:
+        """
+    import numpy as np
+    from PIL import Image
+    from keras.applications.imagenet_utils import decode_predictions
+    from keras.applications.inception_v3 import InceptionV3, preprocess_input
+    import dtlpy as dl
+
+    ##############
+    # load model #
+    ##############
+    model = InceptionV3()
+
+    ##########################
+    # init platform instance #
+    ##########################
+    project = dl.projects.get(project_name='ImageNet')
+    dataset = project.datasets.get(dataset_name='sample')
+
+    # get pages of images from dataset
+    pages = dataset.items.list()
+    ####################
+    # start annotating #
+    ####################
+    for page in pages:
+        for item in page:
+            if item.type == 'dir':
+                continue
+            img_batch = [item.download(save_locally=False)]
+            # load images
+            img_batch = [Image.open(buf) for buf in img_batch]
+            # get original images shapes before reshaping for model
+            orig_img_shape = [img.size[::-1] for img in img_batch]
+            # reshape and load images
+            batch = np.array([np.array(img.resize((299, 299))) for img in img_batch])
+            # preprocess batch
+            batch = preprocess_input(batch)
+            # inference the model
+            predictions = model.predict(batch)
+            # get ImageNet labels
+            labels = decode_predictions(predictions, top=1)
+            # create platform annotations instance
+            builder = item.annotations.builder()
+            for i_pred, label in enumerate(labels):
+
+                # add the class labels
+                ##############################
+                # If model is classification #
+                ##############################
+                builder.add(annotation_definition=dl.Classification(label=label[0][1]))
+                #############################
+                # If model outputs polygons #
+                #############################
+                builder.add(annotation_definition=dl.Polyline(geo=pred['polygon_pts'],
+                                                              label=labels[i_pred][0][1]))
+                #########################
+                # If model outputs mask #
+                #########################
+                builder.add(annotation_definition=dl.Segmentation(geo=pred['mask'],
+                                                                  label=labels[i_pred][0][1]))
+            # upload a annotations to matching items in platform
+            builder.upload()

dtlpy/examples/annotate_video_using_model_and_tracker.py

@@ -1,75 +1,75 @@
-def main():
-    """
-    Detect and track (using model and some tracker) and upload annotation to platform
-    :return:
-    """
-    import cv2
-    import dtlpy as dl
-
-    ##########################
-    # Load model and tracker #
-    ##########################
-    # load your model for detection
-    model = load_some_model()
-    # load any tracking algorithm to track detected elements
-    tracker = load_some_tracker()
-
-    ##############
-    # load video #
-    ##############
-    video_path = 'some/video/path'
-
-    vid = cv2.VideoCapture(video_path)
-    if not vid.isOpened():
-        raise IOError("Couldn't open webcam or video")
-    video_fps = vid.get(cv2.CAP_PROP_FPS)
-    video_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
-                  int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))
-    video_frames = int(vid.get(cv2.CAP_PROP_FRAME_COUNT))
-
-    ############
-    # Platform #
-    ###########
-    # get the item from platform
-    item = dl.projects.get(project_name='MyProject') \
-        .datasets.get(dataset_name='MyDataset') \
-        .items.get(filepath='/path/to/video.mp4')
-    builder = item.annotations.builder()
-
-    #######
-    # Run #
-    #######
-    frame_num = 0
-    while True:
-        # get new frame from video
-        return_value, frame = vid.read()
-        if not return_value:
-            break
-
-        # get detection
-        detections = model.predict(frame)
-
-        # update tracker
-        tracked_elements = tracker.update(detections, frame)
-
-        # update annotations object
-        for element in tracked_elements:
-            # element.bb - format of the bounding box is 2 points in 1 array - [x_left, y_top, x_right, y_bottom])
-            # tracking id of each element is in element.id. to keep the ids of the detected elements
-            left, top, bottom, right = element.bb  # points bounding box annotation
-            builder.add(annotation_definition=dl.Box(top=top,
-                                                     left=left,
-                                                     right=right,
-                                                     bottom=bottom,
-                                                     label=element.label),
-                        object_id=element.id,
-                        frame_num=frame_num)
-        # increase frame number
-        frame_num += 1
-        if cv2.waitKey(1) & 0xFF == ord('q'):
-            break
-
-    ##################################
-    # Upload annotations to platform #
-    ##################################
-    item.annotations.upload(builder.to_platform())
+def main():
+    """
+    Detect and track (using model and some tracker) and upload annotation to platform
+    :return:
+    """
+    import cv2
+    import dtlpy as dl
+
+    ##########################
+    # Load model and tracker #
+    ##########################
+    # load your model for detection
+    model = load_some_model()
+    # load any tracking algorithm to track detected elements
+    tracker = load_some_tracker()
+
+    ##############
+    # load video #
+    ##############
+    video_path = 'some/video/path'
+
+    vid = cv2.VideoCapture(video_path)
+    if not vid.isOpened():
+        raise IOError("Couldn't open webcam or video")
+    video_fps = vid.get(cv2.CAP_PROP_FPS)
+    video_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
+                  int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))
+    video_frames = int(vid.get(cv2.CAP_PROP_FRAME_COUNT))
+
+    ############
+    # Platform #
+    ###########
+    # get the item from platform
+    item = dl.projects.get(project_name='MyProject') \
+        .datasets.get(dataset_name='MyDataset') \
+        .items.get(filepath='/path/to/video.mp4')
+    builder = item.annotations.builder()
+
+    #######
+    # Run #
+    #######
+    frame_num = 0
+    while True:
+        # get new frame from video
+        return_value, frame = vid.read()
+        if not return_value:
+            break
+
+        # get detection
+        detections = model.predict(frame)
+
+        # update tracker
+        tracked_elements = tracker.update(detections, frame)
+
+        # update annotations object
+        for element in tracked_elements:
+            # element.bb - format of the bounding box is 2 points in 1 array - [x_left, y_top, x_right, y_bottom])
+            # tracking id of each element is in element.id. to keep the ids of the detected elements
+            left, top, bottom, right = element.bb  # points bounding box annotation
+            builder.add(annotation_definition=dl.Box(top=top,
+                                                     left=left,
+                                                     right=right,
+                                                     bottom=bottom,
+                                                     label=element.label),
+                        object_id=element.id,
+                        frame_num=frame_num)
+        # increase frame number
+        frame_num += 1
+        if cv2.waitKey(1) & 0xFF == ord('q'):
+            break
+
+    ##################################
+    # Upload annotations to platform #
+    ##################################
+    item.annotations.upload(builder.to_platform())

dtlpy/examples/annotations_convert_to_voc.py

@@ -1,9 +1,9 @@
-def main():
-    import dtlpy as dl
-
-    project = dl.projects.get(project_name='Ocean')
-    dataset = project.datasets.get(dataset_name='Sharks')
-
-    converter = dl.Converter()
-    converter.convert_dataset(dataset=dataset, to_format='voc',
-                              local_path='home/voc_annotations/sharks')
+def main():
+    import dtlpy as dl
+
+    project = dl.projects.get(project_name='Ocean')
+    dataset = project.datasets.get(dataset_name='Sharks')
+
+    converter = dl.Converter()
+    converter.convert_dataset(dataset=dataset, to_format='voc',
+                              local_path='home/voc_annotations/sharks')

dtlpy/examples/annotations_convert_to_yolo.py

@@ -1,9 +1,9 @@
-def main():
-    import dtlpy as dl
-    
-    project = dl.projects.get(project_name='Jungle')
-    dataset = project.datasets.get(dataset_name='Tigers')
-
-    converter = dl.Converter()
-    converter.convert_dataset(dataset=dataset, to_format='yolo',
-                              local_path='home/yolo_annotations/tigers')
+def main():
+    import dtlpy as dl
+    
+    project = dl.projects.get(project_name='Jungle')
+    dataset = project.datasets.get(dataset_name='Tigers')
+
+    converter = dl.Converter()
+    converter.convert_dataset(dataset=dataset, to_format='yolo',
+                              local_path='home/yolo_annotations/tigers')