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

dtlpy/examples/convert_annotation_types.py

@@ -1,51 +1,51 @@
-def main():
-    """
-    Convert annotation types
-    :return:
-    """
-    from PIL import Image
-    import matplotlib.pyplot as plt
-    import dtlpy as dl
-
-    # Get project and dataset
-    project = dl.projects.get(project_name='Toilet Paper')
-    dataset = project.datasets.get(dataset_name='Extra Soft')
-
-    # Get item and binaries
-    item = dataset.items.get(filepath='/with_puppies.jpg')
-    buffer = item.download(save_locally=False)
-
-    #######################################
-    # Convert mask annotations to polygon #
-    #######################################
-    img = Image.open(buffer)
-    builder = item.annotations.builder()
-    builder.add(dl.Polygon.from_segmentation(mask=mask,  # binary mask of the annotation
-                                             label='roll'))
-    # plot the annotation
-    plt.figure()
-    plt.imshow(builder.show())
-    # plot annotation on the image
-    plt.figure()
-    plt.imshow(builder.show(image=img))
-
-    # upload annotation to platform
-    item.annotations.update(builder)
-
-    #######################################
-    # Convert polygon annotations to mask #
-    #######################################
-    img = Image.open(buffer)
-    builder = item.annotations.builder()
-    builder.add(dl.Segmentation.from_polygon(geo=[[x1, y1], [x2, y2], [x3, y3]],  # list of coordinates
-                                             shape=img.size[::-1],  # (h,w)
-                                             label='roll'))
-    # plot the annotation
-    plt.figure()
-    plt.imshow(builder.show())
-    # plot annotation on the image
-    plt.figure()
-    plt.imshow(builder.show(image=img))
-
-    # upload annotation to platform
-    item.annotations.update(builder)
+def main():
+    """
+    Convert annotation types
+    :return:
+    """
+    from PIL import Image
+    import matplotlib.pyplot as plt
+    import dtlpy as dl
+
+    # Get project and dataset
+    project = dl.projects.get(project_name='Toilet Paper')
+    dataset = project.datasets.get(dataset_name='Extra Soft')
+
+    # Get item and binaries
+    item = dataset.items.get(filepath='/with_puppies.jpg')
+    buffer = item.download(save_locally=False)
+
+    #######################################
+    # Convert mask annotations to polygon #
+    #######################################
+    img = Image.open(buffer)
+    builder = item.annotations.builder()
+    builder.add(dl.Polygon.from_segmentation(mask=mask,  # binary mask of the annotation
+                                             label='roll'))
+    # plot the annotation
+    plt.figure()
+    plt.imshow(builder.show())
+    # plot annotation on the image
+    plt.figure()
+    plt.imshow(builder.show(image=img))
+
+    # upload annotation to platform
+    item.annotations.update(builder)
+
+    #######################################
+    # Convert polygon annotations to mask #
+    #######################################
+    img = Image.open(buffer)
+    builder = item.annotations.builder()
+    builder.add(dl.Segmentation.from_polygon(geo=[[x1, y1], [x2, y2], [x3, y3]],  # list of coordinates
+                                             shape=img.size[::-1],  # (h,w)
+                                             label='roll'))
+    # plot the annotation
+    plt.figure()
+    plt.imshow(builder.show())
+    # plot annotation on the image
+    plt.figure()
+    plt.imshow(builder.show(image=img))
+
+    # upload annotation to platform
+    item.annotations.update(builder)

dtlpy/examples/converter.py

