The pysolotools from dinoimhof

pysolotools's Introduction

The open-source tool for loading and analyzing Unity SOLO datasets

Introduction

pysolotools is a python package for managing SOLO datasets. It helps to perform following tasks:

Parse SOLO datasets generated with Unity Perception
Dataset iterables
Convert SOLO to COCO format
Compute common statistics

You can read more about SOLO schema here.

Pre-Requisites

Install Anaconda or Miniconda (recommended). Install pre-commit.
Make sure pip is installed.

Installation

pip install pysolotools --index-url=https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple

SOLO Dataset

Load and iterate frames

from pysolotools.consumers import Solo
solo = Solo(data_path="src_data_path")

for frame in solo.frames():
    # perform operations on frame

SOLO2COCO conversion

Supports conversion for these labels: 2d bbox, keypoints, instance, semantic.

from pysolotools.converters.solo2coco import SOLO2COCOConverter
from pysolotools.consumers import Solo

solo = Solo("src_data_path")
dataset = SOLO2COCOConverter(solo)
dataset.convert(output_path="output_path")

Stats computation

Supports bbox, keypoints and image analysis on SOLO dataset.

from pysolotools.consumers import Solo
from pysolotools.stats.analyzers.bbox_analyzer import BBoxHeatMapStatsAnalyzer, BBoxSizeStatsAnalyzer
from pysolotools.stats.handler import StatsHandler

bbheat=BBoxHeatMapStatsAnalyzer()
bbsize=BBoxSizeStatsAnalyzer()
solo = Solo("data_path")
bbh= StatsHandler(solo=solo)
bbh.handle(analyzers=[bbheat,bbsize],cat_ids=[])

Community and Feedback

The Unity Computer Vision demos are open-source and we encourage and welcome contributions. If you wish to contribute, be sure to review our contribution guidelines and code of conduct.