Dronebase Machine Learning Data Utils

PUBLIC FACING

This package contains commonly used data functions for ML Engineers

import ml_dronebase_data_utils as data_utils
...

Object Detection Annotation Formatting

This package provides a Pascal VOC writer that renders *.xml annotation files for object detection tasks. It supports regular object detection and oriented object detection annotations with an additional <angle>some angle</angle> parameter.

from ml_dronebase_data_utils import PascalVOCWriter
writer = PascalVOCWriter()

for box in boxes:
    xmin, ymin, xmax, ymax, angle = box
    writer.addObject(
        name="some class name",
        xmin=xmin,
        ymin=ymin,
        xmax=xmax,
        ymax=ymax,
        angle=angle # Optional parameter
    )
writer.save(annotation_path)

S3 Data Utils

This package also provides common AWS S3 data functions like downloading data, uploading data (data or trained models), train/test split, etc.

Installation from source

Clone and cd into the root directory of this repo, then run the following:

pip install -e .

Installation using pip

pip install ml-dronebase-data-utils