@@ -1,143 +1,143 @@
-def main():
-    import dtlpy as dl
-
-    dataset = dl.projects.get("project_name").datasets.get("dataset_name")
-
-    ############################
-    # convert annotations list #
-    ############################
-
-    # convert from dataloop to other formats #
-    ##########################################
-
-    # known format #
-
-    item = dataset.items.get(item_id="item_id")
-    annotations = item.annotations.list()
-    converter = dl.Converter()
-    converted_annotations = converter.convert(annotations=annotations,
-                                              from_format='dataloop',
-                                              to_format='yolo')
-
-    # custom format #
-
-    # custom conversion function
-    # converts 1 dataloop annotation to custom format annotation
-    # returns converted annotation
-    def my_converter(annotation):
-        """
-        :param annotation: dataloop Annotation object
-        :return: format of new annotation
-        """
-        ann = {'label': annotation.label, 'type': annotation.type}
-        return ann
-
-    converted_annotations = converter.convert(annotations=annotations,
-                                              from_format='dataloop',
-                                              to_format='my_format',
-                                              conversion_func=my_converter)
-
-    # convert from other formats to dataloop format #
-    #################################################
-
-    # known format #
-
-    # load yolo annotations from file
-    annotations = list()
-    with open('annotations_filepath.txt', 'r') as f:
-        line = f.readline()
-        while line:
-            annotations.append(line)
-            line = f.readline()
-
-    # create converter object
-    converter = dl.Converter()
-
-    # convert
-    converted_annotations = converter.convert(annotations=annotations,
-                                              from_format='yolo',
-                                              to_format='dataloop')
-
-    # custom format #
-
-    # load yolo annotations from file
-    annotations = list()
-    with open('annotations_filepath.txt', 'r') as f:
-        line = f.readline()
-        while line:
-            annotations.append(line)
-            line = f.readline()
-
-    # create converter object
-    converter = dl.Converter()
-
-    # custom conversion function
-    # converts 1 custom annotation to dataloop annotation
-    # dataloop annotation
-    def my_converter(annotation):
-        """
-        :param annotation: custom annotation format
-        :type annotation: dict
-        :return: dataloop Annotation object
-        """
-        annotations = dl.Annotation.new(annotation_definition=dl.Box(top=annotation['top'],
-                                                                     bottom=annotation['bottom'],
-                                                                     left=annotation['left'],
-                                                                     right=annotation['right'],
-                                                                     label=annotation['label']))
-
-        return annotation
-
-    converted_annotations = converter.convert(annotations=annotations,
-                                              from_format='my_format',
-                                              to_format='dataloop',
-                                              conversion_func=my_converter)
-
-    ###############################
-    # convert dataset annotations #
-    ###############################
-
-    # known format #
-    ################
-
-    converter = dl.Converter()
-
-    converter.save_to_format = 'json'
-    converter.convert_dataset(dataset=dataset,
-                              to_format='coco',
-                              local_path='some/local/path/to/download/converted/annotations')
-
-    # custom format #
-    #################
-
-    def my_converter(annotation):
-        """
-        :param annotation: dataloop Annotation object
-        :return: format of new annotation
-        """
-        ann = {'label': annotation.label, 'type': annotation.type}
-        return ann
-
-    converter = dl.Converter()
-
-    converter.save_to_format = 'json'
-    converter.convert_dataset(dataset=dataset,
-                              to_format='my_format',
-                              conversion_func=my_converter,
-                              local_path='some/local/path/to/download/converted/annotations')
-
-    ################
-    # save to file #
-    ################
-
-    item = dataset.items.get(item_id="item_id")
-    annotations = item.annotations.list()
-    converter = dl.Converter()
-    converter.convert(annotations=annotations,
-                      from_format='dataloop',
-                      to_format='yolo')
-
-    # what file format to save to
-    converter.save_to_format = '.txt'
-    # save
-    converter.save_to_file(save_to='local_path', to_format='yolo')
+def main():
+    import dtlpy as dl
+
+    dataset = dl.projects.get("project_name").datasets.get("dataset_name")
+
+    ############################
+    # convert annotations list #
+    ############################
+
+    # convert from dataloop to other formats #
+    ##########################################
+
+    # known format #
+
+    item = dataset.items.get(item_id="item_id")
+    annotations = item.annotations.list()
+    converter = dl.Converter()
+    converted_annotations = converter.convert(annotations=annotations,
+                                              from_format='dataloop',
+                                              to_format='yolo')
+
+    # custom format #
+
+    # custom conversion function
+    # converts 1 dataloop annotation to custom format annotation
+    # returns converted annotation
+    def my_converter(annotation):
+        """
+        :param annotation: dataloop Annotation object
+        :return: format of new annotation
+        """
+        ann = {'label': annotation.label, 'type': annotation.type}
+        return ann
+
+    converted_annotations = converter.convert(annotations=annotations,
+                                              from_format='dataloop',
+                                              to_format='my_format',
+                                              conversion_func=my_converter)
+
+    # convert from other formats to dataloop format #
+    #################################################
+
+    # known format #
+
+    # load yolo annotations from file
+    annotations = list()
+    with open('annotations_filepath.txt', 'r') as f:
+        line = f.readline()
+        while line:
+            annotations.append(line)
+            line = f.readline()
+
+    # create converter object
+    converter = dl.Converter()
+
+    # convert
+    converted_annotations = converter.convert(annotations=annotations,
+                                              from_format='yolo',
+                                              to_format='dataloop')
+
+    # custom format #
+
+    # load yolo annotations from file
+    annotations = list()
+    with open('annotations_filepath.txt', 'r') as f:
+        line = f.readline()
+        while line:
+            annotations.append(line)
+            line = f.readline()
+
+    # create converter object
+    converter = dl.Converter()
+
+    # custom conversion function
+    # converts 1 custom annotation to dataloop annotation
+    # dataloop annotation
+    def my_converter(annotation):
+        """
+        :param annotation: custom annotation format
+        :type annotation: dict
+        :return: dataloop Annotation object
+        """
+        annotations = dl.Annotation.new(annotation_definition=dl.Box(top=annotation['top'],
+                                                                     bottom=annotation['bottom'],
+                                                                     left=annotation['left'],
+                                                                     right=annotation['right'],
+                                                                     label=annotation['label']))
+
+        return annotation
+
+    converted_annotations = converter.convert(annotations=annotations,
+                                              from_format='my_format',
+                                              to_format='dataloop',
+                                              conversion_func=my_converter)
+
+    ###############################
+    # convert dataset annotations #
+    ###############################
+
+    # known format #
+    ################
+
+    converter = dl.Converter()
+
+    converter.save_to_format = 'json'
+    converter.convert_dataset(dataset=dataset,
+                              to_format='coco',
+                              local_path='some/local/path/to/download/converted/annotations')
+
+    # custom format #
+    #################
+
+    def my_converter(annotation):
+        """
+        :param annotation: dataloop Annotation object
+        :return: format of new annotation
+        """
+        ann = {'label': annotation.label, 'type': annotation.type}
+        return ann
+
+    converter = dl.Converter()
+
+    converter.save_to_format = 'json'
+    converter.convert_dataset(dataset=dataset,
+                              to_format='my_format',
+                              conversion_func=my_converter,
+                              local_path='some/local/path/to/download/converted/annotations')
+
+    ################
+    # save to file #
+    ################
+
+    item = dataset.items.get(item_id="item_id")
+    annotations = item.annotations.list()
+    converter = dl.Converter()
+    converter.convert(annotations=annotations,
+                      from_format='dataloop',
+                      to_format='yolo')
+
+    # what file format to save to
+    converter.save_to_format = '.txt'
+    # save
+    converter.save_to_file(save_to='local_path', to_format='yolo')

