Skip to content

yilei/pd-sdk

 
 

Repository files navigation

Parallel Domain SDK

Introduction

The Parallel Domain SDK (or short: PD SDK) allows the community to access Parallel Domain's synthetic data as Python objects.

The PD SDK can decode different data formats into its Python objects, including Dataset Governance Policy (DGP) format as well as CityScapes, NuImages and NuScenes (more public dataset formats will be supported in the future). Currently, local file system and s3 buckets are supported as dataset locations for decoding.

In order to support diffrent data formats PD SDK uses dataset format specific Decoders that are tasked with converting the respective dataset format into the PD SDK common Python objects (aka model classes). To support this the model is designed to represent an arbitrary sensor rig that collects sequential or non-sequential data that may be annotated with different annotations. PD SDK follows the principal of lazy loading data, meaning that any data is loaded as late as possible to ensure quick browsing through datasets. Furthermore PD SDK contains an encoding module tasked with saving model classes into specific dataset formats. This can be useful if you have an existing data pipeline that works on a certain format and you want to convert a dataset to this format to be compatible with your infrastructure.

The two main use cases PD SDK is designed for are:

  • to load data in ML data pipelines from local or cloud storage directly into RAM.
  • to encode data into different dataset formats. Currently it's possible to convert into DGP format from DGP, CityScapes, NuImages and NuScenes.

Quick Start

To use the PD SDK, you can simply install it using pip in your Python environment. Activate your Python environment before-hand, if wanted.

# Clone latest PD SDK release
$ git clone https://github.com/parallel-domain/pd-sdk.git

# Change directory
$ cd pd-sdk

# Optional: Parallelize build process for dependencies using gcc, e.g., `opencv-python-headless`
$ export MAKEFLAGS="-j$(nproc)"

# Install PD SDK from local clone
$ pip install .

Supported Python Versions:

  • Python3.7
  • Python3.8
  • Python3.9

Developers

If you are looking to extend or debug the PD SDK, use the following install procedure.

# Clone latest PD SDK release
$ git clone git@github.com:parallel-domain/pd-sdk.git

# Change directory
$ cd pd-sdk

# Optional: Parallelize build process for dependencies using gcc, e.g., `opencv-python-headless`
$ export MAKEFLAGS="-j$(nproc)"

# Install PD SDK from local clone with developer dependencies
$ pip install -e .[dev]

To make code contributions more stream-lined, we provide local pre-commit hooks that check / adapt to the PD SDK style guidelines automatically on commit. It is optional to set those up, but it helps to keep all code in the same style.

# within local PD SDK directory
$ pre-commit install

After setting up pre-commit hooks, every file in a commit will be checked by a number of hooks. If any of the hooks fail, the commit is rejected. Fortunately, most of the hooks are auto-correcting, so they edit the files per the style guidelines and you can just commit again. Only flake8 lint errors can just be solved by human editing.

Tests

Tests can be executed using pytest and providing a location to a DGP dataset which should be used during the test run.

# within local PD SDK directory - set env variable DATASET_PATH to a local or s3 location.
$ DATASET_PATH=/data/test_dataset pytest .

Documentation

You can find more details about adding to and building PD SDK's documentation in docs/README.md

Documentation

Tutorials

There are several tutorials available covering common use cases. Those can be found under Documentation -> Tutorials. In case you are missing an important tutorial, feel free to request it via a Github Issue or create a PR, in case you have written one already yourself.

API Reference

Public classes / methods / properties are annotated with Docstrings. The compiled API Reference can be found under Documentation -> API Reference

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%