dtlpy/examples/copy_annotations.py

@@ -1,22 +1,22 @@
-def main(first_project_name, second_project_name, first_dataset_name, second_dataset_name, first_remote_filepath, second_remote_filepath):
-    """
-    Copy annotations between items
-    :return:
-    """
-    import dtlpy as dl
-
-    # FROM get the annotations from item
-    project = dl.projects.get(project_name=first_project_name)
-    dataset = project.datasets.get(dataset_name=first_dataset_name)
-    item = dataset.items.get(filepath=first_remote_filepath)
-
-    # get annotations
-    annotations = item.annotations.list()
-
-    # TO post annotations to other item
-    project = dl.projects.get(project_name=second_project_name)
-    dataset = project.datasets.get(dataset_name=second_dataset_name)
-    item = dataset.items.get(filepath=second_remote_filepath)
-
-    # post
-    item.annotations.upload(annotations=annotations)
+def main(first_project_name, second_project_name, first_dataset_name, second_dataset_name, first_remote_filepath, second_remote_filepath):
+    """
+    Copy annotations between items
+    :return:
+    """
+    import dtlpy as dl
+
+    # FROM get the annotations from item
+    project = dl.projects.get(project_name=first_project_name)
+    dataset = project.datasets.get(dataset_name=first_dataset_name)
+    item = dataset.items.get(filepath=first_remote_filepath)
+
+    # get annotations
+    annotations = item.annotations.list()
+
+    # TO post annotations to other item
+    project = dl.projects.get(project_name=second_project_name)
+    dataset = project.datasets.get(dataset_name=second_dataset_name)
+    item = dataset.items.get(filepath=second_remote_filepath)
+
+    # post
+    item.annotations.upload(annotations=annotations)

dtlpy/examples/copy_folder.py

@@ -1,31 +1,31 @@
-def main(first_project_name, second_project_name, first_dataset_name, second_dataset_name):
-    """
-    Copy folder from project/dataset/folder
-    :return:
-    """
-    import dtlpy as dl
-
-    # Get source project and dataset
-    project = dl.projects.get(project_name=first_project_name)
-    dataset_from = project.datasets.get(dataset_name=first_dataset_name)
-    # filter to get all files of a specific folder
-    filters = dl.Filters()
-    filters.add(field='type', values='file')  # get only files
-    filters.add(field='filename', values='/source_folder/**')  # get all items in folder (recursive)
-    pages = dataset_from.items.list(filters=filters)
-
-    # Get destination project and annotations
-    project = dl.projects.get(project_name=second_project_name)
-    dataset_to = project.datasets.get(dataset_name=second_dataset_name)
-
-    # go over all projects and copy file from src to dst
-    for page in pages:
-        for item in page:
-            # download item (without save to disk)
-            buffer = item.download(save_locally=False)
-            # give the items name to the buffer
-            buffer.name = item.name
-            # upload item
-            new_item = dataset_to.items.upload(local_path=buffer,
-                                               remote_path='/destination_folder')
-            print(new_item.filename)
+def main(first_project_name, second_project_name, first_dataset_name, second_dataset_name):
+    """
+    Copy folder from project/dataset/folder
+    :return:
+    """
+    import dtlpy as dl
+
+    # Get source project and dataset
+    project = dl.projects.get(project_name=first_project_name)
+    dataset_from = project.datasets.get(dataset_name=first_dataset_name)
+    # filter to get all files of a specific folder
+    filters = dl.Filters()
+    filters.add(field='type', values='file')  # get only files
+    filters.add(field='filename', values='/source_folder/**')  # get all items in folder (recursive)
+    pages = dataset_from.items.list(filters=filters)
+
+    # Get destination project and annotations
+    project = dl.projects.get(project_name=second_project_name)
+    dataset_to = project.datasets.get(dataset_name=second_dataset_name)
+
+    # go over all projects and copy file from src to dst
+    for page in pages:
+        for item in page:
+            # download item (without save to disk)
+            buffer = item.download(save_locally=False)
+            # give the items name to the buffer
+            buffer.name = item.name
+            # upload item
+            new_item = dataset_to.items.upload(local_path=buffer,
+                                               remote_path='/destination_folder')
+            print(new_item.filename)

dtlpy/examples/create_annotations.py

@@ -1,51 +1,51 @@
-def main():
-    import dtlpy as dl
-    import numpy as np
-    import matplotlib.pyplot as plt
-
-    # Get project and dataset
-    project = dl.projects.get(project_name='Food')
-    dataset = project.datasets.get(dataset_name='BeansDataset')
-
-    # get item from platform
-    item = dataset.items.get(filepath='/image.jpg')
-
-    # Create a builder instance
-    builder = item.annotations.builder()
-
-    # add annotations of type box and label person
-    builder.add(annotation_definition=dl.Box(top=10,
-                                             left=10,
-                                             bottom=100,
-                                             right=100,
-                                             label='black_bean'))
-
-    # add annotations of type point with attribute
-    builder.add(annotation_definition=dl.Point(x=80,
-                                               y=80,
-                                               label='pea'),
-                attribute=['large'])
-
-    # add annotations of type polygon
-    builder.add(annotation_definition=dl.Polyline(geo=[[80, 40],
-                                                       [100, 120],
-                                                       [110, 130]],
-                                                  label='beans_can'))
-
-    # create a mask
-    mask = np.zeros(shape=(item.height, item.width), dtype=np.uint8)
-    # mark some part in the middle
-    mask[50:100, 200:250] = 1
-    # add annotations of type segmentation
-    builder.add(annotation_definition=dl.Segmentation(geo=mask,
-                                                      label='tomato_sauce'))
-
-    # plot the all of the annotations you created
-    plt.figure()
-    plt.imshow(builder.show())
-    for annotation in builder:
-        plt.figure()
-        plt.imshow(annotation.show())
-        plt.title(annotation.label)
-    # upload annotations to the item
-    item.annotations.upload(builder)
+def main():
+    import dtlpy as dl
+    import numpy as np
+    import matplotlib.pyplot as plt
+
+    # Get project and dataset
+    project = dl.projects.get(project_name='Food')
+    dataset = project.datasets.get(dataset_name='BeansDataset')
+
+    # get item from platform
+    item = dataset.items.get(filepath='/image.jpg')
+
+    # Create a builder instance
+    builder = item.annotations.builder()
+
+    # add annotations of type box and label person
+    builder.add(annotation_definition=dl.Box(top=10,
+                                             left=10,
+                                             bottom=100,
+                                             right=100,
+                                             label='black_bean'))
+
+    # add annotations of type point with attribute
+    builder.add(annotation_definition=dl.Point(x=80,
+                                               y=80,
+                                               label='pea'),
+                attribute=['large'])
+
+    # add annotations of type polygon
+    builder.add(annotation_definition=dl.Polyline(geo=[[80, 40],
+                                                       [100, 120],
+                                                       [110, 130]],
+                                                  label='beans_can'))
+
+    # create a mask
+    mask = np.zeros(shape=(item.height, item.width), dtype=np.uint8)
+    # mark some part in the middle
+    mask[50:100, 200:250] = 1
+    # add annotations of type segmentation
+    builder.add(annotation_definition=dl.Segmentation(geo=mask,
+                                                      label='tomato_sauce'))
+
+    # plot the all of the annotations you created
+    plt.figure()
+    plt.imshow(builder.show())
+    for annotation in builder:
+        plt.figure()
+        plt.imshow(annotation.show())
+        plt.title(annotation.label)
+    # upload annotations to the item
+    item.annotations.upload(builder)